def test_iso_idempotency(self):
# Setup
report = UpgradeStepReport()
units._initialize_content_types(self.tmp_test_db.database)
units._isos(self.v1_test_db.database, self.tmp_test_db.database, report)
# Test
success = units._isos(self.v1_test_db.database, self.tmp_test_db.database, report)
# Verify
self.assertTrue(success)
v1_count = self.v1_test_db.database.file.find().count()
v2_count = self.tmp_test_db.database.units_iso.find().count()
self.assertEqual(v1_count, v2_count)
def test_iso(self):
# Test
report = UpgradeStepReport()
success = units._isos(self.v1_test_db.database, self.tmp_test_db.database, report)
# Verify
self.assertTrue(success)
# Verify - Units
v1_files_coll = self.v1_test_db.database.file
v2_iso_coll = self.tmp_test_db.database.units_iso
v1_files = v1_files_coll.find()
for v1_file in v1_files:
v2_query = {'name' : v1_file['filename']}
v2_iso = v2_iso_coll.find_one(v2_query)
self.assertTrue(v2_iso is not None)
self.assertEqual(v2_iso['name'], v1_file['filename'])
self.assertEqual(v2_iso['checksum'], v1_file['checksum'].values()[0])
self.assertEqual(v2_iso['size'], v1_file['size'])
# Verify - Associations
v1_repo_coll = self.v1_test_db.database.repos
v1_file_coll = self.v1_test_db.database.file
v2_ass_coll = self.tmp_test_db.database.repo_content_units
v1_repos = v1_repo_coll.find({'content_types' : 'iso'})
for v1_repo in v1_repos:
for v1_file_id in v1_repo['files']:
v1_file = v1_file_coll.find_one({'_id' : v1_file_id})
v2_iso = v2_iso_coll.find_one({'name' : v1_file['filename']})
spec = {'unit_id' : v2_iso['_id'],
'unit_type_id' : 'iso',
'repo_id' : v1_repo['id']}
association = v2_ass_coll.find_one(spec)
self.assertTrue(association is not None)
def test_iso(self):
# Test
report = UpgradeStepReport()
success = units._isos(self.v1_test_db.database, self.tmp_test_db.database, report)
# Verify
self.assertTrue(success)
# Verify - Units
v1_files_coll = self.v1_test_db.database.file
v2_iso_coll = self.tmp_test_db.database.units_iso
v1_files = v1_files_coll.find()
for v1_file in v1_files:
v2_query = {"name": v1_file["filename"]}
v2_iso = v2_iso_coll.find_one(v2_query)
self.assertTrue(v2_iso is not None)
self.assertEqual(v2_iso["name"], v1_file["filename"])
self.assertEqual(v2_iso["checksum"], v1_file["checksum"].values()[0])
self.assertEqual(v2_iso["size"], v1_file["size"])
# Verify - Associations
v1_repo_coll = self.v1_test_db.database.repos
v1_file_coll = self.v1_test_db.database.file
v2_ass_coll = self.tmp_test_db.database.repo_content_units
v1_repos = v1_repo_coll.find({"content_types": "iso"})
for v1_repo in v1_repos:
for v1_file_id in v1_repo["files"]:
v1_file = v1_file_coll.find_one({"_id": v1_file_id})
v2_iso = v2_iso_coll.find_one({"name": v1_file["filename"]})
spec = {"unit_id": v2_iso["_id"], "unit_type_id": "iso", "repo_id": v1_repo["id"]}
association = v2_ass_coll.find_one(spec)
self.assertTrue(association is not None)
