def test_functionality(self): port = 8080 for fmt in SUPPORTED_FILE_FORMATS: sync_address = "127.0.0.1:" + str(port) + "/dummy.tar.gz" orig_tempdir = TemporaryDirectory() orig_path = os.path.join(orig_tempdir.name, "db") os.makedirs(orig_path) orig_db = PackageDB(orig_path, preferred_category_format=fmt) orig_db.add_category("app-test1") orig_db.add_category("app-test2") ebuild_data = { "test1": "tst1", "test2": "tst2", "test3": serializable_elist([ DeserializableClass("1", "2"), DeserializableClass("3", "4") ]) } common_data = { "common1": "cmn1", "common2": "cmn2", "common3": serializable_elist([ DeserializableClass("c1", "c2"), DeserializableClass("c3", "c4") ]) } 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.set_common_data("app-test1", common_data) full_data = dict(ebuild_data) full_data.update(common_data) orig_db.write() os.system("cd " + orig_tempdir.name + " && tar cvzf good.tar.gz db") os.system("echo invalid >> " + orig_tempdir.name + "/db/app-test1/packages." + fmt) os.system("cd " + orig_tempdir.name + " && tar cvzf dummy.tar.gz db") test_db = PackageDB(self.tempdir.name) self.assertRaises(SyncError, test_db.sync, sync_address) srv = Server(orig_tempdir.name, port=port) srv.start() try: self.assertRaises(IntegrityError, test_db.sync, sync_address) os.system("cd " + orig_tempdir.name + " && mv good.tar.gz dummy.tar.gz") test_db.sync(sync_address) finally: srv.shutdown() srv.join() test_db.read() self.assertEqual(orig_db.database, test_db.database) self.assertEqual(orig_db.get_common_data("app-test1"), test_db.get_common_data("app-test1")) self.assertEqual(orig_db.get_common_data("app-test2"), test_db.get_common_data("app-test2")) 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]), full_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) if package.category == "app-test1": self.assertEqual(data, full_data) else: self.assertEqual(data, ebuild_data) pkg_set.remove(package) self.assertTrue(not pkg_set) self.assertEqual(orig_db.database, test_db.database) port = port + 1
def test_functionality(self): port = 8080 for fmt in SUPPORTED_FILE_FORMATS: sync_address = "127.0.0.1:" + str(port) + "/dummy.tar.gz" orig_tempdir = TemporaryDirectory() orig_path = os.path.join(orig_tempdir.name, "db") os.makedirs(orig_path) orig_db = PackageDB(orig_path, preferred_category_format=fmt) orig_db.add_category("app-test1") orig_db.add_category("app-test2") ebuild_data = {"test1": "tst1", "test2": "tst2", "test3": serializable_elist([DeserializableClass("1", "2"), DeserializableClass("3", "4")])} common_data = {"common1": "cmn1", "common2": "cmn2", "common3": serializable_elist([DeserializableClass("c1", "c2"), DeserializableClass("c3", "c4")])} 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.set_common_data("app-test1", common_data) full_data = dict(ebuild_data) full_data.update(common_data) orig_db.write() os.system("cd " + orig_tempdir.name + " && tar cvzf good.tar.gz db") os.system("echo invalid >> " + orig_tempdir.name + "/db/app-test1/packages." + fmt) os.system("cd " + orig_tempdir.name + " && tar cvzf dummy.tar.gz db") test_db = PackageDB(self.tempdir.name) self.assertRaises(SyncError, test_db.sync, sync_address) srv = Server(orig_tempdir.name, port=port) srv.start() self.assertRaises(IntegrityError, test_db.sync, sync_address) os.system("cd " + orig_tempdir.name + " && mv good.tar.gz dummy.tar.gz") test_db.sync(sync_address) srv.shutdown() srv.join() test_db.read() self.assertEqual(orig_db.database, test_db.database) self.assertEqual(orig_db.get_common_data("app-test1"), test_db.get_common_data("app-test1")) self.assertEqual(orig_db.get_common_data("app-test2"), test_db.get_common_data("app-test2")) 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]), full_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) if package.category == "app-test1": self.assertEqual(data, full_data) else: self.assertEqual(data, ebuild_data) pkg_set.remove(package) self.assertTrue(not pkg_set) self.assertEqual(orig_db.database, test_db.database) port = port + 1