def test_simple(self):
packets = ["abc\r\n", "abc\r", "\nd\r\nef\r\ngh", "a\rbc\r", "def\n",
"\r", "\n\r"]
expected = ["abc\n", "abc\n", "d\nef\ngh", "a\rbc\r", "def\n", "\n",
"\r"]
crlf = streamio.TextCRLFFilter(TSource(packets))
def f():
blocks = []
while True:
block = crlf.read(100)
if not block:
break
blocks.append(block)
assert blocks == expected
self.interpret(f, [])
def test_read1(self):
s_input = "abc\r\nabc\nd\r\nef\r\ngha\rbc\rdef\n\r\n\r"
s_output = "abc\nabc\nd\nef\ngha\rbc\rdef\n\n\r"
assert s_output == s_input.replace('\r\n', '\n')
packets = list(s_input)
expected = list(s_output)
crlf = streamio.TextCRLFFilter(TSource(packets))
def f():
blocks = []
while True:
block = crlf.read(1)
if not block:
break
blocks.append(block)
assert blocks == expected
self.interpret(f, [])
def test_write(self):
data = "line1\r\nline2\rline3\r\n"
crlf = streamio.TextCRLFFilter(TReaderWriter(data))
def f():
line = crlf.readline()
assert line == 'line1\n'
line = crlf.read(6)
assert line == 'line2\r'
pos = crlf.tell()
crlf.write('line3\n')
crlf.seek(pos,0)
line = crlf.readline()
assert line == 'line3\n'
line = crlf.readline()
assert line == ''
self.interpret(f, [])
def test_seek_relative(self):
packets = ["abc\r\n", "abc\r", "\nd\r\nef\r"]
expected = ["abc\n", "abc\n", "d\n","ef\r"]
crlf = streamio.TextCRLFFilter(TSource(packets))
def f():
lines = []
while True:
pos = crlf.tell()
line = crlf.readline()
if not line:
break
crlf.seek(0, 1)
lines.append(line)
assert lines == expected
self.interpret(f, [])
def test_readline_and_seek(self):
packets = ["abc\r\n", "abc\r", "\nd\r\nef\r\ngh", "a\rbc\r", "def\n",
"\r", "\n\r"]
expected = ["abc\n", "abc\n", "d\n","ef\n", "gha\rbc\rdef\n", "\n",
"\r"]
crlf = streamio.TextCRLFFilter(TSource(packets))
def f():
lines = []
while True:
pos = crlf.tell()
line = crlf.readline()
if not line:
break
crlf.seek(pos, 0)
line2 = crlf.readline()
assert line2 == line
lines.append(line)
assert lines == expected
self.interpret(f, [])
