Unify separation tools and add exclusive control mode

This commit is contained in:
Mikei386 committed 2026-10-01 16:56:13 +02:00
1 parent 088f5f6f91
commit bd5aea7656
13 files changed
+81 -28

No files matched your search

+6
View File
@@ -90,6 +90,12 @@ class EndpointTests(unittest.TestCase):
response=conn.getresponse();self.assertEqual(response.status,200);self.assertEqual(response.getheader('Content-Type'),'audio/wav');self.assertEqual(response.read(),b'RIFFsynthetic-WAV');conn.close()
self.assertEqual(self.request('/v1/audio/music',dict(model='Music',lyrics='Synthetic',style='pop',unsupported=True))[0],400)
self.assertEqual(self.request('/v1/audio/music/models',token='bad')[0],401)
def test_separator_mode_rejects_chat_without_starting_worker_or_relay(self):
self.ep.scheduler.gpu_mode='separator';self.ep.video=Mock()
status,error=self.request('/v1/chat/completions',dict(model='alpha',messages=[dict(role='user',content='Synthetic')]))
self.assertEqual(status,503);self.assertEqual(error['error']['code'],'separator_mode_active')
self.assertEqual(self.worker.started,[]);self.ep.video.relay.assert_not_called()
self.assertEqual(self.request('/v1/models')[0],503)
def enable(self,name):self.ep.enable(dict(id=name,enabled=True))
def test_explicit_publication_auth_rotation_and_missing_audio(self):
self.assertEqual(self.request()[1]['data'],[]);self.assertEqual(self.request(token='bad')[0],401)