Fixes mock socket in unit tests

cnpem-emi · Sep 12, 2022 · adb3847 · adb3847
1 parent 9434c52
commit adb3847
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 4 deletions.
diff --git a/tests/unit/test_read_vars_common.py b/tests/unit/test_read_vars_common.py
@@ -7,8 +7,10 @@
 
 
 class TestCommon(unittest.TestCase):
+    @patch("socket.socket")
     @patch("pydrs.EthDRS._transfer")
-    def setUp(self, transfer):
+    def setUp(self, transfer, socket):
+        socket = socket.return_value
         self.drs = GenericDRS("127.0.0.1", 5000)
         self.drs._transfer = transfer
 

diff --git a/tests/unit/test_read_vars_fac.py b/tests/unit/test_read_vars_fac.py
@@ -7,8 +7,10 @@
 
 
 class TestFAC(unittest.TestCase):
+    @patch("socket.socket")
     @patch("pydrs.EthDRS._transfer")
-    def setUp(self, transfer):
+    def setUp(self, transfer, socket):
+        socket = socket.return_value
         self.drs = GenericDRS("127.0.0.1", 5000)
         self.drs._transfer = transfer
 

diff --git a/tests/unit/test_read_vars_fap.py b/tests/unit/test_read_vars_fap.py
@@ -7,8 +7,10 @@
 
 
 class TestFAP(unittest.TestCase):
+    @patch("socket.socket")
     @patch("pydrs.EthDRS._transfer")
-    def setUp(self, transfer):
+    def setUp(self, transfer, socket):
+        socket = socket.return_value
         self.drs = GenericDRS("127.0.0.1", 5000)
         self.drs._transfer = transfer
 

diff --git a/tests/unit/test_read_vars_fbp.py b/tests/unit/test_read_vars_fbp.py
@@ -7,8 +7,10 @@
 
 
 class TestFBP(unittest.TestCase):
+    @patch("socket.socket")
     @patch("pydrs.EthDRS._transfer")
-    def setUp(self, transfer):
+    def setUp(self, transfer, socket):
+        socket = socket.return_value
         self.drs = GenericDRS("127.0.0.1", 5000)
         self.drs._transfer = transfer