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

Added tests for RemoveSupplierProfile,RemoveAttributesProfile,RemoveStatQueueProfile,RemoveThresholdProfile,RemoveResourceProfile

Browse Source
This commit is contained in:
edwardro22
2018-01-06 06:12:39 +02:00
committed by Dan Christian Bogos
parent b09ad09e25
commit 170e70281d
4 changed files with 273 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
engine/onstor_it_test.go
View File

@@ -2218,7 +2218,7 @@ func testOnStorITCRUDResourceProfile(t *testing.T) {
// if err = onStor.DataDB().SelectDatabase(onStorCfg); err != nil {
// t.Error(err)
// }
if err := onStor.RemoveResourceProfile(rL.Tenant, rL.ID, utils.NonTransactional); err != nil {
if err := onStor.RemoveResourceProfile(rL.Tenant, rL.ID, utils.NonTransactional, false); err != nil {
t.Error(err)
}
if _, rcvErr := onStor.GetResourceProfile(rL.Tenant, rL.ID, true, utils.NonTransactional); rcvErr != utils.ErrNotFound {
@@ -2375,7 +2375,7 @@ func testOnStorITCRUDStatQueueProfile(t *testing.T) {
if _, ok := cache.Get(utils.StatQueueProfilePrefix + sq.ID); ok != false {
t.Error("Should not be in cache")
}
if err := onStor.RemoveStatQueueProfile(sq.Tenant, sq.ID, utils.NonTransactional); err != nil {
if err := onStor.RemoveStatQueueProfile(sq.Tenant, sq.ID, utils.NonTransactional, false); err != nil {
t.Error(err)
}
if _, ok := cache.Get(utils.StatQueueProfilePrefix + sq.ID); ok != false {
@@ -2478,7 +2478,7 @@ func testOnStorITCRUDThresholdProfile(t *testing.T) {
} else if !reflect.DeepEqual(th, rcv) {
t.Errorf("Expecting: %v, received: %v", th, rcv)
}
if err := onStor.RemoveThresholdProfile(th.Tenant, th.ID, utils.NonTransactional); err != nil {
if err := onStor.RemoveThresholdProfile(th.Tenant, th.ID, utils.NonTransactional, false); err != nil {
t.Error(err)
}
if _, rcvErr := onStor.GetThresholdProfile(th.Tenant, th.ID, true, utils.NonTransactional); rcvErr != utils.ErrNotFound {
@@ -2598,7 +2598,7 @@ func testOnStorITCRUDSupplierProfile(t *testing.T) {
} else if !reflect.DeepEqual(splProfile, rcv) {
t.Errorf("Expecting: %v, received: %v", splProfile, rcv)
}
if err := onStor.RemoveSupplierProfile(splProfile.Tenant, splProfile.ID, utils.NonTransactional); err != nil {
if err := onStor.RemoveSupplierProfile(splProfile.Tenant, splProfile.ID, utils.NonTransactional, false); err != nil {
t.Error(err)
}
if _, rcvErr := onStor.GetSupplierProfile("cgrates.org", "SPRF_1", true, utils.NonTransactional); rcvErr != nil && rcvErr != utils.ErrNotFound {
@@ -2642,7 +2642,7 @@ func testOnStorITCRUDAttributeProfile(t *testing.T) {
} else if !reflect.DeepEqual(attrProfile, rcv) {
t.Errorf("Expecting: %v, received: %v", attrProfile, rcv)
}
if err := onStor.RemoveAttributeProfile(attrProfile.Tenant, attrProfile.ID, utils.NonTransactional); err != nil {
if err := onStor.RemoveAttributeProfile(attrProfile.Tenant, attrProfile.ID, utils.NonTransactional, false); err != nil {
t.Error(err)
}
if _, rcvErr := onStor.GetAttributeProfile("cgrates.org", "AttrPrf1", true, utils.NonTransactional); rcvErr != nil && rcvErr != utils.ErrNotFound {

4
engine/storage_mongo_datadb.go
View File

@@ -1969,10 +1969,10 @@ func (ms *MongoStorage) SetFilterIndexesDrv(dbKey string, indexes map[string]uti
return
}
func (ms *MongoStorage) RemoveFilterIndexesDrv(id string) (err error) {
func (ms *MongoStorage) RemoveFilterIndexesDrv(dbKey string) (err error) {
session, col := ms.conn(colRFI)
defer session.Close()
err = col.Remove(bson.M{"key": id})
err = col.Remove(bson.M{"key": dbKey})
if err == mgo.ErrNotFound {
err = nil
}