diff options
Diffstat (limited to 'app/tests/test_backends_probe.py')
| -rw-r--r-- | app/tests/test_backends_probe.py | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/app/tests/test_backends_probe.py b/app/tests/test_backends_probe.py index 08f9fd9..742e208 100644 --- a/app/tests/test_backends_probe.py +++ b/app/tests/test_backends_probe.py @@ -63,6 +63,13 @@ class IdentifyServerTests(unittest.TestCase): self.assertEqual(probe.identify_server("http://x:7861"), "qwen-clone") + def test_qwen_design_identified(self): + with patch.object(probe.common, "server_running", return_value=True), \ + self._patch_http({"/info": {"named_endpoints": + {"/run_voice_design": {}}}}): + self.assertEqual(probe.identify_server("http://x:7860"), + "qwen-design") + def test_unreachable_returns_none(self): with patch.object(probe.common, "server_running", return_value=False): self.assertIsNone(probe.identify_server("http://127.0.0.1:8080")) |