Add one more test

sonic-net · Nov 25, 2024 · 19d7415 · 19d7415
1 parent e3e9e79
commit 19d7415
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/sonic-py-common/sonic_py_common/device_info.py b/src/sonic-py-common/sonic_py_common/device_info.py
@@ -630,7 +630,7 @@ def is_dpu():
     # Retrieve platform.json data
     platform_data = get_platform_json_data()
     if platform_data:
-        return "DPU" in platform_data
+        return 'DPU' in platform_data
 
     return False
 

diff --git a/src/sonic-py-common/tests/device_info_test.py b/src/sonic-py-common/tests/device_info_test.py
@@ -249,6 +249,10 @@ def test_is_dpu(self, mock_get_platform, mock_is_smartswitch, mock_get_platform_
         mock_get_platform_json_data.return_value={"DPU": {}}
         assert device_info.is_dpu() == True
 
+        # Test case where platform is a smart switch and DPU data is present in platform.json
+        mock_get_platform_json_data.return_value={"DPUS": {}}
+        assert device_info.is_dpu() == False
+
     @mock.patch("sonic_py_common.device_info.get_platform_json_data")
     @mock.patch("sonic_py_common.device_info.get_platform")
     def test_get_dpu_info(self, mock_get_platform, mock_get_platform_json_data):