tests/test_PackageDB: test database functionality

1 files changed, 34 insertions, 6 deletions

diff --git a/tests/test_PackageDB.py b/tests/test_PackageDB.py
index 87cf3ea..373fa4b 100644
--- a/tests/test_PackageDB.py
+++ b/tests/test_PackageDB.py
@@ -16,7 +16,7 @@ import time
 import unittest
 
 from g_sorcery.compatibility import TemporaryDirectory
-from g_sorcery.exceptions import SyncError
+from g_sorcery.exceptions import IntegrityError, InvalidKeyError, SyncError
 from g_sorcery.g_collections import Package
 from g_sorcery.package_db import PackageDB
 
@@ -30,7 +30,7 @@ class TestDB(PackageDB):
 
 
 class TestPackageDB(BaseTest):
-    
+
     def test_functionality(self):
         orig_tempdir = TemporaryDirectory()
         orig_path = os.path.join(orig_tempdir.name, "db")
@@ -39,11 +39,14 @@ class TestPackageDB(BaseTest):
         orig_db.add_category("app-test1")
         orig_db.add_category("app-test2")
         ebuild_data = {"test1": "test1", "test2": "test2"}
-        orig_db.add_package(Package("app-test1", "test", "1"), ebuild_data)
-        orig_db.add_package(Package("app-test1", "test", "2"))
-        orig_db.add_package(Package("app-test1", "test1", "1"), ebuild_data)
-        orig_db.add_package(Package("app-test2", "test2", "1"), ebuild_data)
+        packages = [Package("app-test1", "test", "1"), Package("app-test1", "test", "2"),
+                    Package("app-test1", "test1", "1"), Package("app-test2", "test2", "1")]
+        for package in packages:
+            orig_db.add_package(package, ebuild_data)
+
         orig_db.write_and_manifest()
+        os.system("cd " + orig_tempdir.name + " && tar cvzf good.tar.gz db")
+        os.system("echo invalid >> " + orig_tempdir.name + "/db/app-test1/packages.json")
         os.system("cd " + orig_tempdir.name + " && tar cvzf dummy.tar.gz db")
 
         test_db = TestDB(self.tempdir.name)
@@ -51,9 +54,34 @@ class TestPackageDB(BaseTest):
 
         srv = Server(orig_tempdir.name)
         srv.start()
+        self.assertRaises(IntegrityError, test_db.sync, "127.0.0.1:8080")
+        os.system("cd " + orig_tempdir.name + " && mv good.tar.gz dummy.tar.gz")
         test_db.sync("127.0.0.1:8080")
         srv.shutdown()
         srv.join()
+        test_db.read()
+        self.assertEqual(orig_db.database, test_db.database)
+        self.assertEqual(set(test_db.list_categories()), set(["app-test1", "app-test2"]))
+        self.assertTrue(test_db.in_category("app-test1", "test"))
+        self.assertFalse(test_db.in_category("app-test2", "test"))
+        self.assertRaises(InvalidKeyError, test_db.in_category, "app-test3", "test")
+        self.assertEqual(set(test_db.list_package_names("app-test1")), set(['test', 'test1']))
+        self.assertEqual(set(test_db.list_catpkg_names()),set(['app-test1/test', 'app-test1/test1', 'app-test2/test2']))
+        self.assertRaises(InvalidKeyError, test_db.list_package_versions, "invalid", "test")
+        self.assertRaises(InvalidKeyError, test_db.list_package_versions, "app-test1", "invalid")
+        self.assertEqual(set(test_db.list_package_versions("app-test1", "test")), set(['1', '2']))
+        self.assertEqual(set(test_db.list_all_packages()), set(packages))
+        self.assertEqual(test_db.get_package_description(packages[0]), ebuild_data)
+        self.assertRaises(KeyError, test_db.get_package_description, Package("invalid", "invalid", "1"))
+        self.assertEqual(test_db.get_max_version("app-test1", "test"), "2")
+        self.assertEqual(test_db.get_max_version("app-test1", "test1"), "1")
+        self.assertRaises(InvalidKeyError, test_db.get_max_version, "invalid", "invalid")
+        pkg_set = set(packages)
+        for package, data in test_db:
+            self.assertTrue(package in pkg_set)
+            self.assertEqual(data, ebuild_data)
+            pkg_set.remove(package)
+        self.assertTrue(not pkg_set)
         
 
 def suite():