cjones/WhisperLive
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

remove whisper_live from patch to mock websocket

Browse Source
This commit is contained in:
makaveli10
2024-02-07 23:47:29 +05:30
parent ca2634bbb6
commit 84a97f5fdd
2 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_server.py
+2 -2
View File
@@ -40,7 +40,7 @@ class TestServerConnection(unittest.TestCase):
def setUp(self):
self.server = TranscriptionServer()
@mock.patch('whisper_live.server.websockets.WebSocketCommonProtocol')
@mock.patch('websockets.WebSocketCommonProtocol')
def test_connection(self, mock_websocket):
mock_websocket.recv.return_value = json.dumps({
'uid': 'test_client',
@@ -51,7 +51,7 @@ class TestServerConnection(unittest.TestCase):
self.server.recv_audio(mock_websocket, "faster_whisper")
@mock.patch('whisper_live.server.websockets.WebSocketCommonProtocol')
@mock.patch('websockets.WebSocketCommonProtocol')
def test_recv_audio_exception_handling(self, mock_websocket):
mock_websocket.recv.side_effect = [json.dumps({
'uid': 'test_client',