def test_impossible_length_error(self):
sock = self.mox.CreateMock(socket.socket)
self.mox.StubOutWithMock(sock, 'recv')
sock.recv(1).AndReturn('ab')
self.mox.ReplayAll()
with self.assertRaises(io.SocketError):
io.recv_bytes(sock, 1)
def test_connection_closed(self):
sock = self.mox.CreateMock(socket.socket)
self.mox.StubOutWithMock(sock, 'recv')
sock.recv(1).AndReturn('')
self.mox.ReplayAll()
with self.assertRaises(io.ConnectionClosedError):
io.recv_bytes(sock, 1)
def test_socket_error(self):
sock = self.mox.CreateMock(socket.socket)
self.mox.StubOutWithMock(sock, 'recv')
sock.recv(1).AndRaise(socket.error)
self.mox.ReplayAll()
with self.assertRaises(socket.error):
io.recv_bytes(sock, 1)
def test_recv_bytes_2_split(self):
sock = self.mox.CreateMock(socket.socket)
self.mox.StubOutWithMock(sock, 'recv')
sock.recv(2).AndReturn('a')
sock.recv(1).AndReturn('b')
self.mox.ReplayAll()
self.assertEqual(io.recv_bytes(sock, 2), 'ab')
def test_recv_bytes_empty(self):
self.assertEqual(io.recv_bytes(None, 0), '')
