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_client.py
+3 -3
View File
@@ -8,7 +8,7 @@ from whisper_live.client import TranscriptionClient, resample
class TestClientWebSocketCommunication(unittest.TestCase): class TestClientWebSocketCommunication(unittest.TestCase):
@patch('whisper_live.client.websocket.WebSocketApp') @patch('websocket.WebSocketApp')
def test_websocket_communication(self, mock_websocket): def test_websocket_communication(self, mock_websocket):
mock_ws_instance = MagicMock() mock_ws_instance = MagicMock()
mock_websocket.return_value = mock_ws_instance mock_websocket.return_value = mock_ws_instance
@@ -23,7 +23,7 @@ class TestClientWebSocketCommunication(unittest.TestCase):
class TestClientCallbacks(unittest.TestCase): class TestClientCallbacks(unittest.TestCase):
@patch('whisper_live.client.websocket.WebSocketApp') @patch('websocket.WebSocketApp')
def setUp(self, mock_websocket): def setUp(self, mock_websocket):
self.mock_ws_app = mock_websocket.return_value self.mock_ws_app = mock_websocket.return_value
self.mock_ws_app.send = MagicMock() self.mock_ws_app.send = MagicMock()
@@ -97,7 +97,7 @@ class TestAudioResampling(unittest.TestCase):
class TestSendingAudioPacket(unittest.TestCase): class TestSendingAudioPacket(unittest.TestCase):
@patch('whisper_live.client.websocket.WebSocketApp') @patch('websocket.WebSocketApp')
def setUp(self, mock_websocket): def setUp(self, mock_websocket):
self.transcription_client = TranscriptionClient("localhost", "9090") self.transcription_client = TranscriptionClient("localhost", "9090")
self.client = self.transcription_client.client self.client = self.transcription_client.client
tests/test_server.py
+2 -2
View File
@@ -40,7 +40,7 @@ class TestServerConnection(unittest.TestCase):
def setUp(self): def setUp(self):
self.server = TranscriptionServer() self.server = TranscriptionServer()
@mock.patch('whisper_live.server.websockets.WebSocketCommonProtocol') @mock.patch('websockets.WebSocketCommonProtocol')
def test_connection(self, mock_websocket): def test_connection(self, mock_websocket):
mock_websocket.recv.return_value = json.dumps({ mock_websocket.recv.return_value = json.dumps({
'uid': 'test_client', 'uid': 'test_client',
@@ -51,7 +51,7 @@ class TestServerConnection(unittest.TestCase):
self.server.recv_audio(mock_websocket, "faster_whisper") 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): def test_recv_audio_exception_handling(self, mock_websocket):
mock_websocket.recv.side_effect = [json.dumps({ mock_websocket.recv.side_effect = [json.dumps({
'uid': 'test_client', 'uid': 'test_client',