diff --git a/console/chargers_profile.go b/console/chargers_profile.go index f518a26cc..cad307803 100644 --- a/console/chargers_profile.go +++ b/console/chargers_profile.go @@ -26,7 +26,7 @@ import ( func init() { c := &CmdGetChargers{ name: "chargers_profile", - rpcMethod: utils.APIerSv1GetChargerProfile, + rpcMethod: utils.AdminSv1GetChargerProfile, rpcParams: &utils.TenantID{}, } commands[c.Name()] = c diff --git a/console/chargers_profile_ids.go b/console/chargers_profile_ids.go index 2548866e0..b66cd052f 100644 --- a/console/chargers_profile_ids.go +++ b/console/chargers_profile_ids.go @@ -25,7 +25,7 @@ import ( func init() { c := &CmdGetChargerIDs{ name: "chargers_profile_ids", - rpcMethod: utils.APIerSv1GetChargerProfileIDs, + rpcMethod: utils.AdminSv1GetChargerProfileIDs, rpcParams: &utils.PaginatorWithTenant{}, } commands[c.Name()] = c diff --git a/console/chargers_profile_rem.go b/console/chargers_profile_rem.go index cc1176689..3f70f2844 100644 --- a/console/chargers_profile_rem.go +++ b/console/chargers_profile_rem.go @@ -23,7 +23,7 @@ import "github.com/cgrates/cgrates/utils" func init() { c := &CmdRemoveChargers{ name: "chargers_profile_remove", - rpcMethod: utils.APIerSv1RemoveChargerProfile, + rpcMethod: utils.AdminSv1RemoveChargerProfile, rpcParams: &utils.TenantIDWithAPIOpts{}, } commands[c.Name()] = c diff --git a/console/chargers_profile_set.go b/console/chargers_profile_set.go index e5f8bfe57..c33eab8ba 100644 --- a/console/chargers_profile_set.go +++ b/console/chargers_profile_set.go @@ -22,7 +22,7 @@ package console func init() { c := &CmdSetChargers{ name: "chargers_profile_set", - rpcMethod: utils.APIerSv1SetChargerProfile, + rpcMethod: utils.AdminSv1SetChargerProfile, rpcParams: &v1.ChargerWithAPIOpts{}, } commands[c.Name()] = c diff --git a/engine/z_filterindexer_it_test.go b/engine/z_filterindexer_it_test.go index 957a64334..8d10ca592 100644 --- a/engine/z_filterindexer_it_test.go +++ b/engine/z_filterindexer_it_test.go @@ -1365,9 +1365,9 @@ func testITChargerProfileIndexes(t *testing.T) { FilterIDs: []string{"CHARGER_FLTR", "*string:~*req.Usage:~*req.Debited"}, Weight: 10, } - if err := dataManager.SetChargerProfile(chrgr1, true); err != nil { + if err := dataManager.SetChargerProfile(context.Background(), chrgr1, true); err != nil { t.Error(err) - } else if err := dataManager.SetChargerProfile(chrgr2, true); err != nil { + } else if err := dataManager.SetChargerProfile(context.Background(), chrgr2, true); err != nil { t.Error(err) } @@ -1399,9 +1399,9 @@ func testITChargerProfileIndexes(t *testing.T) { } chrgr1.ID = "CHANGED_CHARGER_PRF1" chrgr2.ID = "CHANGED_CHARGER_PRF2" - if err := dataManager.SetChargerProfile(chrgr1, true); err != nil { + if err := dataManager.SetChargerProfile(context.Background(), chrgr1, true); err != nil { t.Error(err) - } else if err := dataManager.SetChargerProfile(chrgr2, true); err != nil { + } else if err := dataManager.SetChargerProfile(context.Background(), chrgr2, true); err != nil { t.Error(err) } diff --git a/engine/z_loader_it_test.go b/engine/z_loader_it_test.go index b9ac12dca..cc5d46fff 100644 --- a/engine/z_loader_it_test.go +++ b/engine/z_loader_it_test.go @@ -320,7 +320,7 @@ func testLoaderITWriteToDatabase(t *testing.T) { } for tenatid, cpp := range loader.chargerProfiles { - rcv, err := loader.dm.GetChargerProfile(tenatid.Tenant, tenatid.ID, false, false, utils.NonTransactional) + rcv, err := loader.dm.GetChargerProfile(context.Background(), tenatid.Tenant, tenatid.ID, false, false, utils.NonTransactional) if err != nil { t.Errorf("Failed GetChargerProfile, tenant: %s, id: %s, error: %s ", cpp.Tenant, cpp.ID, err.Error()) } diff --git a/engine/z_onstor_it_test.go b/engine/z_onstor_it_test.go index c1ee4b95f..cfcf9002c 100644 --- a/engine/z_onstor_it_test.go +++ b/engine/z_onstor_it_test.go @@ -908,15 +908,15 @@ func testOnStorITChargerProfile(t *testing.T) { AttributeIDs: []string{"ATTR_1"}, Weight: 20, } - if _, rcvErr := onStor.GetChargerProfile("cgrates.org", "CPP_1", + if _, rcvErr := onStor.GetChargerProfile(context.Background(), "cgrates.org", "CPP_1", true, false, utils.NonTransactional); rcvErr != nil && rcvErr != utils.ErrNotFound { t.Error(rcvErr) } - if err := onStor.SetChargerProfile(cpp, false); err != nil { + if err := onStor.SetChargerProfile(context.Background(), cpp, false); err != nil { t.Error(err) } //get from database - if rcv, err := onStor.GetChargerProfile("cgrates.org", "CPP_1", + if rcv, err := onStor.GetChargerProfile(context.Background(), "cgrates.org", "CPP_1", false, false, utils.NonTransactional); err != nil { t.Error(err) } else if !(reflect.DeepEqual(cpp, rcv)) { @@ -930,23 +930,23 @@ func testOnStorITChargerProfile(t *testing.T) { } //update cpp.FilterIDs = []string{"*string:~*req.Accout:1001", "*prefix:~*req.Destination:10"} - if err := onStor.SetChargerProfile(cpp, false); err != nil { + if err := onStor.SetChargerProfile(context.Background(), cpp, false); err != nil { t.Error(err) } //get from database - if rcv, err := onStor.GetChargerProfile("cgrates.org", "CPP_1", + if rcv, err := onStor.GetChargerProfile(context.Background(), "cgrates.org", "CPP_1", false, false, utils.NonTransactional); err != nil { t.Error(err) } else if !(reflect.DeepEqual(cpp, rcv)) { t.Errorf("Expecting: %v, received: %v", cpp, rcv) } - if err := onStor.RemoveChargerProfile(cpp.Tenant, cpp.ID, + if err := onStor.RemoveChargerProfile(context.Background(), cpp.Tenant, cpp.ID, false); err != nil { t.Error(err) } //check database if removed - if _, rcvErr := onStor.GetChargerProfile("cgrates.org", "CPP_1", + if _, rcvErr := onStor.GetChargerProfile(context.Background(), "cgrates.org", "CPP_1", false, false, utils.NonTransactional); rcvErr != nil && rcvErr != utils.ErrNotFound { t.Error(rcvErr) } diff --git a/migrator/chargers_it_test.go b/migrator/chargers_it_test.go index d5aaf9368..65ba5167e 100644 --- a/migrator/chargers_it_test.go +++ b/migrator/chargers_it_test.go @@ -21,6 +21,7 @@ along with this program. If not, see package migrator import ( + "github.com/cgrates/birpc/context" "log" "path" "reflect" @@ -194,10 +195,10 @@ func testChrgITMigrateAndMove(t *testing.T) { switch chrgAction { case utils.Migrate: // for the momment only one version of chargers exists case utils.Move: - if err := chrgMigrator.dmIN.DataManager().SetChargerProfile(chrgPrf, false); err != nil { + if err := chrgMigrator.dmIN.DataManager().SetChargerProfile(context.Background(), chrgPrf, false); err != nil { t.Error(err) } - if err := chrgMigrator.dmIN.DataManager().SetChargerProfile(chrgPrf2, false); err != nil { + if err := chrgMigrator.dmIN.DataManager().SetChargerProfile(context.Background(), chrgPrf2, false); err != nil { t.Error(err) } currentVersion := engine.CurrentDataDBVersions() @@ -206,7 +207,7 @@ func testChrgITMigrateAndMove(t *testing.T) { t.Error("Error when setting version for Chargers ", err.Error()) } - _, err = chrgMigrator.dmOut.DataManager().GetChargerProfile("cgrates.org", + _, err = chrgMigrator.dmOut.DataManager().GetChargerProfile(context.Background(), "cgrates.org", "CHRG_1", false, false, utils.NonTransactional) if err != utils.ErrNotFound { t.Error(err) @@ -216,23 +217,23 @@ func testChrgITMigrateAndMove(t *testing.T) { if err != nil { t.Error("Error when migrating Chargers ", err.Error()) } - if result, err := chrgMigrator.dmOut.DataManager().GetChargerProfile("cgrates.org", + if result, err := chrgMigrator.dmOut.DataManager().GetChargerProfile(context.Background(), "cgrates.org", "CHRG_1", false, false, utils.NonTransactional); err != nil { t.Fatal(err) } else if !reflect.DeepEqual(result, chrgPrf) { t.Errorf("Expecting: %+v, received: %+v", chrgPrf, result) } - if result, err := chrgMigrator.dmOut.DataManager().GetChargerProfile("cgrates.com", + if result, err := chrgMigrator.dmOut.DataManager().GetChargerProfile(context.Background(), "cgrates.com", "CHRG_1", false, false, utils.NonTransactional); err != nil { t.Fatal(err) } else if !reflect.DeepEqual(result, chrgPrf2) { t.Errorf("Expecting: %+v, received: %+v", chrgPrf2, result) } - if _, err = chrgMigrator.dmIN.DataManager().GetChargerProfile("cgrates.org", + if _, err = chrgMigrator.dmIN.DataManager().GetChargerProfile(context.Background(), "cgrates.org", "CHRG_1", false, false, utils.NonTransactional); err == nil || err != utils.ErrNotFound { t.Error(err) } - if _, err = chrgMigrator.dmIN.DataManager().GetChargerProfile("cgrates.com", + if _, err = chrgMigrator.dmIN.DataManager().GetChargerProfile(context.Background(), "cgrates.com", "CHRG_1", false, false, utils.NonTransactional); err == nil || err != utils.ErrNotFound { t.Error(err) } else if chrgMigrator.stats[utils.Chargers] != 2 { diff --git a/utils/consts.go b/utils/consts.go index 233b49827..eab9549bf 100644 --- a/utils/consts.go +++ b/utils/consts.go @@ -1322,10 +1322,10 @@ const ( ChargerSv1Ping = "ChargerSv1.Ping" ChargerSv1GetChargersForEvent = "ChargerSv1.GetChargersForEvent" ChargerSv1ProcessEvent = "ChargerSv1.ProcessEvent" - APIerSv1GetChargerProfile = "APIerSv1.GetChargerProfile" - APIerSv1RemoveChargerProfile = "APIerSv1.RemoveChargerProfile" - APIerSv1SetChargerProfile = "APIerSv1.SetChargerProfile" - APIerSv1GetChargerProfileIDs = "APIerSv1.GetChargerProfileIDs" + AdminSv1GetChargerProfile = "AdminSv1.GetChargerProfile" + AdminSv1RemoveChargerProfile = "AdminSv1.RemoveChargerProfile" + AdminSv1SetChargerProfile = "AdminSv1.SetChargerProfile" + AdminSv1GetChargerProfileIDs = "AdminSv1.GetChargerProfileIDs" ) // ThresholdS APIs