1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950 |
- import io
- import unittest
- from toolchain.writers.file_concatenator import FileConcatenator
- class TestFileConcatenator(unittest.TestCase):
- def setUp(self):
- self.output = io.StringIO()
- self.concatenator = FileConcatenator()
- def tearDown(self):
- self.output.close()
- def make_input(self, lines):
- input = io.StringIO()
- for line in lines:
- input.write(line + "\n")
- input.seek(0)
- return input
- def test_empty(self):
- input = self.make_input([])
- self.concatenator.concatenate_file(input, self.output)
- self.assertEqual(self.output.getvalue(), "")
- def test_single_line(self):
- input = self.make_input(["aniseed"])
- self.concatenator.concatenate_file(input, self.output)
- self.assertEqual(self.output.getvalue(), "aniseed\n")
- def test_multiple_line(self):
- input = self.make_input(["aniseed", "basil", "cinnamon dill", "elderflower"])
- self.concatenator.concatenate_file(input, self.output)
- self.assertEqual(self.output.getvalue(), "aniseed\nbasil\ncinnamon dill\nelderflower\n")
- if __name__ == "__main__":
- unittest.main()
|