Fix pytest

Author: emmuhamm

src/drunc/process_manager/interface/commands.py

@@ -284,7 +284,9 @@ def logs_impl(
     obj: ProcessManagerContext, how_far: int, grep: str, query: ProcessQuery
 ) -> None:
     log = get_logger("process_manager.shell")
-    log.info(f"Running logs with query {query}")
+    # TODO: MOVE BACK TO DEBUG BEFORE MERGE
+    # THIS IS USEFUL FOR TESTING THOUGH
+    log.error(f"Running logs with query {query}")
     log_req = LogRequest(
         how_far=how_far,
         query=query,

tests/process_manager/interface/test_commands.py

@@ -102,6 +102,10 @@ def logs(self, log_request):
         mock_result.lines = []
         return mock_result
 
+    def send_msg(self, msg: str) -> None:
+        # simulate sending a message; tests don't assert on this, so store it
+        self._last_sent_msg = msg
+
 
 class MockContext:
     """
@@ -115,6 +119,9 @@ def __init__(self, driver=None):
     def get_driver(self, name):
         return self.driver
 
+    def get_shell_id(self):
+        return "mock-shell"
+
     def print(self, msg, justify=None, overflow=None, soft_wrap=None):
         self.output.append(str(msg))