Update test_shutdownmanager.py

Add callback to the shutdown tests
canonical · Nov 22, 2023 · 07b8cde · 07b8cde
1 parent cbda93b
commit 07b8cde
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/landscape/client/manager/tests/test_shutdownmanager.py b/landscape/client/manager/tests/test_shutdownmanager.py
@@ -37,6 +37,7 @@ def check(_):
             bus_object.get_object.assert_called_once()
             bus_object.Reboot.assert_called_once()
 
+        deferred.addCallback(check)
         return deferred
 
     def test_shutdown(self):
@@ -52,4 +53,5 @@ def check(_):
             bus_object.get_object.assert_called_once()
             bus_object.PowerOff.assert_called_once()
 
+        deferred.addCallback(check)
         return deferred