Files
cgrates/apier/v1/attributes_it_test.go
2019-04-11 09:25:00 +02:00

1574 lines
49 KiB
Go

// +build integration
/*
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 v1
import (
"net/rpc"
"net/rpc/jsonrpc"
"path"
"reflect"
"testing"
"time"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/engine"
"github.com/cgrates/cgrates/utils"
)
var (
alsPrfCfgPath string
alsPrfCfg *config.CGRConfig
attrSRPC *rpc.Client
alsPrfDataDir = "/usr/share/cgrates"
alsPrf *AttributeWithCache
alsPrfConfigDIR string //run tests for specific configuration
)
var sTestsAlsPrf = []func(t *testing.T){
testAttributeSInitCfg,
testAttributeSInitDataDb,
testAttributeSResetStorDb,
testAttributeSStartEngine,
testAttributeSRPCConn,
testAttributeSLoadFromFolder,
testAttributeSGetAttributeForEvent,
testAttributeSGetAttributeForEventNotFound,
testAttributeSGetAttributeForEventWithMetaAnyContext,
testAttributeSProcessEvent,
testAttributeSProcessEventNotFound,
testAttributeSProcessEventMissing,
testAttributeSProcessEventWithNoneSubstitute,
testAttributeSProcessEventWithNoneSubstitute2,
testAttributeSProcessEventWithNoneSubstitute3,
testAttributeSProcessEventWithHeader,
testAttributeSGetAttPrfIDs,
testAttributeSGetAlsPrfBeforeSet,
testAttributeSSetAlsPrf,
testAttributeSUpdateAlsPrf,
testAttributeSRemAlsPrf,
testAttributeSSetAlsPrf2,
testAttributeSSetAlsPrf3,
testAttributeSSetAlsPrf4,
testAttributeSPing,
testAttributeSProcessEventWithSearchAndReplace,
testAttributeSProcessWithMultipleRuns,
testAttributeSProcessWithMultipleRuns2,
testAttributeSKillEngine,
//start test for cache options
testAttributeSInitCfg,
testAttributeSInitDataDb,
testAttributeSResetStorDb,
testAttributeSStartEngine,
testAttributeSRPCConn,
testAttributeSCachingMetaNone,
testAttributeSCachingMetaLoad,
testAttributeSCachingMetaReload1,
testAttributeSCachingMetaReload2,
testAttributeSCachingMetaRemove,
testAttributeSKillEngine,
}
//Test start here
func TestAttributeSITMySql(t *testing.T) {
alsPrfConfigDIR = "tutmysql"
for _, stest := range sTestsAlsPrf {
t.Run(alsPrfConfigDIR, stest)
}
}
func TestAttributeSITMongo(t *testing.T) {
alsPrfConfigDIR = "tutmongo"
for _, stest := range sTestsAlsPrf {
t.Run(alsPrfConfigDIR, stest)
}
}
func testAttributeSInitCfg(t *testing.T) {
var err error
alsPrfCfgPath = path.Join(alsPrfDataDir, "conf", "samples", alsPrfConfigDIR)
alsPrfCfg, err = config.NewCGRConfigFromPath(alsPrfCfgPath)
if err != nil {
t.Error(err)
}
alsPrfCfg.DataFolderPath = alsPrfDataDir // Share DataFolderPath through config towards StoreDb for Flush()
config.SetCgrConfig(alsPrfCfg)
}
func testAttributeSInitDataDb(t *testing.T) {
if err := engine.InitDataDb(alsPrfCfg); err != nil {
t.Fatal(err)
}
}
// Wipe out the cdr database
func testAttributeSResetStorDb(t *testing.T) {
if err := engine.InitStorDb(alsPrfCfg); err != nil {
t.Fatal(err)
}
}
// Start CGR Engine
func testAttributeSStartEngine(t *testing.T) {
if _, err := engine.StopStartEngine(alsPrfCfgPath, *waitRater); err != nil {
t.Fatal(err)
}
}
// Connect rpc client to rater
func testAttributeSRPCConn(t *testing.T) {
var err error
attrSRPC, err = jsonrpc.Dial("tcp", alsPrfCfg.ListenCfg().RPCJSONListen) // We connect over JSON so we can also troubleshoot if needed
if err != nil {
t.Fatal(err)
}
}
func testAttributeSGetAlsPrfBeforeSet(t *testing.T) {
var reply *engine.AttributeProfile
if err := attrSRPC.Call("ApierV1.GetAttributeProfile",
&utils.TenantID{Tenant: "cgrates.org", ID: "ApierTest"},
&reply); err == nil || err.Error() != utils.ErrNotFound.Error() {
t.Error(err)
}
}
func testAttributeSLoadFromFolder(t *testing.T) {
var reply string
attrs := &utils.AttrLoadTpFromFolder{FolderPath: path.Join(*dataDir, "tariffplans", "oldtutorial")}
if err := attrSRPC.Call("ApierV1.LoadTariffPlanFromFolder", attrs, &reply); err != nil {
t.Error(err)
}
time.Sleep(500 * time.Millisecond)
}
func testAttributeSGetAttributeForEvent(t *testing.T) {
ev := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaSessionS),
CGREvent: utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSGetAttributeForEvent",
Event: map[string]interface{}{
utils.Account: "1007",
utils.Destination: "+491511231234",
},
},
}
eAttrPrf := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: ev.Tenant,
ID: "ATTR_1",
FilterIDs: []string{"*string:~Account:1007"},
Contexts: []string{utils.MetaSessionS, utils.MetaCDRs},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 1, 14, 0, 0, 0, 0, time.UTC)},
Attributes: []*engine.Attribute{
{
FieldName: utils.Account,
Value: config.NewRSRParsersMustCompile("1001", true, utils.INFIELD_SEP),
},
{
FieldName: utils.Subject,
Value: config.NewRSRParsersMustCompile("1001", true, utils.INFIELD_SEP),
},
},
Weight: 10.0,
},
}
eAttrPrf.Compile()
var attrReply *engine.AttributeProfile
if err := attrSRPC.Call(utils.AttributeSv1GetAttributeForEvent,
ev, &attrReply); err != nil {
t.Error(err)
}
if attrReply == nil {
t.Errorf("Expecting attrReply to not be nil")
// attrReply shoud not be nil so exit function
// to avoid nil segmentation fault;
// if this happens try to run this test manualy
return
}
attrReply.Compile() // Populate private variables in RSRParsers
if !reflect.DeepEqual(eAttrPrf.Attributes[0].Value[0], attrReply.Attributes[0].Value[0]) {
t.Errorf("Expecting: %+v, received: %+v", eAttrPrf.Attributes[0].Value[0], attrReply.Attributes[0].Value[0])
}
}
func testAttributeSGetAttributeForEventNotFound(t *testing.T) {
ev := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaCDRs),
CGREvent: utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSGetAttributeForEventWihMetaAnyContext",
Event: map[string]interface{}{
utils.Account: "dan",
utils.Destination: "+491511231234",
},
},
}
eAttrPrf2 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: ev.Tenant,
ID: "ATTR_3",
FilterIDs: []string{"*string:~Account:dan"},
Contexts: []string{utils.MetaSessionS},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 1, 14, 0, 0, 0, 0, time.UTC)},
Attributes: []*engine.Attribute{
{
FieldName: utils.Account,
Value: config.NewRSRParsersMustCompile("1001", true, utils.INFIELD_SEP),
},
},
Weight: 10.0,
},
}
eAttrPrf2.Compile()
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", eAttrPrf2, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply *engine.AttributeProfile
if err := attrSRPC.Call("ApierV1.GetAttributeProfile",
&utils.TenantID{Tenant: "cgrates.org", ID: "ATTR_3"}, &reply); err != nil {
t.Fatal(err)
}
reply.Compile()
if !reflect.DeepEqual(eAttrPrf2.AttributeProfile, reply) {
t.Errorf("Expecting : %+v, received: %+v", eAttrPrf2, reply)
}
var attrReply *engine.AttributeProfile
if err := attrSRPC.Call(utils.AttributeSv1GetAttributeForEvent,
ev, &attrReply); err == nil || err.Error() != utils.ErrNotFound.Error() {
t.Error(err)
}
}
func testAttributeSGetAttributeForEventWithMetaAnyContext(t *testing.T) {
ev := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaCDRs),
CGREvent: utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSGetAttributeForEventWihMetaAnyContext",
Event: map[string]interface{}{
utils.Account: "dan",
utils.Destination: "+491511231234",
},
},
}
eAttrPrf2 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: ev.Tenant,
ID: "ATTR_2",
FilterIDs: []string{"*string:~Account:dan"},
Contexts: []string{utils.META_ANY},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 1, 14, 0, 0, 0, 0, time.UTC)},
Attributes: []*engine.Attribute{
{
FieldName: utils.Account,
Value: config.NewRSRParsersMustCompile("1001", true, utils.INFIELD_SEP),
},
},
Weight: 10.0,
},
}
eAttrPrf2.Compile()
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", eAttrPrf2, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply *engine.AttributeProfile
if err := attrSRPC.Call("ApierV1.GetAttributeProfile",
&utils.TenantID{Tenant: "cgrates.org", ID: "ATTR_2"}, &reply); err != nil {
t.Fatal(err)
}
reply.Compile()
if !reflect.DeepEqual(eAttrPrf2.AttributeProfile, reply) {
t.Errorf("Expecting : %+v, received: %+v", eAttrPrf2.AttributeProfile, reply)
}
var attrReply *engine.AttributeProfile
if err := attrSRPC.Call(utils.AttributeSv1GetAttributeForEvent,
ev, &attrReply); err != nil {
t.Fatal(err)
}
attrReply.Compile()
if !reflect.DeepEqual(eAttrPrf2.AttributeProfile, attrReply) {
t.Errorf("Expecting: %s, received: %s", utils.ToJSON(eAttrPrf2.AttributeProfile), utils.ToJSON(attrReply))
}
}
func testAttributeSProcessEvent(t *testing.T) {
ev := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaSessionS),
CGREvent: utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSProcessEvent",
Event: map[string]interface{}{
utils.Account: "1007",
utils.Destination: "+491511231234",
},
},
}
eRply := &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"ATTR_1"},
AlteredFields: []string{utils.Subject, utils.Account},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSProcessEvent",
Event: map[string]interface{}{
utils.Account: "1001",
utils.Subject: "1001",
utils.Destination: "+491511231234",
},
},
}
eRply2 := &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"ATTR_1"},
AlteredFields: []string{utils.Account, utils.Subject},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSProcessEvent",
Event: map[string]interface{}{
utils.Account: "1001",
utils.Subject: "1001",
"Destination": "+491511231234",
},
},
}
var rplyEv engine.AttrSProcessEventReply
if err := attrSRPC.Call(utils.AttributeSv1ProcessEvent,
ev, &rplyEv); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(eRply, &rplyEv) &&
!reflect.DeepEqual(eRply2, &rplyEv) { // second for reversed order of attributes
t.Errorf("Expecting: %s, received: %s",
utils.ToJSON(eRply), utils.ToJSON(rplyEv))
}
}
func testAttributeSProcessEventNotFound(t *testing.T) {
ev := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaSessionS),
CGREvent: utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSProcessEventNotFound",
Event: map[string]interface{}{
utils.Account: "Inexistent",
utils.Destination: "+491511231234",
},
},
}
var rplyEv engine.AttrSProcessEventReply
if err := attrSRPC.Call(utils.AttributeSv1ProcessEvent,
ev, &rplyEv); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Error(err)
}
}
func testAttributeSProcessEventMissing(t *testing.T) {
ev := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaSessionS),
CGREvent: utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSProcessEvent",
Event: map[string]interface{}{
utils.Account: "NonExist",
utils.Category: "*attributes",
utils.Destination: "+491511231234",
},
},
}
var rplyEv engine.AttrSProcessEventReply
if err := attrSRPC.Call(utils.AttributeSv1ProcessEvent,
ev, &rplyEv); err == nil ||
err.Error() != utils.ErrMandatoryIeMissing.Error() {
t.Error(err)
}
}
func testAttributeSProcessEventWithNoneSubstitute(t *testing.T) {
ev := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaSessionS),
CGREvent: utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSWithNoneSubstitute",
Event: map[string]interface{}{
utils.Account: "1008",
utils.Destination: "+491511231234",
},
},
}
alsPrf = &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: "cgrates.org",
ID: "AttributeWithNonSubstitute",
Contexts: []string{utils.MetaSessionS, utils.MetaCDRs},
FilterIDs: []string{"*string:~Account:1008"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
ExpiryTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FilterIDs: []string{"*string:~Account:1008"},
FieldName: utils.Account,
Value: config.NewRSRParsersMustCompile("1001", true, utils.INFIELD_SEP),
},
{
FieldName: utils.Subject,
Value: config.NewRSRParsersMustCompile(utils.META_NONE, true, utils.INFIELD_SEP),
},
},
Weight: 20,
},
}
alsPrf.Compile()
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", alsPrf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
eRply := &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"AttributeWithNonSubstitute"},
AlteredFields: []string{utils.Account, utils.Subject},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSWithNoneSubstitute",
Event: map[string]interface{}{
utils.Account: "1001",
utils.Destination: "+491511231234",
},
},
}
var rplyEv engine.AttrSProcessEventReply
if err := attrSRPC.Call(utils.AttributeSv1ProcessEvent,
ev, &rplyEv); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(eRply, &rplyEv) {
t.Errorf("Expecting: %s, received: %s",
utils.ToJSON(eRply), utils.ToJSON(rplyEv))
}
}
func testAttributeSProcessEventWithNoneSubstitute2(t *testing.T) {
ev := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaSessionS),
CGREvent: utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSWithNoneSubstitute",
Event: map[string]interface{}{
utils.Account: "1008",
utils.Subject: "1008",
utils.Destination: "+491511231234",
},
},
}
alsPrf = &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: "cgrates.org",
ID: "AttributeWithNonSubstitute",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Account:1008"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
ExpiryTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FilterIDs: []string{"*string:~Account:1008"},
FieldName: utils.Account,
Value: config.NewRSRParsersMustCompile("1001", true, utils.INFIELD_SEP),
},
{
FieldName: utils.Subject,
Value: config.NewRSRParsersMustCompile(utils.META_NONE, true, utils.INFIELD_SEP),
},
},
Weight: 20,
},
}
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", alsPrf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
eRply := &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"AttributeWithNonSubstitute"},
AlteredFields: []string{"Account", "Subject"},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSWithNoneSubstitute",
Event: map[string]interface{}{
utils.Account: "1001",
utils.Destination: "+491511231234",
},
},
}
eRply2 := &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"AttributeWithNonSubstitute"},
AlteredFields: []string{utils.Subject, utils.Account},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSWithNoneSubstitute",
Event: map[string]interface{}{
utils.Account: "1001",
utils.Destination: "+491511231234",
},
},
}
var rplyEv engine.AttrSProcessEventReply
if err := attrSRPC.Call(utils.AttributeSv1ProcessEvent,
ev, &rplyEv); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(eRply, &rplyEv) &&
!reflect.DeepEqual(eRply2, &rplyEv) {
t.Errorf("Expecting: %s, received: %s",
utils.ToJSON(eRply), utils.ToJSON(rplyEv))
}
}
func testAttributeSProcessEventWithNoneSubstitute3(t *testing.T) {
ev := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaSessionS),
CGREvent: utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSWithNoneSubstitute",
Event: map[string]interface{}{
utils.Account: "1008",
utils.Subject: "1001",
utils.Destination: "+491511231234",
},
},
}
alsPrf = &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: "cgrates.org",
ID: "AttributeWithNonSubstitute",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Account:1008"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
ExpiryTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FilterIDs: []string{"*string:~Account:1008"},
FieldName: utils.Account,
Value: config.NewRSRParsersMustCompile("1001", true, utils.INFIELD_SEP),
},
{
FilterIDs: []string{"*string:~Subject:1008"},
FieldName: utils.Subject,
Value: config.NewRSRParsersMustCompile(utils.META_NONE, true, utils.INFIELD_SEP),
},
},
Weight: 20,
},
}
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", alsPrf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
eRply := &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"AttributeWithNonSubstitute"},
AlteredFields: []string{"Account"},
CGREvent: &utils.CGREvent{
Tenant: "cgrates.org",
ID: "testAttributeSWithNoneSubstitute",
Event: map[string]interface{}{
utils.Account: "1001",
utils.Subject: "1001",
utils.Destination: "+491511231234",
},
},
}
var rplyEv engine.AttrSProcessEventReply
if err := attrSRPC.Call(utils.AttributeSv1ProcessEvent,
ev, &rplyEv); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(eRply, &rplyEv) {
t.Errorf("Expecting: %s, received: %s",
utils.ToJSON(eRply), utils.ToJSON(rplyEv))
}
}
func testAttributeSProcessEventWithHeader(t *testing.T) {
attrPrf1 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_Header",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Field1:Value1"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field2",
Value: config.NewRSRParsersMustCompile("~Field1", true, utils.INFIELD_SEP),
},
},
Blocker: true,
Weight: 10,
},
}
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf1, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
attrArgs := &engine.AttrArgsProcessEvent{
ProcessRuns: utils.IntPointer(1),
Context: utils.StringPointer(utils.MetaSessionS),
CGREvent: utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "HeaderEventForAttribute",
Event: map[string]interface{}{
"Field1": "Value1",
},
},
}
eRply := &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"ATTR_Header"},
AlteredFields: []string{"Field2"},
CGREvent: &utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "HeaderEventForAttribute",
Event: map[string]interface{}{
"Field1": "Value1",
"Field2": "Value1",
},
},
}
var rplyEv engine.AttrSProcessEventReply
if err := attrSRPC.Call(utils.AttributeSv1ProcessEvent,
attrArgs, &rplyEv); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(eRply, &rplyEv) {
t.Errorf("Expecting: %s, received: %s",
utils.ToJSON(eRply), utils.ToJSON(rplyEv))
}
}
func testAttributeSGetAttPrfIDs(t *testing.T) {
expected := []string{"ATTR_2", "ATTR_1", "ATTR_3", "ATTR_Header", "AttributeWithNonSubstitute"}
var result []string
if err := attrSRPC.Call("ApierV1.GetAttributeProfileIDs", "", &result); err == nil ||
err.Error() != utils.NewErrMandatoryIeMissing("Tenant").Error() {
t.Errorf("Expected error recived reply %+v with err=%v", result, err)
}
if err := attrSRPC.Call("ApierV1.GetAttributeProfileIDs", "cgrates.org", &result); err != nil {
t.Error(err)
} else if len(expected) != len(result) {
t.Errorf("Expecting : %+v, received: %+v", expected, result)
}
}
func testAttributeSSetAlsPrf(t *testing.T) {
alsPrf = &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: "cgrates.org",
ID: "ApierTest",
Contexts: []string{utils.MetaSessionS, utils.MetaCDRs},
FilterIDs: []string{"FLTR_ACNT_dan", "FLTR_DST_DE"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
ExpiryTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "FL1",
Value: config.NewRSRParsersMustCompile("Al1", true, utils.INFIELD_SEP),
},
},
Weight: 20,
},
}
alsPrf.Compile()
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", alsPrf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply *engine.AttributeProfile
if err := attrSRPC.Call("ApierV1.GetAttributeProfile",
&utils.TenantID{Tenant: "cgrates.org", ID: "ApierTest"}, &reply); err != nil {
t.Fatal(err)
}
reply.Compile()
if !reflect.DeepEqual(alsPrf.AttributeProfile, reply) {
t.Errorf("Expecting : %+v, received: %+v", alsPrf.AttributeProfile, reply)
}
}
func testAttributeSUpdateAlsPrf(t *testing.T) {
alsPrf.Attributes = []*engine.Attribute{
{
FieldName: "FL1",
Value: config.NewRSRParsersMustCompile("Al1", true, utils.INFIELD_SEP),
},
{
FieldName: "FL2",
Value: config.NewRSRParsersMustCompile("Al2", true, utils.INFIELD_SEP),
},
}
alsPrf.Compile()
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", alsPrf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply *engine.AttributeProfile
if err := attrSRPC.Call("ApierV1.GetAttributeProfile",
&utils.TenantID{Tenant: "cgrates.org", ID: "ApierTest"}, &reply); err != nil {
t.Fatal(err)
}
reply.Compile()
if !reflect.DeepEqual(alsPrf.AttributeProfile, reply) {
t.Errorf("Expecting : %+v, received: %+v", alsPrf.AttributeProfile, reply)
}
}
func testAttributeSRemAlsPrf(t *testing.T) {
var resp string
if err := attrSRPC.Call("ApierV1.RemoveAttributeProfile",
&utils.TenantIDWithCache{Tenant: alsPrf.Tenant, ID: alsPrf.ID}, &resp); err != nil {
t.Error(err)
} else if resp != utils.OK {
t.Error("Unexpected reply returned", resp)
}
var reply *engine.AttributeProfile
if err := attrSRPC.Call("ApierV1.GetAttributeProfile",
&utils.TenantID{Tenant: "cgrates.org", ID: "ApierTest"},
&reply); err == nil || err.Error() != utils.ErrNotFound.Error() {
t.Error(err)
}
// remove twice shoud return not found
resp = ""
if err := attrSRPC.Call("ApierV1.RemoveAttributeProfile",
&utils.TenantIDWithCache{Tenant: alsPrf.Tenant, ID: alsPrf.ID}, &resp); err.Error() != utils.ErrNotFound.Error() {
t.Errorf("Expected error: %v recived: %v", utils.ErrNotFound, err)
}
}
func testAttributeSSetAlsPrf2(t *testing.T) {
alsPrf = &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: "golant",
ID: "ATTR_972587832508_SESSIONAUTH",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Account:972587832508"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
ExpiryTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: utils.Subject,
Value: config.RSRParsers{
&config.RSRParser{
Rules: "roam",
AllFiltersMatch: true,
},
},
},
},
Blocker: false,
Weight: 10,
},
}
alsPrf.Compile()
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", alsPrf, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply *engine.AttributeProfile
if err := attrSRPC.Call("ApierV1.GetAttributeProfile",
&utils.TenantID{Tenant: "golant", ID: "ATTR_972587832508_SESSIONAUTH"}, &reply); err != nil {
t.Fatal(err)
}
reply.Compile()
if !reflect.DeepEqual(alsPrf.AttributeProfile, reply) {
t.Errorf("Expecting : %+v, received: %+v", alsPrf.AttributeProfile, reply)
}
}
func testAttributeSSetAlsPrf3(t *testing.T) {
alsPrf = &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: "golant",
ID: "ATTR_972587832508_SESSIONAUTH",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:Account:972587832508"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
ExpiryTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: utils.Subject,
Value: config.RSRParsers{
&config.RSRParser{
Rules: "",
},
},
},
},
Blocker: false,
Weight: 10,
},
}
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", alsPrf, &result); err == nil {
t.Error(err)
}
}
func testAttributeSSetAlsPrf4(t *testing.T) {
alsPrf = &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: "golant",
ID: "ATTR_972587832508_SESSIONAUTH",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Account:972587832508"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
ExpiryTime: time.Date(2014, 7, 14, 14, 35, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: utils.Subject,
Value: config.RSRParsers{
&config.RSRParser{},
},
},
},
Blocker: false,
Weight: 10,
},
}
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", alsPrf, &result); err == nil {
t.Error(err)
}
}
func testAttributeSPing(t *testing.T) {
var resp string
if err := attrSRPC.Call(utils.AttributeSv1Ping, new(utils.CGREvent), &resp); err != nil {
t.Error(err)
} else if resp != utils.Pong {
t.Error("Unexpected reply returned", resp)
}
}
func testAttributeSProcessEventWithSearchAndReplace(t *testing.T) {
attrPrf1 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_Search_and_replace",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Category:call"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Category",
Value: config.NewRSRParsersMustCompile("~Category:s/(.*)/${1}_suffix/", true, utils.INFIELD_SEP),
},
},
Blocker: true,
Weight: 10,
},
}
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf1, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
attrArgs := &engine.AttrArgsProcessEvent{
ProcessRuns: utils.IntPointer(1),
Context: utils.StringPointer(utils.MetaSessionS),
CGREvent: utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "HeaderEventForAttribute",
Event: map[string]interface{}{
"Category": "call",
},
},
}
eRply := &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"ATTR_Search_and_replace"},
AlteredFields: []string{"Category"},
CGREvent: &utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "HeaderEventForAttribute",
Event: map[string]interface{}{
"Category": "call_suffix",
},
},
}
var rplyEv engine.AttrSProcessEventReply
if err := attrSRPC.Call(utils.AttributeSv1ProcessEvent,
attrArgs, &rplyEv); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(eRply, &rplyEv) {
t.Errorf("Expecting: %s, received: %s",
utils.ToJSON(eRply), utils.ToJSON(rplyEv))
}
}
func testAttributeSProcessWithMultipleRuns(t *testing.T) {
attrPrf1 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_1",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~InitialField:InitialValue"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field1",
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
},
},
Weight: 10,
},
}
attrPrf2 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_2",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Field1:Value1"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field2",
Value: config.NewRSRParsersMustCompile("Value2", true, utils.INFIELD_SEP),
},
},
Weight: 20,
},
}
attrPrf3 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_3",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~NotFound:NotFound"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field3",
Value: config.NewRSRParsersMustCompile("Value3", true, utils.INFIELD_SEP),
},
},
Weight: 30,
},
}
// Add attribute in DM
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf1, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf2, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf3, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
attrArgs := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaSessionS),
ProcessRuns: utils.IntPointer(4),
CGREvent: utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: utils.GenUUID(),
Event: map[string]interface{}{
"InitialField": "InitialValue",
},
},
}
eRply := &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"ATTR_1", "ATTR_2"},
AlteredFields: []string{"Field1", "Field2"},
CGREvent: &utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: utils.GenUUID(),
Event: map[string]interface{}{
"InitialField": "InitialValue",
"Field1": "Value1",
"Field2": "Value2",
},
},
}
var rplyEv engine.AttrSProcessEventReply
if err := attrSRPC.Call(utils.AttributeSv1ProcessEvent,
attrArgs, &rplyEv); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(eRply.MatchedProfiles, rplyEv.MatchedProfiles) {
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, rplyEv.MatchedProfiles)
} else if !reflect.DeepEqual(eRply.AlteredFields, rplyEv.AlteredFields) {
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, rplyEv.AlteredFields)
} else if !reflect.DeepEqual(eRply.CGREvent.Event, rplyEv.CGREvent.Event) {
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, rplyEv.CGREvent.Event)
}
}
func testAttributeSProcessWithMultipleRuns2(t *testing.T) {
attrPrf1 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_1",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~InitialField:InitialValue"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field1",
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
},
},
Weight: 10,
},
}
attrPrf2 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_2",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Field1:Value1"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field2",
Value: config.NewRSRParsersMustCompile("Value2", true, utils.INFIELD_SEP),
},
},
Weight: 20,
},
}
attrPrf3 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_3",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Field2:Value2"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field3",
Value: config.NewRSRParsersMustCompile("Value3", true, utils.INFIELD_SEP),
},
},
Weight: 30,
},
}
// Add attributeProfiles
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf1, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf2, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf3, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
attrArgs := &engine.AttrArgsProcessEvent{
Context: utils.StringPointer(utils.MetaSessionS),
ProcessRuns: utils.IntPointer(4),
CGREvent: utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: utils.GenUUID(),
Event: map[string]interface{}{
"InitialField": "InitialValue",
},
},
}
eRply := &engine.AttrSProcessEventReply{
MatchedProfiles: []string{"ATTR_1", "ATTR_2", "ATTR_3"},
AlteredFields: []string{"Field1", "Field2", "Field3"},
CGREvent: &utils.CGREvent{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: utils.GenUUID(),
Event: map[string]interface{}{
"InitialField": "InitialValue",
"Field1": "Value1",
"Field2": "Value2",
"Field3": "Value3",
},
},
}
var rplyEv engine.AttrSProcessEventReply
if err := attrSRPC.Call(utils.AttributeSv1ProcessEvent,
attrArgs, &rplyEv); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(eRply.MatchedProfiles, rplyEv.MatchedProfiles) {
t.Errorf("Expecting %+v, received: %+v", eRply.MatchedProfiles, rplyEv.MatchedProfiles)
} else if !reflect.DeepEqual(eRply.AlteredFields, rplyEv.AlteredFields) {
t.Errorf("Expecting %+v, received: %+v", eRply.AlteredFields, rplyEv.AlteredFields)
} else if !reflect.DeepEqual(eRply.CGREvent.Event, rplyEv.CGREvent.Event) {
t.Errorf("Expecting %+v, received: %+v", eRply.CGREvent.Event, rplyEv.CGREvent.Event)
}
}
func testAttributeSKillEngine(t *testing.T) {
if err := engine.KillEngine(100); err != nil {
t.Error(err)
}
}
//Start tests for caching
func testAttributeSCachingMetaNone(t *testing.T) {
//*none option should not add attribute in cache only in Datamanager
attrPrf1 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_1",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~InitialField:InitialValue"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field1",
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
},
},
Weight: 10,
},
Cache: utils.StringPointer(utils.META_NONE),
}
// set the profile
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf1, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply bool
argsCache := engine.ArgsGetCacheItem{
CacheID: utils.CacheAttributeProfiles,
ItemID: "cgrates.org:ATTR_1",
}
if err := attrSRPC.Call(utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if reply {
t.Errorf("Expected: false, received:%v", reply)
}
var rcvKeys []string
argsCache2 := engine.ArgsGetCacheItemIDs{
CacheID: utils.CacheAttributeProfiles,
}
if err := attrSRPC.Call(utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Fatalf("Expected error: %s received error: %s and reply: %v ",
utils.ErrNotFound, err.Error(), rcvKeys)
}
//check in dataManager
expected := []string{"ATTR_1"}
var rcvIDs []string
if err := attrSRPC.Call("ApierV1.GetAttributeProfileIDs", "cgrates.org", &rcvIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcvIDs) {
t.Errorf("Expecting : %+v, received: %+v", expected, rcvIDs)
}
}
func testAttributeSCachingMetaLoad(t *testing.T) {
//*load option should add attribute in cache and in Datamanager
attrPrf1 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_1",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~InitialField:InitialValue"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field1",
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
},
},
Weight: 10,
},
Cache: utils.StringPointer(utils.MetaLoad),
}
// set the profile
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf1, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply bool
argsCache := engine.ArgsGetCacheItem{
CacheID: utils.CacheAttributeProfiles,
ItemID: "cgrates.org:ATTR_1",
}
if err := attrSRPC.Call(utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if !reply {
t.Errorf("Expected: true, received:%v", reply)
}
var rcvKeys []string
expectedIDs := []string{"cgrates.org:ATTR_1"}
argsCache2 := engine.ArgsGetCacheItemIDs{
CacheID: utils.CacheAttributeProfiles,
}
if err := attrSRPC.Call(utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err != nil {
t.Fatal(err)
} else if !reflect.DeepEqual(rcvKeys, expectedIDs) {
t.Errorf("Expecting : %+v, received: %+v", expectedIDs, rcvKeys)
}
//check in dataManager
expected := []string{"ATTR_1"}
var rcvIDs []string
if err := attrSRPC.Call("ApierV1.GetAttributeProfileIDs", "cgrates.org", &rcvIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcvIDs) {
t.Errorf("Expecting : %+v, received: %+v", expected, rcvIDs)
}
//remove from cache and DataManager the profile
var resp string
if err := attrSRPC.Call("ApierV1.RemoveAttributeProfile",
&utils.TenantIDWithCache{Tenant: attrPrf1.Tenant, ID: attrPrf1.ID,
Cache: utils.StringPointer(utils.MetaRemove)}, &resp); err != nil {
t.Error(err)
} else if resp != utils.OK {
t.Error("Unexpected reply returned", resp)
}
argsCache = engine.ArgsGetCacheItem{
CacheID: utils.CacheAttributeProfiles,
ItemID: "cgrates.org:ATTR_1",
}
if err := attrSRPC.Call(utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if reply {
t.Errorf("Expected: false, received:%v", reply)
}
if err := attrSRPC.Call(utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Fatalf("Expected error: %s received error: %s and reply: %v ",
utils.ErrNotFound, err, rcvKeys)
}
//check in dataManager
if err := attrSRPC.Call("ApierV1.GetAttributeProfileIDs", "cgrates.org", &rcvIDs); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Fatalf("Expected error: %s received error: %s and reply: %v ",
utils.ErrNotFound, err, rcvIDs)
}
}
func testAttributeSCachingMetaReload1(t *testing.T) {
//*reload add the attributes in cache if was there before
attrPrf1 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_1",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~InitialField:InitialValue"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field1",
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
},
},
Weight: 10,
},
Cache: utils.StringPointer(utils.MetaReload),
}
// set the profile
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf1, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply bool
argsCache := engine.ArgsGetCacheItem{
CacheID: utils.CacheAttributeProfiles,
ItemID: "cgrates.org:ATTR_1",
}
if err := attrSRPC.Call(utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if reply {
t.Errorf("Expected: false, received:%v", reply)
}
var rcvKeys []string
argsCache2 := engine.ArgsGetCacheItemIDs{
CacheID: utils.CacheAttributeProfiles,
}
if err := attrSRPC.Call(utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Fatalf("Expected error: %s received error: %s and reply: %v ",
utils.ErrNotFound, err, rcvKeys)
}
//check in dataManager
expected := []string{"ATTR_1"}
var rcvIDs []string
if err := attrSRPC.Call("ApierV1.GetAttributeProfileIDs", "cgrates.org", &rcvIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcvIDs) {
t.Errorf("Expecting : %+v, received: %+v", expected, rcvIDs)
}
}
func testAttributeSCachingMetaReload2(t *testing.T) {
//add cache with *load option
attrPrf1 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_1",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~InitialField:InitialValue"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field1",
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
},
},
Weight: 10,
},
Cache: utils.StringPointer(utils.MetaLoad),
}
// set the profile
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf1, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply *engine.AttributeProfile
if err := attrSRPC.Call("ApierV1.GetAttributeProfile",
&utils.TenantID{Tenant: "cgrates.org", ID: "ATTR_1"}, &reply); err != nil {
t.Fatal(err)
}
attrPrf1.Compile()
reply.Compile()
if !reflect.DeepEqual(attrPrf1.AttributeProfile, reply) {
t.Errorf("Expecting : %+v, received: %+v", attrPrf1.AttributeProfile, reply)
}
//add cache with *reload option
// should overwrite the first
attrPrf2 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_1",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Test:Test"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field1",
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
},
},
Weight: 10,
},
Cache: utils.StringPointer(utils.MetaReload),
}
// set the profile
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf2, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
if err := attrSRPC.Call("ApierV1.GetAttributeProfile",
&utils.TenantID{Tenant: "cgrates.org", ID: "ATTR_1"}, &reply); err != nil {
t.Fatal(err)
}
attrPrf2.Compile()
reply.Compile()
if !reflect.DeepEqual(attrPrf2.AttributeProfile, reply) {
t.Errorf("Expecting : %+v, received: %+v", attrPrf2.AttributeProfile, reply)
}
}
func testAttributeSCachingMetaRemove(t *testing.T) {
//add cache with *load option
attrPrf1 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_1",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~InitialField:InitialValue"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field1",
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
},
},
Weight: 10,
},
Cache: utils.StringPointer(utils.MetaLoad),
}
// set the profile
var result string
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf1, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
var reply bool
argsCache := engine.ArgsGetCacheItem{
CacheID: utils.CacheAttributeProfiles,
ItemID: "cgrates.org:ATTR_1",
}
if err := attrSRPC.Call(utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if !reply {
t.Errorf("Expected: true, received:%v", reply)
}
var rcvKeys []string
expectedIDs := []string{"cgrates.org:ATTR_1"}
argsCache2 := engine.ArgsGetCacheItemIDs{
CacheID: utils.CacheAttributeProfiles,
}
if err := attrSRPC.Call(utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err != nil {
t.Fatal(err)
} else if !reflect.DeepEqual(rcvKeys, expectedIDs) {
t.Errorf("Expecting : %+v, received: %+v", expectedIDs, rcvKeys)
}
// add with *remove cache option
// should delete it from cache
attrPrf2 := &AttributeWithCache{
AttributeProfile: &engine.AttributeProfile{
Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
ID: "ATTR_1",
Contexts: []string{utils.MetaSessionS},
FilterIDs: []string{"*string:~Test:Test"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
},
Attributes: []*engine.Attribute{
{
FieldName: "Field1",
Value: config.NewRSRParsersMustCompile("Value1", true, utils.INFIELD_SEP),
},
},
Weight: 10,
},
Cache: utils.StringPointer(utils.MetaRemove),
}
// set the profile
if err := attrSRPC.Call("ApierV1.SetAttributeProfile", attrPrf2, &result); err != nil {
t.Error(err)
} else if result != utils.OK {
t.Error("Unexpected reply returned", result)
}
if err := attrSRPC.Call(utils.CacheSv1HasItem, argsCache, &reply); err != nil {
t.Error(err)
} else if reply {
t.Errorf("Expected: false, received:%v", reply)
}
if err := attrSRPC.Call(utils.CacheSv1GetItemIDs, argsCache2, &rcvKeys); err == nil ||
err.Error() != utils.ErrNotFound.Error() {
t.Fatalf("Expected error: %s received error: %s and reply: %v ",
utils.ErrNotFound, err.Error(), rcvKeys)
}
//check in dataManager
expected := []string{"ATTR_1"}
var rcvIDs []string
if err := attrSRPC.Call("ApierV1.GetAttributeProfileIDs", "cgrates.org", &rcvIDs); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expected, rcvIDs) {
t.Errorf("Expecting : %+v, received: %+v", expected, rcvIDs)
}
}