diff --git a/tests/test_client.py b/tests/test_client.py index f38d28b..b140fc5 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -49,8 +49,6 @@ class TestClientCallbacks(BaseTestCase): "task": self.client.task, "model": self.client.model, "use_vad": True, - "max_clients": 4, - "max_connection_time": 600, "send_last_n_segments": 10, "no_speech_thresh": 0.45, "clip_audio": False, diff --git a/tests/test_server.py b/tests/test_server.py index de78702..71f2fb2 100644 --- a/tests/test_server.py +++ b/tests/test_server.py @@ -42,6 +42,8 @@ class TestGetWaitTime(unittest.TestCase): class TestServerConnection(unittest.TestCase): def setUp(self): 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') def test_connection(self, mock_websocket):