From: Mathieu Bridon <[email protected]>

This ensures that we can read/write lines in the new format, as well as
mix lines in both format within a single sources file.
---
 test/test_sources.py | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/test/test_sources.py b/test/test_sources.py
index edff838..c0722b1 100644
--- a/test/test_sources.py
+++ b/test/test_sources.py
@@ -25,6 +25,7 @@ class formatLineTestCase(unittest.TestCase):
     def test_correct_entry(self):
         CORRECT_ENTRIES = [
             (['foo', 'bar'], ('foo  bar')),
+            (['foo', 'bar', 'baz'], ('foo  bar  baz')),
         ]
         for entry, line in CORRECT_ENTRIES:
             self.assertEqual(line,
@@ -53,7 +54,8 @@ class parseLineTestCase(unittest.TestCase):
     def test_correct_line(self):
         CORRECT_LINES = [
             ('foo  bar\n', ['foo', 'bar']),
-            ('foo   bar\n', ['foo', ' bar'])
+            ('foo   bar\n', ['foo', ' bar']),
+            ('foo   bar  baz\n', ['foo', ' bar', 'baz'])
         ]
         for line, entry in CORRECT_LINES:
             self.assertEqual(entry, sources._parse_line(line))
@@ -80,6 +82,10 @@ class ReaderTestCase(unittest.TestCase):
             ('foo  bar\nfooo  baaar\n', [['foo', 'bar'],
                                          ['fooo', 'baaar'],
                                          ]),
+            ('foo  bar  baz\n', [['foo', 'bar', 'baz']]),
+            ('foo  bar\nfoo  bar  baz\n', [['foo', 'bar'],
+                                           ['foo', 'bar', 'baz'],
+                                           ]),
         ]
         for buffer, entries in CORRECT_SOURCES:
             fp = StringIO.StringIO(buffer)
@@ -95,6 +101,9 @@ class WriterTestCase(unittest.TestCase):
             ([['foo', 'bar'],
               ['fooo', 'baaar'],
               ], 'foo  bar\nfooo  baaar\n'),
+            ([['foo', 'bar', 'baz']], 'foo  bar  baz\n'),
+            ([['foo', 'bar'], ['foo', 'bar', 'baz']],
+             'foo  bar\nfoo  bar  baz\n'),
         ]
         for entries, buffer in CORRECT_SOURCES:
             fp = StringIO.StringIO()
-- 
2.1.0

--
buildsys mailing list
[email protected]
https://admin.fedoraproject.org/mailman/listinfo/buildsys

Reply via email to