From 93c16cc2cdc723344e1792828a42c3db78a17780 Mon Sep 17 00:00:00 2001 From: Thomas Adamcik Date: Thu, 28 Jul 2011 22:43:54 +0200 Subject: [PATCH] Add tests for socket sending --- tests/utils/network/connection_test.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/utils/network/connection_test.py b/tests/utils/network/connection_test.py index 7241dc5a..16ec3979 100644 --- a/tests/utils/network/connection_test.py +++ b/tests/utils/network/connection_test.py @@ -501,6 +501,20 @@ class ConnectionTest(unittest.TestCase): network.Connection.send(self.mock, 'data') self.assertEqual(0, self.mock.stop.call_count) + def test_send_calls_socket_send(self): + self.mock.sock = Mock(spec=socket.SocketType) + self.mock.sock.send.return_value = 4 + + self.assertEqual('', network.Connection.send(self.mock, 'data')) + self.mock.sock.send.assert_called_once_with('data') + + def test_send_calls_socket_send_partial_send(self): + self.mock.sock = Mock(spec=socket.SocketType) + self.mock.sock.send.return_value = 2 + + self.assertEqual('ta', network.Connection.send(self.mock, 'data')) + self.mock.sock.send.assert_called_once_with('data') + def test_send_unrecoverable_error(self): self.mock.sock = Mock(spec=socket.SocketType) self.mock.sock.send.side_effect = socket.error