def test_add_lines(self):
dr = DataReader(None)
dr.add_lines('\r\ntwo\r\n.three\r\nfour')
assert_equal(['\r\n', 'two\r\n', 'three\r\n', 'four'], dr.lines)
assert_equal(3, dr.i)
assert_equal(None, dr.EOD)
def test_handle_finished_line_EOD(self):
dr = DataReader(None)
dr.lines = ['.\r\n']
dr.handle_finished_line()
assert_equal(0, dr.EOD)
def test_handle_finished_line_initial_period(self):
dr = DataReader(None)
dr.lines = ['..stuff\r\n']
dr.handle_finished_line()
assert_equal('.stuff\r\n', dr.lines[0])
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)
