diff --git a/migrator/storage_map_datadb_test.go b/migrator/storage_map_datadb_test.go new file mode 100644 index 000000000..851aca0b1 --- /dev/null +++ b/migrator/storage_map_datadb_test.go @@ -0,0 +1,220 @@ +/* +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 +*/ + +package migrator + +import ( + "testing" + + "github.com/cgrates/cgrates/engine" + "github.com/cgrates/cgrates/utils" +) + +func TestInternalMigratorSetV1Account(t *testing.T) { + iDBMig := &internalMigrator{} + account := &v1Account{ + Id: "id", + BalanceMap: map[string]v1BalanceChain{ + "chain1": {}, + }, + UnitCounters: []*v1UnitsCounter{ + {}, + }, + ActionTriggers: v1ActionTriggers{}, + AllowNegative: true, + Disabled: false, + } + err := iDBMig.setV1Account(account) + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorSetAndRemV2Account(t *testing.T) { + iDBMig := &internalMigrator{} + Account := &v2Account{ + ID: "id", + BalanceMap: make(map[string]engine.Balances), + UnitCounters: engine.UnitCounters{}, + ActionTriggers: engine.ActionTriggers{}, + AllowNegative: true, + Disabled: false, + } + ID := "id" + err := iDBMig.setV2Account(Account) + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error for setV2Account to be ErrNotImplemented, but got %v", err) + } + err = iDBMig.remV2Account(ID) + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error for remV2Account to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorMethods(t *testing.T) { + iDBMig := &internalMigrator{} + v1aps, err := iDBMig.getV1ActionPlans() + if v1aps != nil { + t.Fatalf("Expected v1ActionPlans to be nil, but got %v", v1aps) + } + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error for getV1ActionPlans to be ErrNotImplemented, but got %v", err) + } + v1acs, err := iDBMig.getV1Actions() + if v1acs != nil { + t.Fatalf("Expected v1Actions to be nil, but got %v", v1acs) + } + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error for getV1Actions to be ErrNotImplemented, but got %v", err) + } + +} + +func TestInternalMigratorGetV1RouteProfile(t *testing.T) { + iDBMig := &internalMigrator{} + v1chrPrf, err := iDBMig.getV1RouteProfile() + if v1chrPrf != nil { + t.Fatalf("Expected v1chrPrf to be nil, but got %v", v1chrPrf) + } + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorDataManager(t *testing.T) { + dataManager := &engine.DataManager{} + iDBMig := &internalMigrator{ + dm: dataManager, + } + returnedDM := iDBMig.DataManager() + if returnedDM != dataManager { + t.Fatalf("Expected DataManager to be %v, but got %v", dataManager, returnedDM) + } +} + +func TestInternalMigratorGetv1Account(t *testing.T) { + iDBMig := &internalMigrator{} + v1Acnt, err := iDBMig.getv1Account() + if v1Acnt != nil { + t.Fatalf("Expected v1Acnt to be nil, but got %v", v1Acnt) + } + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorRemV1Account(t *testing.T) { + iDBMig := &internalMigrator{} + testID := "id" + err := iDBMig.remV1Account(testID) + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorGetv2Account(t *testing.T) { + iDBMig := &internalMigrator{} + v2Acnt, err := iDBMig.getv2Account() + if v2Acnt != nil { + t.Fatalf("Expected v2Account to be nil, but got %v", v2Acnt) + } + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorProfileMethods(t *testing.T) { + iDBMig := &internalMigrator{} + v1chrPrf, err := iDBMig.getV1ChargerProfile() + if v1chrPrf != nil { + t.Fatalf("Expected ChargerProfile to be nil, but got %v", v1chrPrf) + } + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error for getV1ChargerProfile to be ErrNotImplemented, but got %v", err) + } + +} + +func TestInternalMigratorGetV1DispatcherProfile(t *testing.T) { + iDBMig := &internalMigrator{} + v1chrPrf, err := iDBMig.getV1DispatcherProfile() + if v1chrPrf != nil { + t.Fatalf("Expected DispatcherProfile to be nil, but got %v", v1chrPrf) + } + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorRemSupplier(t *testing.T) { + iDBMig := &internalMigrator{} + tenant := "cgrates.org" + id := "ID" + err := iDBMig.remSupplier(tenant, id) + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorSetSupplier(t *testing.T) { + iDBMig := &internalMigrator{} + supplierProfile := &SupplierProfile{ + Tenant: "cgrates.org", + ID: "1", + FilterIDs: []string{"id1", "id2", "id3"}, + Sorting: "sorting", + Weight: 10, + } + err := iDBMig.setSupplier(supplierProfile) + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorGetSupplier(t *testing.T) { + iDBMig := &internalMigrator{} + supplier, err := iDBMig.getSupplier() + if supplier != nil { + t.Fatalf("Expected supplier to be nil, but got %v", supplier) + } + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorRemV1Filter(t *testing.T) { + iDBMig := &internalMigrator{} + tenant := "cgrates.org" + id := "ID" + err := iDBMig.remV1Filter(tenant, id) + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +} + +func TestInternalMigratorSetV1Filter(t *testing.T) { + iDBMig := &internalMigrator{} + sampleFilter := &v1Filter{ + Tenant: "cgrates.org", + ID: "ID", + } + err := iDBMig.setV1Filter(sampleFilter) + if err != utils.ErrNotImplemented { + t.Fatalf("Expected error to be ErrNotImplemented, but got %v", err) + } +}