andpp/mopidy
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Add tests for socket sending

Browse Source
This commit is contained in:
Thomas Adamcik 2011-07-28 22:43:54 +02:00
parent 4f6ddd3532
commit 93c16cc2cd
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/utils/network/connection_test.py
View File

@ -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