From 7c7a44647841d9ec8fec9256f7fcfcf39aa62dd0 Mon Sep 17 00:00:00 2001
From: makaveli <39617050+makaveli10@users.noreply.github.com>
Date: Fri, 15 Mar 2024 12:29:32 +0530
Subject: [PATCH] Fix: mock pyaudio for ci to pass the server tests

---
 tests/test_server.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/tests/test_server.py b/tests/test_server.py
index a7c8ddf..f836be7 100644
--- a/tests/test_server.py
+++ b/tests/test_server.py
@@ -69,6 +69,10 @@ class TestServerConnection(unittest.TestCase):
 class TestServerInferenceAccuracy(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
+        cls.mock_pyaudio_patch = mock.patch('pyaudio.PyAudio')
+        cls.mock_pyaudio = cls.mock_pyaudio_patch.start()
+        cls.mock_pyaudio.return_value.open.return_value = mock.MagicMock()
+        
         cls.server_process = subprocess.Popen(["python", "run_server.py"])
         time.sleep(2)
 
@@ -77,11 +81,7 @@ class TestServerInferenceAccuracy(unittest.TestCase):
         cls.server_process.terminate()
         cls.server_process.wait()
 
-    @mock.patch('pyaudio.PyAudio')
-    def setUp(self, mock_pyaudio):
-        self.mock_pyaudio = mock_pyaudio.return_value
-        self.mock_stream = mock.MagicMock()
-        self.mock_pyaudio.open.return_value = self.mock_stream
+    def setUp(self):
         self.metric = evaluate.load("wer")
         self.normalizer = EnglishTextNormalizer()
 
@@ -117,6 +117,7 @@ class TestServerInferenceAccuracy(unittest.TestCase):
         self.check_prediction("transcript1.srt")
         self.check_prediction("transcript2.srt")
 
+
 class TestExceptionHandling(unittest.TestCase):
     def setUp(self):
         self.server = TranscriptionServer()