uppal/cgrates
1
0
Fork 1
mirror of https://github.com/cgrates/cgrates.git synced 2026-02-21 15:18:44 +05:00
Code Issues Packages Projects Releases Wiki Activity

Added tests for Get/Set/Remove Account profile in z_onstor_it_test.go

Browse Source
This commit is contained in:
andronache
2021-01-07 10:41:59 +02:00
committed by Dan Christian Bogos
parent 3b230e084f
commit 8bcbd789e6
1 changed files with 70 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
engine/z_onstor_it_test.go
View File

@@ -77,7 +77,7 @@ var (
testOnStorITDispatcherProfile,
testOnStorITRateProfile,
testOnStorITActionProfile,
testOnStorITAccountProfile,
//testOnStorITCacheActionTriggers,
//testOnStorITCRUDActionTriggers,
}
@@ -2313,3 +2313,72 @@ func testOnStorITActionProfile(t *testing.T) {
t.Error(err)
}
}
func testOnStorITAccountProfile(t *testing.T) {
acctPrf := &utils.AccountProfile{
Tenant: "cgrates.org",
ID: "RP1",
FilterIDs: []string{"test_filterId"},
ActivationInterval: &utils.ActivationInterval{
ActivationTime: time.Date(2014, 7, 14, 14, 25, 0, 0, time.UTC),
ExpiryTime: time.Date(2014, 7, 15, 14, 25, 0, 0, time.UTC),
},
Weight: 2,
Balances: []*utils.Balance{{
ID: "VoiceBalance",
FilterIDs: []string{"FLTR_RES_GR2"},
Weight: 10,
Type: utils.VOICE,
Units: 3600,
Opts: map[string]interface{}{
"key1": "val1",
},
}},
ThresholdIDs: []string{"test_thrs"},
}
//empty in database
if _, err := onStor.GetAccountProfile("cgrates.org", "RP1",
true, false, utils.NonTransactional); err != utils.ErrNotFound {
t.Error(err)
}
//get from database
if err := onStor.SetAccountProfile(acctPrf, false); err != nil {
t.Error(err)
}
if rcv, err := onStor.GetAccountProfile("cgrates.org", "RP1",
true, false, utils.NonTransactional); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(rcv, acctPrf) {
t.Errorf("Expecting: %v, received: %v", acctPrf, rcv)
}
//craft akeysFromPrefix
expectedKey := []string{"anp_cgrates.org:RP1"}
if rcv, err := onStor.DataDB().GetKeysForPrefix(utils.AccountProfilePrefix); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(expectedKey, rcv) {
t.Errorf("Expecting: %v, received: %v", expectedKey, rcv)
}
//updateFilters
acctPrf.FilterIDs = []string{"*prefix:~*req.Destination:10"}
if err := onStor.SetAccountProfile(acctPrf, false); err != nil {
t.Error(err)
} else if rcv, err := onStor.GetAccountProfile("cgrates.org", "RP1",
false, false, utils.NonTransactional); err != nil {
t.Error(err)
} else if !reflect.DeepEqual(acctPrf, rcv) {
t.Errorf("Expecting: %v, received: %v", acctPrf, rcv)
}
//remove from database
if err := onStor.RemoveAccountProfile("cgrates.org", "RP1",
utils.NonTransactional, false); err != nil {
t.Error(err)
} else if _, err := onStor.GetAccountProfile("cgrates.org", "RP1",
false, false, utils.NonTransactional); err != utils.ErrNotFound {
t.Error(err)
}
}