updated db_migrator tests

MuhammadUmarAsad · MuhammadUmarAsad · commit 94111966d0ce · 2023-02-25T03:51:01.000+05:00
diff --git a/scripts/db_migrator.py b/scripts/db_migrator.py
@@ -469,6 +469,9 @@ def migrate_config_db_switchport_mode(self):
                 if pc_key in vlan_member_keys:
                     pc_value["mode"] = "trunk"
                     self.configDB.set_entry("PORTCHANNEL", pc_key, pc_value)
+                else:
+                    pc_value["mode"] = "routed"
+                    self.configDB.set_entry("PORTCHANNEL", pc_key, pc_value)
 
 
     def migrate_qos_db_fieldval_reference_remove(self, table_list, db, db_num, db_delimeter):
diff --git a/tests/db_migrator_input/config_db/switchport-expected.json b/tests/db_migrator_input/config_db/switchport-expected.json
@@ -53,7 +53,7 @@
         "mtu": "9100",
         "speed": "40000"
     },
-    
+
     "VLAN|Vlan2": {
         "vlanid": "2"
     },
@@ -132,6 +132,6 @@
     "PORTCHANNEL_MEMBER|PortChannel0002|Ethernet28": {},
 
     "VERSIONS|DATABASE": {
-        "VERSION": "version_4_0_0"
+        "VERSION": "version_4_0_1"
     }
 }
diff --git a/tests/db_migrator_test.py b/tests/db_migrator_test.py
@@ -266,9 +266,9 @@ def test_switchport_mode_migrator(self):
         expected_db = Db()
         advance_version_for_expected_database(dbmgtr.configDB, expected_db.cfgdb, 'version_4_0_1')
 
-        # assert dbmgtr.configDB.get_table('PORT') == expected_db.cfgdb.get_table('PORT')
-        # assert dbmgtr.configDB.get_table('PORTCHANNEL') == expected_db.cfgdb.get_table('PORTCHANNEL')
-        # assert dbmgtr.configDB.get_table('VERSIONS') == expected_db.cfgdb.get_table('VERSIONS')
+        assert dbmgtr.configDB.get_table('PORT') == expected_db.cfgdb.get_table('PORT')
+        assert dbmgtr.configDB.get_table('PORTCHANNEL') == expected_db.cfgdb.get_table('PORTCHANNEL')
+        assert dbmgtr.configDB.get_table('VERSIONS') == expected_db.cfgdb.get_table('VERSIONS')
 
 class TestInitConfigMigrator(object):
     @classmethod
