1 files changed, 52 insertions, 0 deletions
diff --git a/tests/test_import_issues.py b/tests/test_import_issues.py
new file mode 100644
index 0000000..d6b2dbc
--- /dev/null
+++ b/tests/test_import_issues.py
@@ -0,0 +1,52 @@
+import os.path
+import textwrap
+import unittest
+
+from import_issues import MAX_SIZE_COMMENT, split_long_str
+
+
+class TestStringSplitting(unittest.TestCase):
+    maxDiff = None
+
+    def setUp(self):
+        __cur_dir = os.path.abspath(os.path.dirname(__file__))
+        with open(os.path.join(__cur_dir, "209_description.txt")) as inf:
+            self.long_str = inf.read()
+
+    def test_one_line(self):
+        in_str = "Testing string."
+        split_str = split_long_str(in_str, 100)
+        self.assertIsInstance(split_str, list)
+        self.assertEqual(len(split_str), 1)
+        self.assertEqual(in_str, "".join(split_str))
+
+    def test_short_str(self):
+        in_str = textwrap.dedent(
+            """\
+        Reader, I married him.
+        A quiet wedding we had: he and I, the parson and clerk, were alone present.
+        When we got back from church, I went into the kitchen of the manor-house, where Mary was cooking the dinner and John cleaning the knives, and I said—
+
+        “Mary, I have been married to Mr. Rochester this morning.”
+
+        The housekeeper and her husband were both of that decent phlegmatic order of people, to whom one may at any time safely communicate a remarkable piece of news without incurring the danger of having one’s ears pierced by some shrill ejaculation, and subsequently stunned by a torrent of wordy wonderment.
+        Mary did look up, and she did stare at me: the ladle with which she was basting a pair of chickens roasting at the fire, did for some three minutes hang suspended in air; and for the same space of time John’s knives also had rest from the polishing process: but Mary, bending again over the roast, said only—
+
+        “Have you, Miss? Well, for sure!”
+
+        """
+        )
+        split_str = split_long_str(in_str, 100)
+        self.assertIsInstance(split_str, list)
+        self.assertEqual(len(split_str), 6)
+        self.assertEqual(in_str, "".join(split_str))
+
+    def test_long_str(self):
+        split_str = split_long_str(self.long_str, MAX_SIZE_COMMENT)
+        self.assertIsInstance(split_str, list)
+        self.assertEqual(len(split_str), 4)
+        self.assertEqual(self.long_str, "".join(split_str))
+
+
+if __name__ == "__main__":
+    unittest.main()