Add test in migrator for tp dispatcher

migrator/migrator.go

@@ -178,6 +178,8 @@ func (m *Migrator) Migrate(taskIDs []string) (err error, stats map[string]int) {
 			err = m.migrateTPDestinations()
 		case utils.MetaTpChargers:
 			err = m.migrateTPChargers()
+		case utils.MetaTpDispatchers:
+			err = m.migrateTPDispatchers()
 			//DATADB ALL
 		case utils.MetaDataDB:
 			if err := m.migrateAccounts(); err != nil {

migrator/tp_dispatchers_it_test.go

@@ -0,0 +1,157 @@
+// +build integration
+
+/*
+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 <http://www.gnu.org/licenses/>
+*/
+
+package migrator
+
+import (
+	"log"
+	"path"
+	"reflect"
+	"testing"
+
+	"github.com/cgrates/cgrates/config"
+	"github.com/cgrates/cgrates/engine"
+	"github.com/cgrates/cgrates/utils"
+)
+
+var (
+	tpDispPathIn   string
+	tpDispPathOut  string
+	tpDispCfgIn    *config.CGRConfig
+	tpDispCfgOut   *config.CGRConfig
+	tpDispMigrator *Migrator
+	tpDisps        []*utils.TPDispatcherProfile
+)
+
+var sTestsTpDispIT = []func(t *testing.T){
+	testTpDispITConnect,
+	testTpDispITFlush,
+	testTpDispITPopulate,
+	testTpDispITMove,
+	testTpDispITCheckData,
+}
+
+func TestTpDispMove(t *testing.T) {
+	for _, stest := range sTestsTpDispIT {
+		t.Run("TestTpDispMove", stest)
+	}
+}
+
+func testTpDispITConnect(t *testing.T) {
+	var err error
+	tpDispPathIn = path.Join(*dataDir, "conf", "samples", "tutmongo")
+	tpDispCfgIn, err = config.NewCGRConfigFromFolder(tpDispPathIn)
+	if err != nil {
+		t.Fatal(err)
+	}
+	tpDispPathOut = path.Join(*dataDir, "conf", "samples", "tutmysql")
+	tpDispCfgOut, err = config.NewCGRConfigFromFolder(tpDispPathOut)
+	if err != nil {
+		t.Fatal(err)
+	}
+	storDBIn, err := NewMigratorStorDB(tpDispCfgIn.StorDbCfg().StorDBType,
+		tpDispCfgIn.StorDbCfg().StorDBHost, tpDispCfgIn.StorDbCfg().StorDBPort,
+		tpDispCfgIn.StorDbCfg().StorDBName, tpDispCfgIn.StorDbCfg().StorDBUser,
+		tpDispCfgIn.StorDbCfg().StorDBPass,
+		config.CgrConfig().StorDbCfg().StorDBMaxOpenConns,
+		config.CgrConfig().StorDbCfg().StorDBMaxIdleConns,
+		config.CgrConfig().StorDbCfg().StorDBConnMaxLifetime,
+		config.CgrConfig().StorDbCfg().StorDBCDRSIndexes)
+	if err != nil {
+		log.Fatal(err)
+	}
+	storDBOut, err := NewMigratorStorDB(tpDispCfgOut.StorDbCfg().StorDBType,
+		tpDispCfgOut.StorDbCfg().StorDBHost, tpDispCfgOut.StorDbCfg().StorDBPort,
+		tpDispCfgOut.StorDbCfg().StorDBName, tpDispCfgOut.StorDbCfg().StorDBUser,
+		tpDispCfgOut.StorDbCfg().StorDBPass,
+		config.CgrConfig().StorDbCfg().StorDBMaxOpenConns,
+		config.CgrConfig().StorDbCfg().StorDBMaxIdleConns,
+		config.CgrConfig().StorDbCfg().StorDBConnMaxLifetime,
+		config.CgrConfig().StorDbCfg().StorDBCDRSIndexes)
+	if err != nil {
+		log.Fatal(err)
+	}
+	tpDispMigrator, err = NewMigrator(nil, nil, storDBIn, storDBOut, false, false, false)
+	if err != nil {
+		log.Fatal(err)
+	}
+}
+
+func testTpDispITFlush(t *testing.T) {
+	if err := tpDispMigrator.storDBIn.StorDB().Flush(
+		path.Join(tpDispCfgIn.DataFolderPath, "storage",
+			tpDispCfgIn.StorDbCfg().StorDBType)); err != nil {
+		t.Error(err)
+	}
+
+	if err := tpDispMigrator.storDBOut.StorDB().Flush(
+		path.Join(tpDispCfgOut.DataFolderPath, "storage",
+			tpDispCfgOut.StorDbCfg().StorDBType)); err != nil {
+		t.Error(err)
+	}
+}
+
+func testTpDispITPopulate(t *testing.T) {
+	tpDisps = []*utils.TPDispatcherProfile{
+		&utils.TPDispatcherProfile{
+			TPid:      "TP1",
+			Tenant:    "cgrates.org",
+			ID:        "Dsp1",
+			FilterIDs: []string{"*string:Account:1002"},
+			ActivationInterval: &utils.TPActivationInterval{
+				ActivationTime: "2014-07-29T15:00:00Z",
+				ExpiryTime:     "",
+			},
+			Strategy: utils.MetaFirst,
+			Hosts:    []string{"localhost"},
+			Weight:   10,
+		},
+	}
+	if err := tpDispMigrator.storDBIn.StorDB().SetTPDispatchers(tpDisps); err != nil {
+		t.Error("Error when setting TpDispatchers ", err.Error())
+	}
+	currentVersion := engine.CurrentStorDBVersions()
+	err := tpDispMigrator.storDBOut.StorDB().SetVersions(currentVersion, false)
+	if err != nil {
+		t.Error("Error when setting version for TpDispatchers ", err.Error())
+	}
+}
+
+func testTpDispITMove(t *testing.T) {
+	err, _ := tpDispMigrator.Migrate([]string{utils.MetaTpDispatchers})
+	if err != nil {
+		t.Error("Error when migrating TpDispatchers ", err.Error())
+	}
+}
+
+func testTpDispITCheckData(t *testing.T) {
+	result, err := tpDispMigrator.storDBOut.StorDB().GetTPDispatchers("TP1", "cgrates.org", "Dsp1")
+	if err != nil {
+		t.Error("Error when getting TpDispatchers ", err.Error())
+	}
+	if !reflect.DeepEqual(tpDisps[0], result[0]) {
+		t.Errorf("Expecting: %+v, received: %+v",
+			utils.ToJSON(tpDisps[0]), utils.ToJSON(result[0]))
+	}
+	result, err = tpDispMigrator.storDBIn.StorDB().GetTPDispatchers("TP1", "cgrates.org", "Dsp1")
+	if err != utils.ErrNotFound {
+		t.Error(err)
+	}
+}

utils/consts.go

@@ -631,6 +631,7 @@ const (
 	MetaTpRatingPlan        = "*tp_rating_plan"
 	MetaTpRatingProfile     = "*tp_rating_profile"
 	MetaTpChargers          = "*tp_chargers"
+	MetaTpDispatchers       = "*tp_dispatchers"
 	MetaDurationSeconds     = "*duration_seconds"
 	MetaDurationNanoseconds = "*duration_nanoseconds"
 	CapAttributes           = "Attributes"