Fix max_client, max_connection_time failed tests
Signed-off-by: makaveli10 <vineet.suryan@collabora.com>
This commit is contained in:
@@ -49,8 +49,6 @@ class TestClientCallbacks(BaseTestCase):
|
|||||||
"task": self.client.task,
|
"task": self.client.task,
|
||||||
"model": self.client.model,
|
"model": self.client.model,
|
||||||
"use_vad": True,
|
"use_vad": True,
|
||||||
"max_clients": 4,
|
|
||||||
"max_connection_time": 600,
|
|
||||||
"send_last_n_segments": 10,
|
"send_last_n_segments": 10,
|
||||||
"no_speech_thresh": 0.45,
|
"no_speech_thresh": 0.45,
|
||||||
"clip_audio": False,
|
"clip_audio": False,
|
||||||
|
|||||||
@@ -42,6 +42,8 @@ class TestGetWaitTime(unittest.TestCase):
|
|||||||
class TestServerConnection(unittest.TestCase):
|
class TestServerConnection(unittest.TestCase):
|
||||||
def setUp(self):
|
def setUp(self):
|
||||||
self.server = TranscriptionServer()
|
self.server = TranscriptionServer()
|
||||||
|
self.server.client_manager = ClientManager(max_clients=4, max_connection_time=600)
|
||||||
|
self.server.cache_path = "~/.cache/whisper-live/"
|
||||||
|
|
||||||
@mock.patch('websockets.WebSocketCommonProtocol')
|
@mock.patch('websockets.WebSocketCommonProtocol')
|
||||||
def test_connection(self, mock_websocket):
|
def test_connection(self, mock_websocket):
|
||||||
|
|||||||
Reference in New Issue
Block a user