Add unit test for the db_migrator

Author: EdenGri

tests/db_migrator_input/config_db/logger_tables_expected.json

@@ -0,0 +1,13 @@
+{
+    "VERSIONS|DATABASE": {
+        "VERSION": "version_3_0_6"
+    },
+    "LOGGER|orchagent": {
+        "LOGLEVEL": "INFO",
+        "LOGOUTPUT": "SYSLOG"
+    },
+    "LOGGER|SAI_API_BUFFER": {
+        "LOGLEVEL": "SAI_LOG_LEVEL_NOTICE",
+        "LOGOUTPUT": "SYSLOG"
+    }
+}

tests/db_migrator_input/config_db/logger_tables_input.json

@@ -0,0 +1,5 @@
+{
+    "VERSIONS|DATABASE": {
+        "VERSION": "version_3_0_5"
+    }
+}

tests/db_migrator_input/loglevel_db/logger_tables_expected.json

@@ -0,0 +1 @@
+{}

tests/db_migrator_input/loglevel_db/logger_tables_input.json

@@ -0,0 +1,11 @@
+{
+    "orchagent:orchagent": {
+        "LOGLEVEL": "INFO",
+        "LOGOUTPUT": "SYSLOG"
+    },
+    "SAI_API_BUFFER:SAI_API_BUFFER": {
+        "LOGLEVEL": "SAI_LOG_LEVEL_NOTICE",
+        "LOGOUTPUT": "SYSLOG"
+    },
+    "JINJA2_CACHE": {}
+}

tests/db_migrator_test.py

@@ -409,4 +409,42 @@ def test_global_dscp_to_tc_map_migrator(self):
         dbmgtr_mlnx.migrate()
         resulting_table = dbmgtr_mlnx.configDB.get_table('PORT_QOS_MAP')
         assert resulting_table == {}
+        
+class TestMoveLoggerTablesInWarmUpgrade(object):
+    @classmethod
+    def setup_class(cls):
+        os.environ['UTILITIES_UNIT_TESTING'] = "2"
+
+    @classmethod
+    def teardown_class(cls):
+        os.environ['UTILITIES_UNIT_TESTING'] = "0"
+        dbconnector.dedicated_dbs['CONFIG_DB'] = None
+        dbconnector.dedicated_dbs['LOGLEVEL_DB'] = None
+    def mock_dedicated_loglevel_db(self, filename):
+        jsonfile = os.path.join(mock_db_path, 'loglevel_db', filename)
+        dbconnector.dedicated_dbs['LOGLEVEL_DB'] = jsonfile
+        loglevel_db = SonicV2Connector(host='127.0.0.1')
+        loglevel_db.connect(loglevel_db.LOGLEVEL_DB)
+        return loglevel_db
+
+    def test_move_logger_tables_in_warm_upgrade(self):
+        dbconnector.dedicated_dbs['CONFIG_DB'] = os.path.join(mock_db_path, 'config_db', 'logger_tables_input')
+        dbconnector.dedicated_dbs['LOGLEVEL_DB'] = os.path.join(mock_db_path, 'loglevel_db', 'logger_tables_input')
+        import db_migrator
+        dbmgtr = db_migrator.DBMigrator(None)
+        dbmgtr.migrate()
+        dbconnector.dedicated_dbs['CONFIG_DB'] = os.path.join(mock_db_path, 'config_db', 'logger_tables_expected')
+        dbconnector.dedicated_dbs['LOGLEVEL_DB'] = os.path.join(mock_db_path, 'loglevel_db', 'logger_tables_expected')
+
+        expected_db = Db()
+
+        resulting_table = dbmgtr.configDB.get_table('LOGGER')
+        expected_table = expected_db.cfgdb.get_table('LOGGER')
+
+        diff = DeepDiff(resulting_table, expected_table, ignore_order=True)
+        assert not diff
+
+
+
+