Iproving coverage

sonic-net · Dec 13, 2024 · 3e2114f · 3e2114f
1 parent ddf4541
commit 3e2114f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/process-reboot-cause_test.py b/tests/process-reboot-cause_test.py
@@ -97,8 +97,8 @@ def test_get_sorted_reboot_cause_files_error(self, mock_listdir):
         self.assertEqual(result, [])
 
     # Test update_dpu_reboot_cause_to_chassis_state_db
-    @patch("builtins.open", new_callable=mock_open, read_data='{"cause": "Non-Hardware", "comment": " Switch rebooted DPU\\n", "device": "DPU0", "time": "Fri Dec 13 01:12:36 AM UTC 2024", "name": "2024_12_13_01_12_36"}')
-    @patch("process_reboot_cause.SonicV2Connector")
+    @patch("builtins.open", new_callable=mock_open, read_data='{"cause": "Non-Hardware", "comment": "Switch rebooted DPU", "device": "DPU0", "time": "Fri Dec 13 01:12:36 AM UTC 2024", "name": "2024_12_13_01_12_36"}')
+    @patch("process_reboot_cause.swsscommon.SonicV2Connector")
     def test_update_dpu_reboot_cause_to_chassis_state_db_update(self, mock_connector, mock_open):
         # Mock the database connection
         mock_db = MagicMock()