class TestPrimaryDatabaseMD(TestPrimaryMD):
"""Test PrimaryDatabaseMD class."""
def setUp(self):
"""Setup example primary_db file."""
self.primary_db = PrimaryDatabaseMD("test_data/repodata/primary_db.sqlite")
def test_invalid_file(self):
"""Test case when file doesn't exist or is invalid."""
with self.assertRaises(OperationalError):
PrimaryDatabaseMD("/file/does/not/exist")
PrimaryDatabaseMD("/dev/null")
def test_packages(self):
"""Test parsed package metadata fields and counts."""
self.assertGreater(self.primary_db.get_package_count(), 0)
packages = self.primary_db.list_packages()
# Test fields of packages in list
for pkg in packages:
self._test_package(pkg)
class TestPrimaryDatabaseMD(unittest.TestCase):
def setUp(self):
"""Setup example primary_db file."""
self.primary_db = PrimaryDatabaseMD("test_data/repodata/primary_db.sqlite")
def _test_package(self, pkg):
intended_fields = ["name", "epoch", "ver", "rel", "arch", "checksum_type", "checksum"]
actual_fields = pkg.keys()
for field in intended_fields:
self.assertTrue(field in actual_fields)
for field in actual_fields:
self.assertTrue(field in intended_fields)
def test_invalid_file(self):
with self.assertRaises(OperationalError):
PrimaryDatabaseMD("/file/does/not/exist")
PrimaryDatabaseMD("/dev/null")
def test_packages(self):
self.assertGreater(self.primary_db.get_package_count(), 0)
packages = self.primary_db.list_packages()
# Test fields of packages in list
for pkg in packages:
self._test_package(pkg)