16ac225013
This also simplifies the test data by removing a lot of the langpacks. Signed-off-by: Lubomír Sedlář <lsedlar@redhat.com>
214 lines
8.0 KiB
Python
214 lines
8.0 KiB
Python
# -*- 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, CompsFilter, CompsValidationError
|
|
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')
|
|
COMPS_FILE_WITH_TYPO = os.path.join(FIXTURE_DIR, 'comps-typo.xml')
|
|
COMPS_FILE_WITH_WHITESPACE = os.path.join(FIXTURE_DIR, 'comps-ws.xml')
|
|
|
|
|
|
class CompsWrapperTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.file = tempfile.NamedTemporaryFile(prefix='comps-wrapper-test-')
|
|
|
|
def assertFilesEqual(self, fn1, fn2):
|
|
with open(fn1, 'rb') as f1:
|
|
lines1 = f1.read().decode('utf-8').splitlines()
|
|
with open(fn2, 'rb') as f2:
|
|
lines2 = f2.read().decode('utf-8').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.assertEqual(
|
|
sorted(comps.get_comps_groups()),
|
|
sorted(['core', 'standard', 'text-internet', 'firefox', 'resilient-storage', 'basic-desktop']))
|
|
|
|
def test_get_packages(self):
|
|
comps = CompsWrapper(COMPS_FILE)
|
|
self.assertEqual(
|
|
sorted(comps.get_packages('text-internet')),
|
|
sorted(['dummy-elinks', 'dummy-tftp']))
|
|
|
|
def test_get_langpacks(self):
|
|
comps = CompsWrapper(COMPS_FILE)
|
|
self.assertEqual(
|
|
comps.get_langpacks(),
|
|
{
|
|
"aspell": "aspell-%s",
|
|
"firefox": "firefox-langpack-%s",
|
|
"kdelibs": "kde-l10n-%s",
|
|
}
|
|
)
|
|
|
|
def test_get_packages_for_non_existing_group(self):
|
|
comps = CompsWrapper(COMPS_FILE)
|
|
with self.assertRaises(KeyError):
|
|
comps.get_packages('foo')
|
|
|
|
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"]}])
|
|
|
|
def test_report_typo_in_package_type(self):
|
|
comps = CompsWrapper(COMPS_FILE_WITH_TYPO)
|
|
with self.assertRaises(RuntimeError) as ctx:
|
|
comps.write_comps(target_file=self.file.name)
|
|
self.assertIn(
|
|
'Package dummy-bash in group core has unknown type',
|
|
str(ctx.exception))
|
|
|
|
def test_validate_correct(self):
|
|
comps = CompsWrapper(COMPS_FILE)
|
|
comps.validate()
|
|
|
|
def test_validate_with_whitespace(self):
|
|
comps = CompsWrapper(COMPS_FILE_WITH_WHITESPACE)
|
|
with self.assertRaises(CompsValidationError) as ctx:
|
|
comps.validate()
|
|
|
|
self.assertIn(
|
|
"Package name foo in group 'core' contains leading or trailing whitespace",
|
|
str(ctx.exception),
|
|
)
|
|
self.assertIn(
|
|
"Package name bar in group 'core' contains leading or trailing whitespace",
|
|
str(ctx.exception),
|
|
)
|
|
self.assertIn(
|
|
"Package name baz in group 'core' contains leading or trailing whitespace",
|
|
str(ctx.exception),
|
|
)
|
|
|
|
|
|
COMPS_IN_FILE = os.path.join(FIXTURE_DIR, 'comps.xml.in')
|
|
|
|
|
|
class CompsFilterTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.filter = CompsFilter(COMPS_IN_FILE, reindent=True)
|
|
self.output = tempfile.NamedTemporaryFile(prefix='comps-filter-test-')
|
|
|
|
def assertOutput(self, filepath):
|
|
self.filter.write(self.output)
|
|
self.output.flush()
|
|
with open(self.output.name, 'r') as f:
|
|
actual = f.read().strip().replace('utf-8', 'UTF-8')
|
|
with open(filepath, 'r') as f:
|
|
expected = f.read().strip()
|
|
self.maxDiff = None
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_filter_packages(self):
|
|
self.filter.filter_packages('ppc64le', None)
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-filtered-packages.xml'))
|
|
|
|
def test_filter_packages_with_variant(self):
|
|
self.filter.filter_packages('ppc64le', 'Server')
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-filtered-packages-variant.xml'))
|
|
|
|
def test_filter_groups(self):
|
|
self.filter.filter_groups('ppc64le', None)
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-filtered-groups.xml'))
|
|
|
|
def test_filter_groups_with_variant(self):
|
|
self.filter.filter_groups('ppc64le', 'Server')
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-filtered-groups-variant.xml'))
|
|
|
|
def test_filter_environments(self):
|
|
self.filter.filter_environments('ppc64le', None)
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-filtered-environments.xml'))
|
|
|
|
def test_filter_environments_variant(self):
|
|
self.filter.filter_environments('ppc64le', 'Client')
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-filtered-environments-variant.xml'))
|
|
|
|
def test_remove_categories(self):
|
|
self.filter.remove_categories()
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-removed-categories.xml'))
|
|
|
|
def test_remove_langpacks(self):
|
|
self.filter.remove_langpacks()
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-removed-langpacks.xml'))
|
|
|
|
def test_remove_translations(self):
|
|
self.filter.remove_translations()
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-removed-translations.xml'))
|
|
|
|
def test_remove_environments(self):
|
|
self.filter.remove_environments()
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-removed-environments.xml'))
|
|
|
|
def test_cleanup(self):
|
|
self.filter.cleanup()
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-cleanup.xml'))
|
|
|
|
def test_cleanup_after_filter(self):
|
|
self.filter.filter_packages('ppc64le', None)
|
|
self.filter.cleanup()
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-cleanup-filter.xml'))
|
|
|
|
def test_cleanup_after_filter_keep_group(self):
|
|
self.filter.filter_packages('ppc64le', None)
|
|
self.filter.cleanup(['standard'])
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-cleanup-keep.xml'))
|
|
|
|
def test_cleanup_all(self):
|
|
self.filter.filter_packages('ppc64le', None)
|
|
self.filter.filter_groups('ppc64le', None)
|
|
self.filter.filter_environments('ppc64le', None)
|
|
self.filter.cleanup()
|
|
self.assertOutput(os.path.join(FIXTURE_DIR, 'comps-cleanup-all.xml'))
|