1 files changed, 70 insertions, 0 deletions
diff --git a/tests/test_policy.py b/tests/test_policy.py
new file mode 100644
index 0000000..ad0032c
--- /dev/null
+++ b/tests/test_policy.py
@@ -0,0 +1,70 @@
+#!/usr/bin/python3
+#
+# Copyright (C) 2012 Michael Vogt <mvo@ubuntu.com>
+#
+# Copying and distribution of this file, with or without modification,
+# are permitted in any medium without royalty provided the copyright
+# notice and this notice are preserved.
+
+import unittest
+
+import apt_pkg
+import testcommon
+
+import apt
+
+
+class TestAptPolicy(testcommon.TestCase):
+    def test_apt_policy_lowlevel(self):
+        return  # TODO: Make tests independent of system state
+        # get a policy
+        cache = apt.Cache()
+        policy = cache._depcache.policy
+        self.assertNotEqual(policy, None)
+        # basic tests
+        pkg = cache["apt"]
+        self.assertEqual(policy.get_priority(pkg._pkg), 0)
+        # get priority for all pkgfiles
+        for ver in pkg.versions:
+            lowlevel_ver = ver._cand
+            for pkgfile, i in lowlevel_ver.file_list:
+                # print pkgfile, i, policy.get_priority(pkgfile)
+                self.assertTrue(policy.get_priority(pkgfile) >= 1)
+                self.assertTrue(policy.get_priority(pkgfile) < 1001)
+
+    def test_apt_policy_lowlevel_files(self):
+        cache = apt_pkg.Cache()
+        depcache = apt_pkg.DepCache(cache)
+        policy = cache.policy
+        dpolicy = depcache.policy
+        self.assertNotEqual(policy, None)
+        self.assertNotEqual(dpolicy, None)
+
+        for f in cache.file_list:
+            policy.get_priority(f)
+            dpolicy.get_priority(f)
+
+    def test_apt_policy_lowlevel_versions(self):
+        cache = apt_pkg.Cache()
+        depcache = apt_pkg.DepCache(cache)
+        policy = cache.policy
+        dpolicy = depcache.policy
+        self.assertNotEqual(policy, None)
+        self.assertNotEqual(dpolicy, None)
+
+        for pkg in cache.packages:
+            for ver in pkg.version_list:
+                policy.get_priority(ver)
+                dpolicy.get_priority(ver)
+
+    def test_apt_policy_highlevel(self):
+        return  # TODO: Make tests independent of system state
+        cache = apt.Cache()
+        pkg = cache["apt"]
+        self.assertTrue(
+            pkg.candidate.policy_priority > 1 and pkg.candidate.policy_priority < 1001
+        )
+
+
+if __name__ == "__main__":
+    unittest.main()