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
