feat: design model support for qwen-tts backend. remove unnecessary port split for qwen models

1 files changed, 39 insertions, 0 deletions

diff --git a/app/tests/test_runview.py b/app/tests/test_runview.py
index 31a8ebe..f2d5180 100644
--- a/app/tests/test_runview.py
+++ b/app/tests/test_runview.py
@@ -363,6 +363,45 @@ class WorkerTests(_FakeTui, unittest.TestCase):
         self.assertIn("ERROR - boom", text)
         self.assertIn("error", [e["kind"] for e in self._drain(view)])
 
+    def test_restart_first_stops_then_boots_the_new_model(self):
+        # The managed qwen server hosts another model than this run picked:
+        # the worker stops it (releasing the single port) before booting
+        # the spec again — whose argv now names the newly-selected model.
+        from types import SimpleNamespace
+        spec = SimpleNamespace(name="qwen")
+        events = []
+
+        def fake_start(spec_, progress=None, cancel=None):
+            events.append(("start", progress))
+            progress({"kind": "ready", "name": "qwen",
+                      "url": "http://127.0.0.1:7860"})
+            return True
+
+        with patch("audiobook.convert") as mk_convert, \
+                patch.object(runview.servers, "stop") as mk_stop, \
+                patch.object(runview.servers, "start",
+                             side_effect=fake_start):
+            view = self.make_view(autostart_spec=spec, restart_first=True,
+                                  server_name="qwen")
+            view._worker_main()
+        mk_stop.assert_called_once_with("qwen")
+        events = self._drain(view)
+        kinds = [e["kind"] for e in events]
+        self.assertIn("ready", kinds)
+        self.assertNotIn("error", kinds)
+        mk_convert.assert_called_once()
+
+    def test_no_restart_without_the_flag(self):
+        # A plain autostart never stops a server first.
+        from types import SimpleNamespace
+        spec = SimpleNamespace(name="qwen")
+        with patch("audiobook.convert"), \
+                patch.object(runview.servers, "stop") as mk_stop, \
+                patch.object(runview.servers, "start", return_value=True):
+            view = self.make_view(autostart_spec=spec, server_name="qwen")
+            view._worker_main()
+        mk_stop.assert_not_called()
+
 
 if __name__ == "__main__":
     unittest.main()