comps-wrapper: Add tests

This tests pretty much all the code that parses, filters and then writes
comps back to a file.

Signed-off-by: Lubomír Sedlář <lsedlar@redhat.com>

tests/test_comps_wrapper.py

@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+
+try:
+    import unittest2 as unittest
+except ImportError:
+    import unittest
+import tempfile
+import difflib
+
+import os
+import sys
+
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
+
+from pungi.wrappers.comps import CompsWrapper
+from tests.helpers import FIXTURE_DIR
+
+COMPS_FILE = os.path.join(FIXTURE_DIR, 'comps.xml')
+COMPS_FORMATTED_FILE = os.path.join(FIXTURE_DIR, 'comps-formatted.xml')
+COMPS_GROUP_FILE = os.path.join(FIXTURE_DIR, 'comps-group.xml')
+COMPS_ENVIRONMENT_FILE = os.path.join(FIXTURE_DIR, 'comps-env.xml')
+
+
+class CompsWrapperTest(unittest.TestCase):
+    def setUp(self):
+        self.file = tempfile.NamedTemporaryFile(prefix='comps-wrapper-test-')
+
+    def assertFilesEqual(self, fn1, fn2):
+        with open(fn1) as f1:
+            lines1 = f1.read().splitlines()
+        with open(fn2) as f2:
+            lines2 = f2.read().splitlines()
+        diff = '\n'.join(difflib.unified_diff(lines1, lines2,
+                                              fromfile='EXPECTED', tofile='ACTUAL'))
+        self.assertEqual(diff, '', 'Files differ:\n' + diff)
+
+    def test_get_groups(self):
+        comps = CompsWrapper(COMPS_FILE)
+        self.assertItemsEqual(
+            comps.get_comps_groups(),
+            ['core', 'standard', 'text-internet', 'firefox', 'resilient-storage', 'basic-desktop'])
+
+    def test_write_comps(self):
+        comps = CompsWrapper(COMPS_FILE)
+        comps.write_comps(target_file=self.file.name)
+        self.assertFilesEqual(COMPS_FORMATTED_FILE, self.file.name)
+
+    def test_filter_groups(self):
+        comps = CompsWrapper(COMPS_FILE)
+        unmatched = comps.filter_groups([
+            {"name": "core", "glob": False, "default": False, "uservisible": True},
+            {"name": "*a*", "glob": True, "default": None, "uservisible": None},
+        ])
+        self.assertEqual(unmatched, set())
+        comps.write_comps(target_file=self.file.name)
+        self.assertFilesEqual(COMPS_GROUP_FILE, self.file.name)
+
+    def test_filter_groups_unused_filter(self):
+        comps = CompsWrapper(COMPS_FILE)
+        unmatched = comps.filter_groups([
+            {"name": "boom", "glob": False, "default": False, "uservisible": True},
+        ])
+        self.assertEqual(unmatched, set(["boom"]))
+
+    def test_filter_environments(self):
+        comps = CompsWrapper(COMPS_FILE)
+        comps.filter_environments([
+            {"name": "minimal", "display_order": 10}
+        ])
+        comps.write_comps(target_file=self.file.name)
+        self.assertFilesEqual(COMPS_ENVIRONMENT_FILE, self.file.name)
+
+    def test_read_display_order(self):
+        comps = CompsWrapper(COMPS_FILE)
+        groups = [
+            {"name": "minimal", "display_order": None}
+        ]
+        comps.filter_environments(groups)
+        self.assertEqual(groups, [{"name": "minimal", "display_order": 99, "groups": ["core"]}])