Add RTX 5080 FLUX hot-swap worker

dev/test_profile_controller.py

@@ -21,6 +21,11 @@ def item(profile, state="exited"):
     }
 
 
+def image_item(state="exited"):
+    return {"Id": "id-flux", "State": state,
+            "Labels": {controller.IMAGE_LABEL_KEY: controller.IMAGE_WORKER}}
+
+
 class ProfileControllerTests(unittest.TestCase):
     def test_rejects_unknown_profile_before_docker_call(self):
         with patch.object(controller, "docker_request") as request:
@@ -38,6 +43,7 @@ class ProfileControllerTests(unittest.TestCase):
             return 204, b""
 
         with patch.object(controller, "containers", return_value=profiles), \
+             patch.object(controller, "image_container", return_value=image_item()), \
              patch.object(controller, "docker_request", side_effect=request):
             result = controller.activate("medium")
 
@@ -49,10 +55,47 @@ class ProfileControllerTests(unittest.TestCase):
 
     def test_fails_if_profile_container_is_missing(self):
         profiles = {name: item(name) for name in controller.ALLOWED[:-1]}
-        with patch.object(controller, "containers", return_value=profiles):
+        with patch.object(controller, "containers", return_value=profiles), \
+             patch.object(controller, "image_container", return_value=image_item()):
             with self.assertRaisesRegex(RuntimeError, "missing"):
                 controller.activate("fast")
 
+    def test_image_start_stops_inference_first(self):
+        profiles = {name: item(name) for name in controller.ALLOWED}
+        profiles["medium"] = item("medium", "running")
+        calls = []
+
+        def request(method, path):
+            calls.append((method, path))
+            return 204, b""
+
+        with patch.object(controller, "containers", return_value=profiles), \
+             patch.object(controller, "image_container", return_value=image_item()), \
+             patch.object(controller, "docker_request", side_effect=request):
+            controller.set_image_worker(True)
+        self.assertEqual(calls, [
+            ("POST", "/containers/id-medium/stop?t=120"),
+            ("POST", "/containers/id-flux/start"),
+        ])
+
+    def test_profile_activation_stops_image_worker_first(self):
+        profiles = {name: item(name) for name in controller.ALLOWED}
+        calls = []
+
+        def request(method, path):
+            calls.append((method, path))
+            return 204, b""
+
+        with patch.object(controller, "containers", return_value=profiles), \
+             patch.object(controller, "image_container",
+                          return_value=image_item("running")), \
+             patch.object(controller, "docker_request", side_effect=request):
+            controller.activate("fast")
+        self.assertEqual(calls, [
+            ("POST", "/containers/id-flux/stop?t=120"),
+            ("POST", "/containers/id-fast/start"),
+        ])
+
 
 if __name__ == "__main__":
     unittest.main()