mirror of https://github.com/docker/compose.git
55 lines
1.4 KiB
Python
55 lines
1.4 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
|
|
from .. import unittest
|
|
from compose.utils import split_buffer
|
|
|
|
|
|
class SplitBufferTest(unittest.TestCase):
|
|
def test_single_line_chunks(self):
|
|
def reader():
|
|
yield b'abc\n'
|
|
yield b'def\n'
|
|
yield b'ghi\n'
|
|
|
|
self.assert_produces(reader, ['abc\n', 'def\n', 'ghi\n'])
|
|
|
|
def test_no_end_separator(self):
|
|
def reader():
|
|
yield b'abc\n'
|
|
yield b'def\n'
|
|
yield b'ghi'
|
|
|
|
self.assert_produces(reader, ['abc\n', 'def\n', 'ghi'])
|
|
|
|
def test_multiple_line_chunk(self):
|
|
def reader():
|
|
yield b'abc\ndef\nghi'
|
|
|
|
self.assert_produces(reader, ['abc\n', 'def\n', 'ghi'])
|
|
|
|
def test_chunked_line(self):
|
|
def reader():
|
|
yield b'a'
|
|
yield b'b'
|
|
yield b'c'
|
|
yield b'\n'
|
|
yield b'd'
|
|
|
|
self.assert_produces(reader, ['abc\n', 'd'])
|
|
|
|
def test_preserves_unicode_sequences_within_lines(self):
|
|
string = u"a\u2022c\n"
|
|
|
|
def reader():
|
|
yield string.encode('utf-8')
|
|
|
|
self.assert_produces(reader, [string])
|
|
|
|
def assert_produces(self, reader, expectations):
|
|
split = split_buffer(reader())
|
|
|
|
for (actual, expected) in zip(split, expectations):
|
|
assert type(actual) == type(expected)
|
|
assert actual == expected
|