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), '')