mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
2220 lines
69 KiB
Go
2220 lines
69 KiB
Go
/*
|
|
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
|
|
Copyright (C) ITsysCOM GmbH
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
*/
|
|
package engine
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
var (
|
|
expTimeAttributes = time.Now().Add(time.Duration(20 * time.Minute))
|
|
attrService *AttributeService
|
|
dmAtr *DataManager
|
|
attrEvs = []*AttrArgsProcessEvent{
|
|
{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
CGREvent: &utils.CGREvent{ //matching AttributeProfile1
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Attribute": "AttributeProfile1",
|
|
utils.AnswerTime: time.Date(2014, 7, 14, 14, 30, 0, 0, time.UTC),
|
|
"UsageInterval": "1s",
|
|
utils.Weight: "20.0",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
CGREvent: &utils.CGREvent{ //matching AttributeProfile2
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Attribute": "AttributeProfile2",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
CGREvent: &utils.CGREvent{ //matching AttributeProfilePrefix
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Attribute": "AttributeProfilePrefix",
|
|
},
|
|
},
|
|
},
|
|
{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
CGREvent: &utils.CGREvent{ //matching AttributeProfilePrefix
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"DistinctMatch": 20,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
atrPs = AttributeProfiles{
|
|
&AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "AttributeProfile1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"FLTR_ATTR_1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
ExpiryTime: expTimeAttributes,
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Account,
|
|
Value: config.NewRSRParsersMustCompile("1010", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 20,
|
|
},
|
|
&AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "AttributeProfile2",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"FLTR_ATTR_2"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
ExpiryTime: expTimeAttributes,
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Account,
|
|
Value: config.NewRSRParsersMustCompile("1010", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 20,
|
|
},
|
|
&AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "AttributeProfilePrefix",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"FLTR_ATTR_3"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
ExpiryTime: expTimeAttributes,
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Account,
|
|
Value: config.NewRSRParsersMustCompile("1010", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 20,
|
|
},
|
|
&AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "AttributeIDMatch",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*gte:~*req.DistinctMatch:20"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
ExpiryTime: expTimeAttributes,
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Account,
|
|
Value: config.NewRSRParsersMustCompile("1010", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 20,
|
|
},
|
|
}
|
|
)
|
|
|
|
func TestAttributePopulateAttrService(t *testing.T) {
|
|
defaultCfg, _ := config.NewDefaultCGRConfig()
|
|
defaultCfg.AttributeSCfg().ProcessRuns = 1
|
|
defaultCfg.AttributeSCfg().StringIndexedFields = nil
|
|
defaultCfg.AttributeSCfg().PrefixIndexedFields = nil
|
|
defaultCfg.AttributeSCfg().PrefixIndexedFields = nil
|
|
data := NewInternalDB(nil, nil, true, defaultCfg.DataDbCfg().Items)
|
|
dmAtr = NewDataManager(data, config.CgrConfig().CacheCfg(), nil)
|
|
attrService, err = NewAttributeService(dmAtr, &FilterS{dm: dmAtr, cfg: defaultCfg}, defaultCfg)
|
|
if err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
}
|
|
|
|
func TestAttributeAddFilters(t *testing.T) {
|
|
fltrAttr1 := &Filter{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "FLTR_ATTR_1",
|
|
Rules: []*FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Attribute",
|
|
Values: []string{"AttributeProfile1"},
|
|
},
|
|
{
|
|
Type: utils.MetaGreaterOrEqual,
|
|
Element: "~*req.UsageInterval",
|
|
Values: []string{(1 * time.Second).String()},
|
|
},
|
|
{
|
|
Type: utils.MetaGreaterOrEqual,
|
|
Element: "~*req." + utils.Weight,
|
|
Values: []string{"9.0"},
|
|
},
|
|
},
|
|
}
|
|
dmAtr.SetFilter(fltrAttr1)
|
|
fltrAttr2 := &Filter{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "FLTR_ATTR_2",
|
|
Rules: []*FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Attribute",
|
|
Values: []string{"AttributeProfile2"},
|
|
},
|
|
},
|
|
}
|
|
dmAtr.SetFilter(fltrAttr2)
|
|
fltrAttrPrefix := &Filter{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "FLTR_ATTR_3",
|
|
Rules: []*FilterRule{
|
|
{
|
|
Type: utils.MetaPrefix,
|
|
Element: "~*req.Attribute",
|
|
Values: []string{"AttributeProfilePrefix"},
|
|
},
|
|
},
|
|
}
|
|
dmAtr.SetFilter(fltrAttrPrefix)
|
|
fltrAttr4 := &Filter{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "FLTR_ATTR_4",
|
|
Rules: []*FilterRule{
|
|
{
|
|
Type: utils.MetaGreaterOrEqual,
|
|
Element: "~*req." + utils.Weight,
|
|
Values: []string{"200.00"},
|
|
},
|
|
},
|
|
}
|
|
dmAtr.SetFilter(fltrAttr4)
|
|
}
|
|
|
|
func TestAttributeCache(t *testing.T) {
|
|
for _, atr := range atrPs {
|
|
if err = dmAtr.SetAttributeProfile(atr, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
}
|
|
//verify each attribute from cache
|
|
for _, atr := range atrPs {
|
|
if tempAttr, err := dmAtr.GetAttributeProfile(atr.Tenant, atr.ID,
|
|
true, false, utils.NonTransactional); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
} else if !reflect.DeepEqual(atr, tempAttr) {
|
|
t.Errorf("Expecting: %+v, received: %+v", atr, tempAttr)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAttributeProfileForEvent(t *testing.T) {
|
|
atrp, err := attrService.attributeProfileForEvent(attrEvs[0])
|
|
if err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(atrPs[0], atrp) {
|
|
t.Errorf("Expecting: %+v, received: %+v", utils.ToJSON(atrPs[0]), utils.ToJSON(atrp))
|
|
}
|
|
|
|
atrp, err = attrService.attributeProfileForEvent(attrEvs[1])
|
|
if err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(atrPs[1], atrp) {
|
|
t.Errorf("Expecting: %+v, received: %+v", utils.ToJSON(atrPs[1]), utils.ToJSON(atrp))
|
|
}
|
|
|
|
atrp, err = attrService.attributeProfileForEvent(attrEvs[2])
|
|
if err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(atrPs[2], atrp) {
|
|
t.Errorf("Expecting: %+v, received: %+v", utils.ToJSON(atrPs[2]), utils.ToJSON(atrp))
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessEvent(t *testing.T) {
|
|
attrEvs[0].CGREvent.Event["Account"] = "1010" //Field added in event after process
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"AttributeProfile1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Account"},
|
|
CGREvent: attrEvs[0].CGREvent,
|
|
}
|
|
atrp, err := attrService.processEvent(attrEvs[0])
|
|
if err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply, atrp) {
|
|
t.Errorf("Expecting: %+v, received: %+v", utils.ToJSON(eRply), utils.ToJSON(atrp))
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessEventWithNotFound(t *testing.T) {
|
|
attrEvs[3].CGREvent.Event["Account"] = "1010" //Field added in event after process
|
|
if _, err := attrService.processEvent(attrEvs[3]); err == nil || err != utils.ErrNotFound {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessEventWithIDs(t *testing.T) {
|
|
attrEvs[3].CGREvent.Event["Account"] = "1010" //Field added in event after process
|
|
attrEvs[3].AttributeIDs = []string{"AttributeIDMatch"}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"AttributeIDMatch"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Account"},
|
|
CGREvent: attrEvs[3].CGREvent,
|
|
}
|
|
if atrp, err := attrService.processEvent(attrEvs[3]); err != nil {
|
|
} else if !reflect.DeepEqual(eRply, atrp) {
|
|
t.Errorf("Expecting: %+v, received: %+v", utils.ToJSON(eRply), utils.ToJSON(atrp))
|
|
}
|
|
}
|
|
|
|
func TestAttributeEventReplyDigest(t *testing.T) {
|
|
eRpl := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.Account, utils.Subject},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "testAttributeSProcessEvent",
|
|
Event: map[string]any{
|
|
utils.Account: "1001",
|
|
utils.Subject: "1001",
|
|
utils.Destinations: "+491511231234",
|
|
},
|
|
},
|
|
}
|
|
expRpl := "Account:1001,Subject:1001"
|
|
val := eRpl.Digest()
|
|
if !reflect.DeepEqual(val, expRpl) {
|
|
t.Errorf("Expecting : %+v, received: %+v", utils.ToJSON(expRpl), utils.ToJSON(val))
|
|
}
|
|
}
|
|
|
|
func TestAttributeEventReplyDigest2(t *testing.T) {
|
|
eRpl := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "testAttributeSProcessEvent",
|
|
Event: map[string]any{
|
|
utils.Account: "1001",
|
|
utils.Subject: "1001",
|
|
utils.Destinations: "+491511231234",
|
|
},
|
|
},
|
|
}
|
|
expRpl := ""
|
|
val := eRpl.Digest()
|
|
if !reflect.DeepEqual(val, expRpl) {
|
|
t.Errorf("Expecting : %+v, received: %+v", utils.ToJSON(expRpl), utils.ToJSON(val))
|
|
}
|
|
}
|
|
|
|
func TestAttributeEventReplyDigest3(t *testing.T) {
|
|
eRpl := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{"*req.Subject"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "testAttributeSProcessEvent",
|
|
Event: map[string]any{
|
|
utils.Account: "1001",
|
|
utils.Subject: "1001",
|
|
utils.Destinations: "+491511231234",
|
|
},
|
|
},
|
|
}
|
|
expRpl := "Subject:1001"
|
|
val := eRpl.Digest()
|
|
if !reflect.DeepEqual(val, expRpl) {
|
|
t.Errorf("Expecting : %+v, received: %+v", utils.ToJSON(expRpl), utils.ToJSON(val))
|
|
}
|
|
}
|
|
|
|
func TestAttributeEventReplyDigest4(t *testing.T) {
|
|
eRpl := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{"*req.Subject"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "testAttributeSProcessEvent",
|
|
Event: map[string]any{
|
|
utils.Account: "1001",
|
|
utils.Destinations: "+491511231234",
|
|
},
|
|
},
|
|
}
|
|
expRpl := ""
|
|
val := eRpl.Digest()
|
|
if !reflect.DeepEqual(val, expRpl) {
|
|
t.Errorf("Expecting : %+v, received: %+v", utils.ToJSON(expRpl), utils.ToJSON(val))
|
|
}
|
|
}
|
|
|
|
func TestAttributeIndexer(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf := &AttributeProfile{
|
|
Tenant: "cgrates.org",
|
|
ID: "AttrPrf",
|
|
Contexts: []string{utils.META_ANY},
|
|
FilterIDs: []string{"*string:~*req.Account:1007"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
ExpiryTime: expTimeAttributes,
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + utils.Account,
|
|
Value: config.NewRSRParsersMustCompile("1010", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 20,
|
|
}
|
|
if err := dmAtr.SetAttributeProfile(attrPrf, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
eIdxes := map[string]utils.StringMap{
|
|
"*string:~*req.Account:1007": {
|
|
"AttrPrf": true,
|
|
},
|
|
}
|
|
rfi1 := NewFilterIndexer(dmAtr, utils.AttributeProfilePrefix,
|
|
utils.ConcatenatedKey(attrPrf.Tenant, utils.META_ANY))
|
|
if rcvIdx, err := dmAtr.GetFilterIndexes(utils.PrefixToIndexCache[rfi1.itemType],
|
|
rfi1.dbKeySuffix, utils.MetaString, nil); err != nil {
|
|
t.Error(err)
|
|
} else {
|
|
if !reflect.DeepEqual(eIdxes, rcvIdx) {
|
|
t.Errorf("Expecting %+v, received: %+v", eIdxes, rcvIdx)
|
|
}
|
|
}
|
|
//Set AttributeProfile with new context (*sessions)
|
|
cpAttrPrf := new(AttributeProfile)
|
|
*cpAttrPrf = *attrPrf
|
|
cpAttrPrf.Contexts = []string{utils.MetaSessionS}
|
|
if err := dmAtr.SetAttributeProfile(cpAttrPrf, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
rfi2 := NewFilterIndexer(dmAtr, utils.AttributeProfilePrefix,
|
|
utils.ConcatenatedKey(attrPrf.Tenant, utils.MetaSessionS))
|
|
if rcvIdx, err := dmAtr.GetFilterIndexes(utils.PrefixToIndexCache[rfi2.itemType],
|
|
rfi2.dbKeySuffix, utils.MetaString, nil); err != nil {
|
|
t.Error(err)
|
|
} else {
|
|
if !reflect.DeepEqual(eIdxes, rcvIdx) {
|
|
t.Errorf("Expecting %+v, received: %+v", eIdxes, rcvIdx)
|
|
}
|
|
}
|
|
//verify if old index was deleted ( context *any)
|
|
if _, err := dmAtr.GetFilterIndexes(utils.PrefixToIndexCache[rfi1.itemType],
|
|
rfi1.dbKeySuffix, utils.MetaString, nil); err != utils.ErrNotFound {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessWithMultipleRuns1(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.InitialField:InitialValue"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field1",
|
|
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 10,
|
|
}
|
|
attrPrf2 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_2",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Value: config.NewRSRParsersMustCompile("Value2", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 20,
|
|
}
|
|
attrPrf3 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_3",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field2:Value2"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field3",
|
|
Value: config.NewRSRParsersMustCompile("Value3", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 30,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf2, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf3, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(4),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1", "ATTR_2", "ATTR_3"},
|
|
AlteredFields: []string{
|
|
utils.MetaReq + utils.NestingSep + "Field1",
|
|
utils.MetaReq + utils.NestingSep + "Field2",
|
|
utils.MetaReq + utils.NestingSep + "Field3",
|
|
},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
"Field1": "Value1",
|
|
"Field2": "Value2",
|
|
"Field3": "Value3",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Fatalf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessWithMultipleRuns2(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.InitialField:InitialValue"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field1",
|
|
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 10,
|
|
}
|
|
attrPrf2 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_2",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Value: config.NewRSRParsersMustCompile("Value2", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 20,
|
|
}
|
|
attrPrf3 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_3",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.NotFound:NotFound"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field3",
|
|
Value: config.NewRSRParsersMustCompile("Value3", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 30,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf2, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf3, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(4),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1", "ATTR_2"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field1",
|
|
utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
"Field1": "Value1",
|
|
"Field2": "Value2",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessWithMultipleRuns3(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.InitialField:InitialValue"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field1",
|
|
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 10,
|
|
}
|
|
attrPrf2 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_2",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Value: config.NewRSRParsersMustCompile("Value2", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 20,
|
|
}
|
|
attrPrf3 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_3",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field2:Value2"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field3",
|
|
Value: config.NewRSRParsersMustCompile("Value3", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 30,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf2, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf3, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(2),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1", "ATTR_2"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field1",
|
|
utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
"Field1": "Value1",
|
|
"Field2": "Value2",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessWithMultipleRuns4(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.InitialField:InitialValue"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field1",
|
|
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 10,
|
|
}
|
|
attrPrf2 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_2",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Value: config.NewRSRParsersMustCompile("Value2", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 20,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf2, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(4),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1", "ATTR_2"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field1",
|
|
utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
"Field1": "Value1",
|
|
"Field2": "Value2",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeMultipleProcessWithBlocker(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
return
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.InitialField:InitialValue"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field1",
|
|
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 10,
|
|
}
|
|
attrPrf2 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_2",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Value: config.NewRSRParsersMustCompile("Value2", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 20,
|
|
}
|
|
attrPrf3 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_3",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field2:Value2"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field3",
|
|
Value: config.NewRSRParsersMustCompile("Value3", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 30,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf2, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf3, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(4),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1", "ATTR_2"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field1",
|
|
utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
"Field1": "Value1",
|
|
"Field2": "Value2",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeMultipleProcessWithBlocker2(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.InitialField:InitialValue"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field1",
|
|
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
attrPrf2 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_2",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Value: config.NewRSRParsersMustCompile("Value2", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 20,
|
|
}
|
|
attrPrf3 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_3",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field2:Value2"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field3",
|
|
Value: config.NewRSRParsersMustCompile("Value3", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 30,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf2, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if err = dmAtr.SetAttributeProfile(attrPrf3, true); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(4),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field1"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"InitialField": "InitialValue",
|
|
"Field1": "Value1",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessValue(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Value: config.NewRSRParsersMustCompile("~*req.Field1", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"Field2": "Value1",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeAttributeFilterIDs(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.META_ANY},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
FilterIDs: []string{"*string:~*req.PassField:Test"},
|
|
Path: utils.MetaReq + utils.NestingSep + "PassField",
|
|
Value: config.NewRSRParsersMustCompile("Pass", true, utils.INFIELD_SEP),
|
|
},
|
|
{
|
|
FilterIDs: []string{"*string:~*req.PassField:RandomValue"},
|
|
Path: utils.MetaReq + utils.NestingSep + "NotPassField",
|
|
Value: config.NewRSRParsersMustCompile("NotPass", true, utils.INFIELD_SEP),
|
|
},
|
|
{
|
|
FilterIDs: []string{"*notexists:~*req.RandomField:"},
|
|
Path: utils.MetaReq + utils.NestingSep + "RandomField",
|
|
Value: config.NewRSRParsersMustCompile("RandomValue", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"PassField": "Test",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "PassField",
|
|
utils.MetaReq + utils.NestingSep + "RandomField"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"PassField": "Pass",
|
|
"RandomField": "RandomValue",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessEventConstant(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.META_CONSTANT,
|
|
Value: config.NewRSRParsersMustCompile("ConstVal", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"Field2": "ConstVal",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessEventVariable(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("~*req.Field1", true, utils.INFIELD_SEP),
|
|
},
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("~*req.TheField", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"TheField": "TheVal",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"Field2": "TheVal",
|
|
"TheField": "TheVal",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessEventComposed(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.META_COMPOSED,
|
|
Value: config.NewRSRParsersMustCompile("~*req.Field1", true, utils.INFIELD_SEP),
|
|
},
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.META_COMPOSED,
|
|
Value: config.NewRSRParsersMustCompile("_", true, utils.INFIELD_SEP),
|
|
},
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.META_COMPOSED,
|
|
Value: config.NewRSRParsersMustCompile("~*req.TheField", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"TheField": "TheVal",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"Field2": "Value1_TheVal",
|
|
"TheField": "TheVal",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessEventSum(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.MetaSum,
|
|
Value: config.NewRSRParsersMustCompile("10;~*req.NumField;20", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"TheField": "TheVal",
|
|
"NumField": "20",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"TheField": "TheVal",
|
|
"NumField": "20",
|
|
"Field2": "50",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessEventUsageDifference(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.META_USAGE_DIFFERENCE,
|
|
Value: config.NewRSRParsersMustCompile("~*req.UnixTimeStamp;~*req.UnixTimeStamp2", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"TheField": "TheVal",
|
|
"UnixTimeStamp": "1554364297",
|
|
"UnixTimeStamp2": "1554364287",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"TheField": "TheVal",
|
|
"UnixTimeStamp": "1554364297",
|
|
"UnixTimeStamp2": "1554364287",
|
|
"Field2": "10s",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributeProcessEventValueExponent(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.MetaValueExponent,
|
|
Value: config.NewRSRParsersMustCompile("~*req.Multiplier;~*req.Pow", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"TheField": "TheVal",
|
|
"Multiplier": "2",
|
|
"Pow": "3",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "Field2"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
"TheField": "TheVal",
|
|
"Multiplier": "2",
|
|
"Pow": "3",
|
|
"Field2": "2000",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func BenchmarkAttributeProcessEventConstant(b *testing.B) {
|
|
defaultCfg, _ := config.NewDefaultCGRConfig()
|
|
defaultCfg.AttributeSCfg().ProcessRuns = 1
|
|
data := NewInternalDB(nil, nil, true, defaultCfg.DataDbCfg().Items)
|
|
dmAtr = NewDataManager(data, config.CgrConfig().CacheCfg(), nil)
|
|
attrService, err = NewAttributeService(dmAtr, &FilterS{dm: dmAtr, cfg: defaultCfg}, defaultCfg)
|
|
if err != nil {
|
|
b.Errorf("Error: %+v", err)
|
|
}
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
b.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
b.Error(err)
|
|
} else if test != true {
|
|
b.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.META_CONSTANT,
|
|
Value: config.NewRSRParsersMustCompile("ConstVal", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
b.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
b.Errorf("Error: %+v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkAttributeProcessEventVariable(b *testing.B) {
|
|
defaultCfg, _ := config.NewDefaultCGRConfig()
|
|
defaultCfg.AttributeSCfg().ProcessRuns = 1
|
|
data := NewInternalDB(nil, nil, true, defaultCfg.DataDbCfg().Items)
|
|
dmAtr = NewDataManager(data, config.CgrConfig().CacheCfg(), nil)
|
|
attrService, err = NewAttributeService(dmAtr, &FilterS{dm: dmAtr, cfg: defaultCfg}, defaultCfg)
|
|
if err != nil {
|
|
b.Errorf("Error: %+v", err)
|
|
}
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
b.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
b.Error(err)
|
|
} else if test != true {
|
|
b.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"*string:~*req.Field1:Value1"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.MetaVariable,
|
|
Value: config.NewRSRParsersMustCompile("~*req.Field1", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
b.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"Field1": "Value1",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
b.ResetTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
b.Errorf("Error: %+v", err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGetAttributeProfileFromInline(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrID := "*sum:*req.Field2:10;~*req.NumField;20"
|
|
expAttrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: attrID,
|
|
Contexts: []string{utils.META_ANY},
|
|
Attributes: []*Attribute{{
|
|
Path: utils.MetaReq + utils.NestingSep + "Field2",
|
|
Type: utils.MetaSum,
|
|
Value: config.NewRSRParsersMustCompile("10;~*req.NumField;20", true, utils.INFIELD_SEP),
|
|
}},
|
|
}
|
|
attr, err := dm.GetAttributeProfile(config.CgrConfig().GeneralCfg().DefaultTenant, attrID, false, false, "")
|
|
if err != nil {
|
|
t.Error(err)
|
|
} else if !reflect.DeepEqual(expAttrPrf1, attr) {
|
|
t.Errorf("Expecting %+v, received: %+v", utils.ToJSON(expAttrPrf1), utils.ToJSON(attr))
|
|
}
|
|
}
|
|
|
|
func TestAttributesProcessEventSIPCID(t *testing.T) {
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if test, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if test != true {
|
|
t.Errorf("\nExpecting: true got :%+v", test)
|
|
}
|
|
attrPrf1 := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "OriginID",
|
|
Type: utils.MetaSIPCID,
|
|
Value: config.NewRSRParsersMustCompile("~*req.cid;~*req.to;~*req.from", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf1, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"cid": "12345",
|
|
"to": "1001",
|
|
"from": "1002",
|
|
},
|
|
},
|
|
}
|
|
eRply := &AttrSProcessEventReply{
|
|
MatchedProfiles: []string{"ATTR_1"},
|
|
AlteredFields: []string{utils.MetaReq + utils.NestingSep + "OriginID"},
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"cid": "12345",
|
|
"to": "1001",
|
|
"from": "1002",
|
|
"OriginID": "12345;1001;1002",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
|
|
attrArgs.CGREvent.Event = map[string]any{
|
|
"cid": "12345",
|
|
"to": "1002",
|
|
"from": "1001",
|
|
}
|
|
eRply.CGREvent.Event = map[string]any{
|
|
"cid": "12345",
|
|
"to": "1002",
|
|
"from": "1001",
|
|
"OriginID": "12345;1001;1002",
|
|
}
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err != nil {
|
|
t.Errorf("Error: %+v", err)
|
|
}
|
|
if !reflect.DeepEqual(eRply.MatchedProfiles, reply.MatchedProfiles) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, reply.MatchedProfiles)
|
|
}
|
|
if !reflect.DeepEqual(eRply.AlteredFields, reply.AlteredFields) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, reply.AlteredFields)
|
|
}
|
|
if !reflect.DeepEqual(eRply.CGREvent.Event, reply.CGREvent.Event) {
|
|
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, reply.CGREvent.Event)
|
|
}
|
|
}
|
|
|
|
func TestAttributesProcessEventSIPCIDNotFoundErr(t *testing.T) {
|
|
defaultCfg, _ := config.NewDefaultCGRConfig()
|
|
data := NewInternalDB(nil, nil, true, defaultCfg.DataDbCfg().Items)
|
|
dmAtr = NewDataManager(data, config.CgrConfig().CacheCfg(), nil)
|
|
attrService, err = NewAttributeService(dmAtr, &FilterS{dm: dmAtr, cfg: defaultCfg}, defaultCfg)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if empty, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if !empty {
|
|
t.Error("expected DataDB to be empty")
|
|
}
|
|
attrPrf := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_ID",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "OriginID",
|
|
Type: utils.MetaSIPCID,
|
|
Value: config.NewRSRParsersMustCompile("~*req.cid;~*req.to;~*req.from", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err == nil ||
|
|
err != utils.ErrNotFound {
|
|
t.Errorf("expected: <%+v>, \nreceived: <%+v>", utils.ErrNotFound, err)
|
|
}
|
|
}
|
|
|
|
func TestAttributesProcessEventSIPCIDWrongPathErr(t *testing.T) {
|
|
defaultCfg, _ := config.NewDefaultCGRConfig()
|
|
data := NewInternalDB(nil, nil, true, defaultCfg.DataDbCfg().Items)
|
|
dmAtr = NewDataManager(data, config.CgrConfig().CacheCfg(), nil)
|
|
attrService, err = NewAttributeService(dmAtr, &FilterS{dm: dmAtr, cfg: defaultCfg}, defaultCfg)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if empty, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if !empty {
|
|
t.Error("expected DataDB to be empty")
|
|
}
|
|
attrPrf := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_ID",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "OriginID",
|
|
Type: utils.MetaSIPCID,
|
|
Value: config.NewRSRParsersMustCompile("~*req.cid;~*req.123.a", true, utils.INFIELD_SEP),
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"cid": "12345",
|
|
"123": struct{}{},
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
experr := `SERVER_ERROR: WRONG_PATH`
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err == nil ||
|
|
err.Error() != experr {
|
|
t.Errorf("expected: <%+v>, \nreceived: <%+v>", experr, err)
|
|
}
|
|
}
|
|
|
|
func TestAttributesProcessEventSIPCIDInvalidArgs(t *testing.T) {
|
|
Cache.Clear(nil)
|
|
defaultCfg, _ := config.NewDefaultCGRConfig()
|
|
data := NewInternalDB(nil, nil, true, defaultCfg.DataDbCfg().Items)
|
|
dmAtr = NewDataManager(data, config.CgrConfig().CacheCfg(), nil)
|
|
attrService, err = NewAttributeService(dmAtr, &FilterS{dm: dmAtr, cfg: defaultCfg}, defaultCfg)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
//refresh the DM
|
|
if err := dmAtr.DataDB().Flush(""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
Cache.Clear(nil)
|
|
if empty, err := dmAtr.DataDB().IsDBEmpty(); err != nil {
|
|
t.Error(err)
|
|
} else if !empty {
|
|
t.Error("expected DataDB to be empty")
|
|
}
|
|
attrPrf := &AttributeProfile{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: "ATTR_ID",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
Attributes: []*Attribute{
|
|
{
|
|
Path: utils.MetaReq + utils.NestingSep + "OriginID",
|
|
Type: utils.MetaSIPCID,
|
|
Value: config.RSRParsers{},
|
|
},
|
|
},
|
|
Blocker: true,
|
|
Weight: 10,
|
|
}
|
|
// Add attribute in DM
|
|
if err := dmAtr.SetAttributeProfile(attrPrf, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
attrArgs := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer(utils.MetaSessionS),
|
|
ProcessRuns: utils.IntPointer(1),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
|
|
ID: utils.GenUUID(),
|
|
Event: map[string]any{
|
|
"cid": "12345",
|
|
"to": "1001",
|
|
"from": "1002",
|
|
},
|
|
},
|
|
}
|
|
var reply AttrSProcessEventReply
|
|
experr := `SERVER_ERROR: invalid number of arguments <[]> to *sipcid`
|
|
if err := attrService.V1ProcessEvent(attrArgs, &reply); err == nil ||
|
|
err.Error() != experr {
|
|
t.Errorf("expected: <%+v>, \n received: <%+v>", experr, err)
|
|
}
|
|
}
|
|
|
|
func TestAttributeGetForEvent(t *testing.T) {
|
|
defaultCfg, _ := config.NewDefaultCGRConfig()
|
|
data := NewInternalDB(nil, nil, true, defaultCfg.DataDbCfg().Items)
|
|
dmAtr = NewDataManager(data, config.CgrConfig().CacheCfg(), nil)
|
|
attrService, err = NewAttributeService(dmAtr, NewFilterS(defaultCfg, nil, dmAtr), defaultCfg)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
args := &AttrArgsProcessEvent{
|
|
Context: utils.StringPointer("*sessions"),
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "testAttributeSGetAttributeForEvent",
|
|
Event: map[string]any{
|
|
utils.EVENT_NAME: "Internal",
|
|
utils.Account: "1003",
|
|
},
|
|
},
|
|
ArgDispatcher: &utils.ArgDispatcher{
|
|
APIKey: utils.StringPointer("attr12345"),
|
|
},
|
|
}
|
|
fltr := &Filter{
|
|
Tenant: "cgrates.org",
|
|
ID: "ATTR_FLTR",
|
|
Rules: []*FilterRule{
|
|
{
|
|
Type: utils.MetaString,
|
|
Element: "~*req.Account",
|
|
Values: []string{"1003"},
|
|
},
|
|
},
|
|
}
|
|
dmAtr.SetFilter(fltr)
|
|
attr := &AttributeProfile{
|
|
Tenant: "cgrates.org",
|
|
ID: "attributeprofile1",
|
|
Contexts: []string{utils.MetaSessionS},
|
|
FilterIDs: []string{"ATTR_FLTR"},
|
|
ActivationInterval: &utils.ActivationInterval{
|
|
ActivationTime: time.Date(2024, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
ExpiryTime: time.Date(2025, 7, 14, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
|
|
Weight: 20,
|
|
}
|
|
if err := dmAtr.SetAttributeProfile(attr, true); err != nil {
|
|
t.Error(err)
|
|
}
|
|
var reply AttributeProfile
|
|
if err := attrService.V1GetAttributeForEvent(args, &reply); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestAttrSGetAttributeForEventErrs(t *testing.T) {
|
|
cfg, _ := config.NewDefaultCGRConfig()
|
|
db := NewInternalDB(nil, nil, true, cfg.DataDbCfg().Items)
|
|
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
|
attrService, err := NewAttributeService(dm, NewFilterS(cfg, nil, dm), cfg)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
testCases := []struct {
|
|
name string
|
|
args *AttrArgsProcessEvent
|
|
attrPrfl *AttributeProfile
|
|
}{
|
|
{
|
|
name: "Missing CgrEvent",
|
|
args: &AttrArgsProcessEvent{},
|
|
},
|
|
{
|
|
name: "AttributeProfile not found",
|
|
args: &AttrArgsProcessEvent{
|
|
CGREvent: &utils.CGREvent{
|
|
Tenant: "cgrates.org",
|
|
ID: "CGREvent1",
|
|
Event: map[string]any{
|
|
utils.Account: "1002",
|
|
utils.Subject: "1002",
|
|
utils.Destination: "1001",
|
|
utils.SetupTime: time.Date(2022, 12, 1, 14, 25, 0, 0, time.UTC),
|
|
utils.Usage: "1m20s",
|
|
},
|
|
},
|
|
},
|
|
attrPrfl: new(AttributeProfile),
|
|
},
|
|
}
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
err := attrService.V1GetAttributeForEvent(tc.args, tc.attrPrfl)
|
|
if err == nil {
|
|
t.Error("expected error")
|
|
}
|
|
})
|
|
}
|
|
}
|