typing

Author: olethanh

src/aleph/vm/models.py

@@ -361,6 +361,7 @@ async def start(self):
 
     async def wait_for_persistent_boot(self):
         """Determine if VM has booted by responding to ping and check if the process is still running"""
+        assert self.vm
         assert self.vm.enable_networking and self.vm.tap_interface, f"Network not enabled for VM {self.vm.vm_id}"
         ip = self.vm.get_ip()
         if not ip:

tests/supervisor/test_instance.py

@@ -23,7 +23,7 @@ class MockSystemDManager(SystemDManager):
     execution: MicroVM | None = None
     process: Process | None = None
 
-    async def enable_and_start(self, service: str) -> tuple[MicroVM, Process]:
+    async def enable_and_start(self, service: str) -> tuple[MicroVM | None, Process | None]:
         vm_hash = service.split("@", maxsplit=1)[1].split(".", maxsplit=1)[0]
 
         config_path = Path(f"{settings.EXECUTION_ROOT}/{vm_hash}-controller.json")

tests/supervisor/test_qemu_instance.py

@@ -23,7 +23,7 @@ class MockSystemDManager(SystemDManager):
     execution: QemuVM | None = None
     process: Process | None = None
 
-    async def enable_and_start(self, service: str) -> tuple[QemuVM, Process]:
+    async def enable_and_start(self, service: str) -> tuple[QemuVM | None, Process | None]:
         # aleph-vm-controller@decadecadecadecadecadecadecadecadecadecadecadecadecadecadecadeca.service-controller.json
         vm_hash = service.split("@", maxsplit=1)[1].split(".", maxsplit=1)[0]