Update tests/unittests/cmd/test_status.py

Co-authored-by: Chad Smith <chad.smith@canonical.com>
canonical · Feb 2, 2024 · c8dea59 · c8dea59
1 parent 8733800
commit c8dea59
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/tests/unittests/cmd/test_status.py b/tests/unittests/cmd/test_status.py
@@ -1049,11 +1049,9 @@ def test_query_systemctl_wait_with_exception_status(self, mocker):
         m_sleep = mocker.patch(f"{M_PATH}sleep")
         m_subp = mocker.patch(
             f"{M_PATH}subp.subp",
-            side_effect=[
-                subp.ProcessExecutionError("Message recipient disconnected"),
-                subp.ProcessExecutionError("Message recipient disconnected"),
-                subp.ProcessExecutionError("Message recipient disconnected"),
-            ],
+            side_effect=subp.ProcessExecutionError(
+                "Message recipient disconnected"
+            ),
         )
 
         assert (