[multi-asic] Refine [override config table] for corner cases

config/main.py

@@ -1894,14 +1894,24 @@ def override_config_table(db, input_config_db, dry_run):
         current_config = config_db.get_config()
         # Serialize to the same format as json input
         sonic_cfggen.FormatConverter.to_serialized(current_config)
-
-        if multi_asic.is_multi_asic():
+        ns_config_input = None
+        if multi_asic.is_multi_asic() and len(config_input):
             # Golden Config will use "localhost" to represent host name
             if ns == DEFAULT_NAMESPACE:
-                ns_config_input = config_input["localhost"]
+                if "localhost" in config_input.keys():
+                    ns_config_input = config_input["localhost"]
+                else:
+                    click.secho("Wrong config format! 'localhost' not found in host config! cannot override.. abort")
+                    sys.exit(1)
             else:
-                ns_config_input = config_input[ns]
-        else:
+                if ns in config_input.keys():
+                    ns_config_input = config_input[ns]
+                else:
+                    click.secho("Wrong config format! {} not found in asic config! cannot override.. abort".format(ns))
+                    sys.exit(1)
+        if not ns_config_input:
+            # if ns_config_input is not defined, define it
+            # it could be single-asic dut, or config_input is empty
             ns_config_input = config_input
         # Generate sysinfo if missing in ns_config_input
         generate_sysinfo(current_config, ns_config_input, ns)

tests/config_override_input/multi_asic_wrong_format.json

@@ -0,0 +1,11 @@
+{
+    "host": {
+        "DEVICE_METADATA": {}
+    },
+    "asic2": {
+        "DEVICE_METADATA": {}
+    },
+    "asic1": {
+        "DEVICE_METADATA": {}
+    }
+}

tests/config_override_test.py

@@ -381,6 +381,27 @@ def read_json_file_side_effect(filename):
             assert platform == "multi_asic"
             assert mac == "11:22:33:44:55:66"
 
+    def test_masic_wrong_format_override(self):
+        def read_json_file_side_effect(filename):
+            with open(MULTI_ASIC_WRONG_FORMAT, "r") as f:
+                wrong_format = json.load(f)
+            return wrong_format
+        db = Db()
+        cfgdb_clients = db.cfgdb_clients
+
+        with mock.patch('config.main.read_json_file',
+                        mock.MagicMock(side_effect=read_json_file_side_effect)):
+            runner = CliRunner()
+            result = runner.invoke(config.config.commands["override-config-table"],
+                                   ['golden_config_db.json'], obj=db)
+            # make sure program aborted with return code 1
+            assert result.exit_code == 1
+
+        for ns, config_db in cfgdb_clients.items():
+            if ns:
+                assert "not found in asic config" in result.output
+            else:
+                assert "'localhost' not found in host config" in result.output
 
     @classmethod
     def teardown_class(cls):