Working on coverage

sonic-net · Nov 19, 2024 · d982499 · d982499
1 parent 7c3ed24
commit d982499
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sonic-chassisd/tests/test_chassisd.py b/sonic-chassisd/tests/test_chassisd.py
@@ -133,8 +133,10 @@ def test_smartswitch_moduleupdater_status_transitions():
     # Create the updater
     module_updater = SmartSwitchModuleUpdater(SYSLOG_IDENTIFIER, chassis)
 
-    # Mock methods that interact with the filesystem
-    with patch.object(module_updater, 'persist_dpu_reboot_time') as mock_persist_reboot_time, \
+    # Mock dependent methods
+    with patch.object(module_updater, 'retrieve_dpu_reboot_time', return_value="2024-11-19T00:00:00") \
+             as mock_retrieve_reboot_time, \
+         patch.object(module_updater, '_is_first_boot', return_value=False) as mock_is_first_boot, \
          patch.object(module_updater, 'persist_dpu_reboot_cause') as mock_persist_reboot_cause, \
          patch.object(module_updater, 'update_dpu_reboot_cause_to_db') as mock_update_reboot_db: