Tests: Add LicenseManager tests

diff --git a/pym/portage/tests/ebuild/test_config.py b/pym/portage/tests/ebuild/test_config.py
index e9b474597dce3be30f47819afe3750ab6f94499e..ef4b585a364631ca7be853d453fa1f7daed5e919 100644 (file)
--- a/pym/portage/tests/ebuild/test_config.py
+++ b/pym/portage/tests/ebuild/test_config.py
@@ -1,7 +1,11 @@
 # Copyright 2010 Gentoo Foundation
 # Distributed under the terms of the GNU General Public License v2
 
+import portage
+from portage import os
+from portage.dep import ExtendedAtomDict
 from portage.package.ebuild.config import config
+from portage.package.ebuild._config.LicenseManager import LicenseManager
 from portage.tests import TestCase
 from portage.tests.resolver.ResolverPlayground import ResolverPlayground
 
@@ -32,3 +36,67 @@ class ConfigTestCase(TestCase):
                        self.assertEqual('noclean' in settings['FEATURES'].split(),True)
                finally:
                        playground.cleanup()
+
+       def testLicenseManager(self):
+
+               user_config = {
+                       "package.license":
+                               (
+                                       "dev-libs/* TEST",
+                                       "dev-libs/A -TEST2", 
+                                       "=dev-libs/A-2 TEST3 @TEST",
+                                       "*/* @EULA TEST2",
+                                       "=dev-libs/C-1 *",
+                                       "=dev-libs/C-2 -*",
+                               ),
+               }
+
+               playground = ResolverPlayground(user_config=user_config)
+               try:
+                       portage.util.noiselimit = -2
+
+                       license_group_locations = (os.path.join(playground.portdir, "profiles"),)
+                       pkg_license = os.path.join(playground.eroot, "etc", "portage")
+
+                       lic_man = LicenseManager()
+
+                       lic_man.parse_license_groups(license_group_locations)
+                       lic_man.read_config_files(pkg_license)
+                       lic_man_backup = LicenseManager(clone=lic_man)
+
+                       self.assertEqual(lic_man._accept_license_str, None)
+                       self.assertEqual(lic_man._accept_license_str, lic_man_backup._accept_license_str)
+                       self.assertEqual(lic_man._accept_license, None)
+                       self.assertEqual(lic_man._accept_license, lic_man_backup._accept_license)
+                       self.assertEqual(lic_man._license_groups, {"EULA": ["TEST"]})
+                       self.assertEqual(lic_man._license_groups, lic_man_backup._license_groups)
+                       self.assertEqual(lic_man._plicensedict, lic_man_backup._plicensedict)
+                       self.assertEqual(lic_man._undef_lic_groups, set(["TEST"]))
+                       self.assertEqual(lic_man._undef_lic_groups, lic_man_backup._undef_lic_groups)
+
+                       self.assertEqual(lic_man.extract_global_changes(), "TEST TEST2")
+                       self.assertEqual(lic_man.extract_global_changes(), "")
+                       self.assertEqual(lic_man_backup.extract_global_changes("TEST3 TEST4"), "TEST3 TEST4 TEST TEST2")
+                       self.assertEqual(lic_man_backup.extract_global_changes("TEST3 TEST4"), "TEST3 TEST4")
+
+                       lic_man.set_accept_license_str("TEST TEST2")
+                       self.assertEqual(lic_man._getPkgAcceptLicense("dev-libs/B-1", "0"), ["TEST", "TEST2", "TEST"])
+                       self.assertEqual(lic_man._getPkgAcceptLicense("dev-libs/A-1", "0"), ["TEST", "TEST2", "TEST", "-TEST2"])
+                       self.assertEqual(lic_man._getPkgAcceptLicense("dev-libs/A-2", "0"), ["TEST", "TEST2", "TEST", "-TEST2", "TEST3", "@TEST"])
+
+                       self.assertEqual(lic_man.get_prunned_accept_license("dev-libs/B-1", [], "TEST", "0"), "TEST")
+                       self.assertEqual(lic_man.get_prunned_accept_license("dev-libs/A-1", [], "-TEST2", "0"), "")
+                       self.assertEqual(lic_man.get_prunned_accept_license("dev-libs/A-2", [], "|| ( TEST TEST2 )", "0"), "TEST")
+                       self.assertEqual(lic_man.get_prunned_accept_license("dev-libs/C-1", [], "TEST5", "0"), "TEST5")
+                       self.assertEqual(lic_man.get_prunned_accept_license("dev-libs/C-2", [], "TEST2", "0"), "")
+
+                       self.assertEqual(lic_man.getMissingLicenses("dev-libs/B-1", [], "TEST", "0"), [])
+                       self.assertEqual(lic_man.getMissingLicenses("dev-libs/A-1", [], "-TEST2", "0"), ["-TEST2"])
+                       self.assertEqual(lic_man.getMissingLicenses("dev-libs/A-2", [], "|| ( TEST TEST2 )", "0"), [])
+                       self.assertEqual(lic_man.getMissingLicenses("dev-libs/A-3", [], "|| ( TEST2 || ( TEST3 TEST4 ) )", "0"), ["TEST2", "TEST3", "TEST4"])
+                       self.assertEqual(lic_man.getMissingLicenses("dev-libs/C-1", [], "TEST5", "0"), [])
+                       self.assertEqual(lic_man.getMissingLicenses("dev-libs/C-2", [], "TEST2", "0"), ["TEST2"])
+                       self.assertEqual(lic_man.getMissingLicenses("dev-libs/D-1", [], "", "0"), [])
+               finally:
+                       portage.util.noiselimit = 0
+                       playground.cleanup()