Fix broken tests due to field name change and status not being correct

Author: kirkbrauer

packages/jumpstarter-testing/jumpstarter_testing/pytest_test.py

@@ -1,6 +1,7 @@
 from jumpstarter_driver_power.driver import MockPower
 from pytest import Pytester
 
+from jumpstarter.common import ExporterStatus
 from jumpstarter.config.env import JMP_DRIVERS_ALLOW, JUMPSTARTER_HOST
 from jumpstarter.exporter import Session
 
@@ -18,6 +19,8 @@ def test_simple(self, client):
 
     with Session(root_device=MockPower()) as session:
         with session.serve_unix() as path:
+            # For local testing, set status to LEASE_READY since there's no lease/hook flow
+            session.update_status(ExporterStatus.LEASE_READY)
             monkeypatch.setenv(JUMPSTARTER_HOST, str(path))
             monkeypatch.setenv(JMP_DRIVERS_ALLOW, "UNSAFE")
             result = pytester.runpytest()

packages/jumpstarter/jumpstarter/client/core.py

@@ -89,7 +89,7 @@ async def check_exporter_status(self):
             status = ExporterStatus.from_proto(response.status)
 
             if status != ExporterStatus.LEASE_READY:
-                raise ExporterNotReady(f"Exporter status is {status}: {response.status_message}")
+                raise ExporterNotReady(f"Exporter status is {status}: {response.message}")
 
         except AioRpcError as e:
             # If GetStatus is not implemented, assume ready for backward compatibility

packages/jumpstarter/jumpstarter/common/utils.py

@@ -22,8 +22,12 @@
 
 @asynccontextmanager
 async def serve_async(root_device: "Driver", portal: BlockingPortal, stack: ExitStack):
+    from jumpstarter.common import ExporterStatus
+
     with Session(root_device=root_device) as session:
         async with session.serve_unix_async() as path:
+            # For local testing, set status to LEASE_READY since there's no lease/hook flow
+            session.update_status(ExporterStatus.LEASE_READY)
             # SAFETY: the root_device instance is constructed locally thus considered trusted
             async with client_from_path(path, portal, stack, allow=[], unsafe=True) as client:
                 try:

packages/jumpstarter/jumpstarter/config/exporter.py

@@ -183,6 +183,7 @@ def delete(cls, alias: str) -> Path:
     @asynccontextmanager
     async def serve_unix_async(self):
         # dynamic import to avoid circular imports
+        from jumpstarter.common import ExporterStatus
         from jumpstarter.exporter import Session
 
         with Session(
@@ -193,6 +194,8 @@ async def serve_unix_async(self):
             ).instantiate(),
         ) as session:
             async with session.serve_unix_async() as path:
+                # For local usage, set status to LEASE_READY since there's no lease/hook flow
+                session.update_status(ExporterStatus.LEASE_READY)
                 yield path
 
     @contextmanager

packages/jumpstarter/jumpstarter/exporter/session.py

@@ -174,5 +174,5 @@ async def GetStatus(self, request, context):
         logger.debug("GetStatus() -> %s", self._current_status)
         return jumpstarter_pb2.GetStatusResponse(
             status=self._current_status.to_proto(),
-            status_message=self._status_message,
+            message=self._status_message,
         )