feat: generate a book with all installed models to compare

1 files changed, 75 insertions, 5 deletions

diff --git a/app/tests/test_tts.py b/app/tests/test_tts.py
index 67c81f8..3c039ab 100644
--- a/app/tests/test_tts.py
+++ b/app/tests/test_tts.py
@@ -1957,6 +1957,7 @@ class AudioCppUnloadModelsTests(unittest.TestCase):
         client.instruction_voice = False
         client.speaker_mode = False
         client.instructions = ""
+        client._unload_models_override = None
         with patch.object(client, "_check_health"), \
                 patch.object(client, "_list_models",
                              return_value=[{"id": client.model_id,
@@ -1971,6 +1972,70 @@ class AudioCppUnloadModelsTests(unittest.TestCase):
             client._connect()
         mock_unload.assert_called_once()
 
+    def test_connect_unload_override_forces_unload(self):
+        # "All (multiple generation)" runs force the unload regardless of
+        # the AUDIOCPP_UNLOAD_MODELS setting, so each model starts clean.
+        client = AudioCppTTSClient.__new__(AudioCppTTSClient)
+        client.api_url = "http://127.0.0.1:8080"
+        client.model_id = _AUDIOCPP_MODEL_ID
+        client.preset_mode = True
+        client.voice = "narrator"
+        client.language = "English"
+        client._seed = -1
+        client.family = "qwen3_tts"
+        client.task = AUDIOCPP_TASK_TTS
+        client.profile = AUDIOCPP_FAMILY_PROFILES["qwen3_tts"]
+        client.design_mode = False
+        client.instruction_voice = False
+        client.speaker_mode = False
+        client.instructions = ""
+        client._unload_models_override = True
+        with patch.object(client, "_check_health"), \
+                patch.object(client, "_list_models",
+                             return_value=[{"id": client.model_id,
+                                            "family": "qwen3_tts",
+                                            "task": "tts"}]), \
+                patch.object(client, "_auto_pick_model_id"), \
+                patch.object(client, "_require_model_id"), \
+                patch.object(client, "_resolve_family"), \
+                patch.object(client, "_resolve_task"), \
+                patch.object(client, "_check_voice"), \
+                patch.object(config, "AUDIOCPP_UNLOAD_MODELS", False), \
+                patch.object(client, "_unload_server_models") as mock_unload:
+            client._connect()
+        mock_unload.assert_called_once()
+
+    def test_connect_unload_override_false_skips_unload(self):
+        client = AudioCppTTSClient.__new__(AudioCppTTSClient)
+        client.api_url = "http://127.0.0.1:8080"
+        client.model_id = _AUDIOCPP_MODEL_ID
+        client.preset_mode = True
+        client.voice = "narrator"
+        client.language = "English"
+        client._seed = -1
+        client.family = "qwen3_tts"
+        client.task = AUDIOCPP_TASK_TTS
+        client.profile = AUDIOCPP_FAMILY_PROFILES["qwen3_tts"]
+        client.design_mode = False
+        client.instruction_voice = False
+        client.speaker_mode = False
+        client.instructions = ""
+        client._unload_models_override = False
+        with patch.object(client, "_check_health"), \
+                patch.object(client, "_list_models",
+                             return_value=[{"id": client.model_id,
+                                            "family": "qwen3_tts",
+                                            "task": "tts"}]), \
+                patch.object(client, "_auto_pick_model_id"), \
+                patch.object(client, "_require_model_id"), \
+                patch.object(client, "_resolve_family"), \
+                patch.object(client, "_resolve_task"), \
+                patch.object(client, "_check_voice"), \
+                patch.object(config, "AUDIOCPP_UNLOAD_MODELS", True), \
+                patch.object(client, "_unload_server_models") as mock_unload:
+            client._connect()
+        mock_unload.assert_not_called()
+
     def test_connect_skips_unload_when_disabled(self):
         client = AudioCppTTSClient.__new__(AudioCppTTSClient)
         client.api_url = "http://127.0.0.1:8080"
@@ -1986,6 +2051,7 @@ class AudioCppUnloadModelsTests(unittest.TestCase):
         client.instruction_voice = False
         client.speaker_mode = False
         client.instructions = ""
+        client._unload_models_override = None
         with patch.object(client, "_check_health"), \
                 patch.object(client, "_list_models",
                              return_value=[{"id": client.model_id,
@@ -2029,7 +2095,8 @@ class BackendWiringTests(unittest.TestCase):
                                               model_id=None,
                                               instructions=None,
                                               request_options={},
-                                              api_url=None, quiet=False)
+                                              api_url=None, quiet=False,
+                                              unload_models=None)
         mock_faster.assert_not_called()
         mock_qwen.assert_not_called()
 
@@ -2042,7 +2109,8 @@ class BackendWiringTests(unittest.TestCase):
                                               model_id=None,
                                               instructions=None,
                                               request_options={},
-                                              api_url=None, quiet=False)
+                                              api_url=None, quiet=False,
+                                              unload_models=None)
 
     def test_audiocpp_backend_model_id_is_wired_through(self):
         with patch("converter.converter.AudioCppTTSClient") as mock_audiocpp:
@@ -2053,7 +2121,8 @@ class BackendWiringTests(unittest.TestCase):
             chunks_dir=converter_mod.CHUNKS_FOLDER,
             voice="narrator", language=config.LANGUAGE,
             model_id="higgs", instructions=None,
-            request_options={}, api_url=None, quiet=False)
+            request_options={}, api_url=None, quiet=False,
+            unload_models=None)
 
     def test_audiocpp_backend_instructions_and_options_are_wired_through(self):
         with patch("converter.converter.AudioCppTTSClient") as mock_audiocpp:
@@ -2068,7 +2137,7 @@ class BackendWiringTests(unittest.TestCase):
             model_id=None,
             instructions="A warm adult narrator",
             request_options={"emotion": "neutral", "speed": "1.1"},
-            api_url=None, quiet=False)
+            api_url=None, quiet=False, unload_models=None)
 
     def test_qwen_backend_uses_qwen_client(self):
         with patch("converter.converter.FasterTTSClient") as mock_faster, \
@@ -2119,7 +2188,8 @@ class BackendWiringTests(unittest.TestCase):
             chunks_dir=converter_mod.CHUNKS_FOLDER,
             voice="narrator", language=config.LANGUAGE, model_id=None,
             instructions=None, request_options={},
-            api_url="http://10.0.0.5:8080", quiet=False)
+            api_url="http://10.0.0.5:8080", quiet=False,
+            unload_models=None)
         with patch("converter.converter.FasterTTSClient") as mock_faster:
             AudiobookConverter(voice_mode=VOICE_MODE_CLONE,
                                backend=BACKEND_FASTER, voice="narrator",