def test_recv(self):
self.sock.recv(IsA(int)).AndReturn(b'\r\nthree\r\n')
self.sock.recv(IsA(int)).AndReturn(b'.\r\nstuff\r\n')
self.mox.ReplayAll()
io = IO(self.sock)
io.recv_buffer = b'one\r\ntwo'
dr = DataReader(io)
self.assertEqual(b'one\r\ntwo\r\nthree\r\n', dr.recv())
def test_recv(self):
self.sock.recv(IsA(int)).AndReturn("\r\nthree\r\n")
self.sock.recv(IsA(int)).AndReturn(".\r\nstuff\r\n")
self.mox.ReplayAll()
io = IO(self.sock)
io.recv_buffer = "one\r\ntwo"
dr = DataReader(io)
self.assertEqual("one\r\ntwo\r\nthree\r\n", dr.recv())
def test_from_recv_buffer(self):
io = IO(None)
io.recv_buffer = b'test\r\ndata'
dr = DataReader(io)
dr.from_recv_buffer()
self.assertEqual([b'test\r\n', b'data'], dr.lines)
def test_from_recv_buffer(self):
io = IO(None)
io.recv_buffer = "test\r\ndata"
dr = DataReader(io)
dr.from_recv_buffer()
self.assertEqual(["test\r\n", "data"], dr.lines)
def test_from_recv_buffer(self):
io = IO(None)
io.recv_buffer = 'test\r\ndata'
dr = DataReader(io)
dr.from_recv_buffer()
assert_equal(['test\r\n', 'data'], dr.lines)
