def setup_serializer(self): self.test_dir = mkdtemp() self.serial = OutputStreamSerializer([self.test_dir]) self.filename = "STARS" self.content = "123\n456\n789" self.expected = [ line.encode('ascii') for line in self.content.split('\n') ]
def setup_serializer(self): self.test_dir = mkdtemp() self.serial = OutputStreamSerializer([self.test_dir]) self.filename = "STARS" self.content = "123\n456\n789" self.expected = [ line for line in self.content.split('\n') ]
class OutputStreamSerializerTestCase(TestCase): @setup def setup_serializer(self): self.test_dir = mkdtemp() self.serial = OutputStreamSerializer([self.test_dir]) self.filename = "STARS" self.content = "123\n456\n789" self.expected = [ line.encode('ascii') for line in self.content.split('\n') ] @teardown def teardown_test_dir(self): shutil.rmtree(self.test_dir) def _write_contents(self): with open(self.serial.full_path(self.filename), 'w') as f: f.write(self.content) def test_open(self): with self.serial.open(self.filename) as fh: fh.write(self.content) with open(self.serial.full_path(self.filename)) as f: assert_equal(f.read(), self.content) @suite('integration') def test_init_with_output_path(self): path = OutputPath(self.test_dir, 'one', 'two', 'three') stream = OutputStreamSerializer(path) assert_equal(stream.base_path, str(path)) def test_tail(self): self._write_contents() assert_equal(self.serial.tail(self.filename), self.expected) def test_tail_num_lines(self): self._write_contents() assert_equal(self.serial.tail(self.filename, 1), self.expected[-1:]) def test_tail_file_does_not_exist(self): file_dne = 'bogusfile123' assert_equal(self.serial.tail(file_dne), [])
class TestOutputStreamSerializer(TestCase): @setup def setup_serializer(self): self.test_dir = mkdtemp() self.serial = OutputStreamSerializer([self.test_dir]) self.filename = "STARS" self.content = "123\n456\n789" self.expected = [ line for line in self.content.split('\n') ] @teardown def teardown_test_dir(self): shutil.rmtree(self.test_dir) def _write_contents(self): with open(self.serial.full_path(self.filename), 'w') as f: f.write(self.content) def test_open(self): with self.serial.open(self.filename) as fh: fh.write(self.content) with open(self.serial.full_path(self.filename)) as f: assert_equal(f.read(), self.content) @suite('integration') def test_init_with_output_path(self): path = OutputPath(self.test_dir, 'one', 'two', 'three') stream = OutputStreamSerializer(path) assert_equal(stream.base_path, str(path)) def test_tail(self): self._write_contents() assert_equal(self.serial.tail(self.filename), self.expected) def test_tail_num_lines(self): self._write_contents() assert_equal(self.serial.tail(self.filename, 1), self.expected[-1:]) def test_tail_file_does_not_exist(self): file_dne = 'bogusfile123' assert_equal(self.serial.tail(file_dne), [])
def test_init_with_output_path(self): path = OutputPath(self.test_dir, 'one', 'two', 'three') stream = OutputStreamSerializer(path) assert_equal(stream.base_path, str(path))