diff --git a/migrator/filters_it_test.go b/migrator/filters_it_test.go index 310719ae0..b25e0e50c 100644 --- a/migrator/filters_it_test.go +++ b/migrator/filters_it_test.go @@ -219,7 +219,7 @@ func testFltrITMigrateAndMove(t *testing.T) { t.Errorf("Unexpected version returned: %d", vrs[utils.RQF]) } //check if Filters was migrate correctly - result, err := engine.GetFilter(fltrMigrator.dmOut.DataManager(), Filters.Tenant, Filters.ID, false, false, utils.NonTransactional) + result, err := fltrMigrator.dmOut.DataManager().GetFilter(Filters.Tenant, Filters.ID, false, false, utils.NonTransactional) if err != nil { t.Fatalf("Error when getting Attributes %v", err.Error()) } @@ -260,7 +260,7 @@ func testFltrITMigrateAndMove(t *testing.T) { t.Error("Error when fltrMigratorrating Filters ", err.Error()) } //check if account was migrate correctly - result, err := engine.GetFilter(fltrMigrator.dmOut.DataManager(), expFilters.Tenant, expFilters.ID, false, false, utils.NonTransactional) + result, err := fltrMigrator.dmOut.DataManager().GetFilter(expFilters.Tenant, expFilters.ID, false, false, utils.NonTransactional) if err != nil { t.Error(err) } @@ -269,7 +269,7 @@ func testFltrITMigrateAndMove(t *testing.T) { t.Errorf("Expecting: %+v, received: %+v", utils.ToJSON(expFilters), utils.ToJSON(result)) } // check if old account was deleted - result, err = engine.GetFilter(fltrMigrator.dmIN.DataManager(), expFilters.Tenant, expFilters.ID, false, false, utils.NonTransactional) + result, err = fltrMigrator.dmIN.DataManager().GetFilter(expFilters.Tenant, expFilters.ID, false, false, utils.NonTransactional) if err != utils.ErrNotFound { t.Error(err) } @@ -385,7 +385,7 @@ func testFltrITMigratev2(t *testing.T) { t.Errorf("Unexpected version returned: %d", vrs[utils.RQF]) } //check if Filters was migrate correctly - result, err := engine.GetFilter(fltrMigrator.dmOut.DataManager(), filters.Tenant, filters.ID, false, false, utils.NonTransactional) + result, err := fltrMigrator.dmOut.DataManager().GetFilter(filters.Tenant, filters.ID, false, false, utils.NonTransactional) if err != nil { t.Fatalf("Error when getting filters %v", err.Error()) } @@ -488,7 +488,7 @@ func testFltrITMigratev3(t *testing.T) { t.Errorf("Unexpected version returned: %d", vrs[utils.RQF]) } //check if Filters was migrate correctly - result, err := engine.GetFilter(fltrMigrator.dmOut.DataManager(), filters.Tenant, filters.ID, false, false, utils.NonTransactional) + result, err := fltrMigrator.dmOut.DataManager().GetFilter(filters.Tenant, filters.ID, false, false, utils.NonTransactional) if err != nil { t.Fatalf("Error when getting filters %v", err.Error()) } diff --git a/migrator/stats_it_test.go b/migrator/stats_it_test.go index fc7c2dca5..ee109e974 100755 --- a/migrator/stats_it_test.go +++ b/migrator/stats_it_test.go @@ -264,7 +264,7 @@ func testStsITMigrateAndMove(t *testing.T) { t.Errorf("Expecting: %+v, received: %+v", sqp, result) } - result1, err := engine.GetFilter(stsMigrator.dmOut.DataManager(), "cgrates.org", v1Sts.Id, false, false, utils.NonTransactional) + result1, err := stsMigrator.dmOut.DataManager().GetFilter("cgrates.org", v1Sts.Id, false, false, utils.NonTransactional) if err != nil { t.Error("Error when getting Stats ", err.Error()) }