fixed unix server/client tests

2022-03-08 10:22:07 +01:00
parent d05f84b2c3
commit eb152e4d75
2 changed files with 6 additions and 6 deletions
@@ -190,9 +190,9 @@ class UnixControlClientTestCase(IsolatedAsyncioTestCase):
        self.mock_base_init.assert_called_once_with(**self.kwargs)

    @patch.object(client, 'print')
-    @patch.object(client, 'open_unix_connection')
-    async def test__open_connection(self, mock_open_unix_connection: AsyncMock, mock_print: MagicMock):
-        mock_open_unix_connection.return_value = expected_output = 'something'
+    async def test__open_connection(self, mock_print: MagicMock):
+        expected_output = 'something'
+        self.client._open_unix_connection = mock_open_unix_connection = AsyncMock(return_value=expected_output)
        kwargs = {'a': 1, 'b': 2}
        output = await self.client._open_connection(**kwargs)
        self.assertEqual(expected_output, output)