Unify separation tools and add exclusive control mode
This commit is contained in:
1 parent
088f5f6f91
commit
bd5aea7656
13 files changed
+81
-28
No files matched your search
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user