stor-toolchain/test/writers/test_file_concatenator.py

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()