diff --git a/tests/utils/network/test_connection.py b/tests/utils/network/test_connection.py index 7435353a..c3200689 100644 --- a/tests/utils/network/test_connection.py +++ b/tests/utils/network/test_connection.py @@ -7,7 +7,7 @@ import unittest import gobject -from mock import Mock, patch, sentinel +from mock import Mock, call, patch, sentinel import pykka @@ -418,8 +418,11 @@ class ConnectionTest(unittest.TestCase): self.assertTrue(network.Connection.recv_callback( self.mock, sentinel.fd, gobject.IO_IN)) - self.mock.disable_recv.assert_called_once_with() - self.mock.actor_ref.tell.assert_called_once_with({'close': True}) + self.assertEqual(self.mock.mock_calls, [ + call.sock.recv(any_int), + call.disable_recv(), + call.actor_ref.tell({'close': True}), + ]) def test_recv_callback_recoverable_error(self): self.mock.sock = Mock(spec=socket.SocketType)