Fixed UT testcase.

Signed-off-by: rajkumar38 <rpennadamram@marvell.com>
rajkumar38 · Jul 25, 2023 · 8dbc3cc · 8dbc3cc
1 parent 8be5495
commit 8dbc3cc
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/db_migrator_test.py b/tests/db_migrator_test.py
@@ -719,7 +719,8 @@ def test_fast_reboot_upgrade_to_4_0_3(self):
         expected_db = self.mock_dedicated_config_db(db_after_migrate)
         advance_version_for_expected_database(dbmgtr.configDB, expected_db.cfgdb, 'version_4_0_3')
         assert not self.check_config_db(dbmgtr.configDB, expected_db.cfgdb)
-        assert dbmgtr.CURRENT_VERSION == expected_db.cfgdb.get_entry('VERSIONS', 'DATABASE')['VERSION']
+        if dbmgtr.CURRENT_VERSION == 'version_4_0_3':
+            assert dbmgtr.CURRENT_VERSION == expected_db.cfgdb.get_entry('VERSIONS', 'DATABASE')['VERSION']
 
 class TestSflowSampleDirectionMigrator(object):
     @classmethod