[UT] Add UT for chassis qos reload

config/main.py

@@ -3001,7 +3001,8 @@ def _qos_update_ports(ctx, ports, dry_run, json_data):
                     if element.startswith('Eth'):
                         port = element
                         break
-                assert port is not None, "Port is not found in config DB"
+                if port is None:
+                    ctx.fail("Port is not found in config DB")
                 if not port in portset_to_handle:
                     continue
                 config_db.set_entry(table_name, '|'.join(key), None)

tests/config_test.py

@@ -911,12 +911,30 @@ def test_qos_update_single(
         )
         assert filecmp.cmp(output_file, expected_result, shallow=False)
 
+    def test_qos_update_multi_dut(
+            self, get_cmd_module, setup_qos_mock_apis
+        ):
+        (config, show) = get_cmd_module
+        json_data = '{"DEVICE_METADATA": {"localhost": {}}, "PORT": {"Ethernet0": {}}}'
+        runner = CliRunner()
+        output_file = os.path.join(os.sep, "tmp", "qos_config_update_multi_dut.json")
+        cmd_vector = ["reload", "--ports", "Ethernet0"]
+        result = runner.invoke(config.config.commands["qos"], cmd_vector)
+        print(result.exit_code)
+        assert result.output == "ok"
+        print(result.output)
+        assert result.exit_code == 0
+        cwd = os.path.dirname(os.path.realpath(__file__))
+        expected_result = os.path.join(
+            cwd, "qos_config_input", "update_qos_multi_dut.json"
+        )
+        assert filecmp.cmp(output_file, expected_result, shallow=False)
+
     @classmethod
     def teardown_class(cls):
         print("TEARDOWN")
         os.environ['UTILITIES_UNIT_TESTING'] = "0"
 
-
 class TestConfigQosMasic(object):
     @classmethod
     def setup_class(cls):
@@ -1009,6 +1027,7 @@ def teardown_class(cls):
         importlib.reload(mock_single_asic)
         dbconnector.load_namespace_config()
 
+
 class TestGenericUpdateCommands(unittest.TestCase):
     def setUp(self):
         os.environ['UTILITIES_UNIT_TESTING'] = "1"

tests/qos_config_input/update_qos_multi_dut.json

@@ -0,0 +1,60 @@
+{
+    "BUFFER_PG": {
+        "Ethernet0|0": {
+            "profile": "ingress_lossy_profile"
+        }
+        "lc1-1|asic0|Ethernet88|0-2": {
+            "profile": "ingress_lossy_profile"
+        }
+    },
+    "BUFFER_QUEUE": {
+        "Ethernet0|0-2": {
+            "profile": "egress_lossy_profile"
+        },
+        "Ethernet0|3-4": {
+            "profile": "egress_lossless_profile"
+        },
+        "Ethernet0|5-6": {
+            "profile": "egress_lossy_profile"
+        }
+    },
+    "CABLE_LENGTH": {
+        "AZURE": {
+            "Ethernet0": "300m"
+        }
+    },
+    "PORT_QOS_MAP": {
+        "Ethernet0": {
+            "dscp_to_tc_map": "AZURE",
+            "pfc_enable": "3,4",
+            "pfc_to_queue_map": "AZURE",
+            "tc_to_pg_map": "AZURE",
+            "tc_to_queue_map": "AZURE"
+        }
+    },
+    "QUEUE": {
+        "Ethernet0|0": {
+            "scheduler": "scheduler.0"
+        },
+        "Ethernet0|1": {
+            "scheduler": "scheduler.0"
+        },
+        "Ethernet0|2": {
+            "scheduler": "scheduler.0"
+        },
+        "Ethernet0|3": {
+            "scheduler": "scheduler.1",
+            "wred_profile": "AZURE_LOSSLESS"
+        },
+        "Ethernet0|4": {
+            "scheduler": "scheduler.1",
+            "wred_profile": "AZURE_LOSSLESS"
+        },
+        "Ethernet0|5": {
+            "scheduler": "scheduler.0"
+        },
+        "Ethernet0|6": {
+            "scheduler": "scheduler.0"
+        }
+    }
+}