def test_sanitizes_checksum_type(self, Distribution, RawConfigParser):
"""
Ensure the function properly sanitizes checksum types.
"""
parser = MagicMock()
parser.has_section.return_value = True
parser.items.return_value = [['path', 'sha:checksum']]
RawConfigParser.return_value = parser
model, files = DistSync.parse_treeinfo_file('/some/path')
self.assertEqual(files[0]['checksumtype'], 'sha1')
def test_rhel5_optional(self):
path = os.path.join(DATA_PATH, 'treeinfo-rhel5-no-optional-keys')
model, files = DistSync.parse_treeinfo_file(path)
self.assertTrue(isinstance(model, models.Distribution))
self.assertEqual(model.distribution_id, 'ks-Red Hat Enterprise Linux Server--5.9-x86_64')
self.assertEqual(len(files), 19)
for item in files:
self.assertTrue(item['relativepath'])
self.assertEquals('', model.variant)
self.assertEquals(None, model.packagedir)
def test_rhel5(self):
path = os.path.join(DATA_PATH, 'treeinfo-rhel5')
model, files = DistSync.parse_treeinfo_file(path)
self.assertTrue(isinstance(model, models.Distribution))
self.assertEqual(model.distribution_id, 'ks-Red Hat Enterprise Linux Server-foo-5.9-x86_64')
self.assertEqual(len(files), 19)
for item in files:
self.assertTrue(item['relativepath'])
self.assertEquals('foo', model.variant)
self.assertEquals('Server', model.packagedir)
self.assertEquals(1354213090.94, model.timestamp)
def setUp(self):
path = os.path.join(DATA_PATH, 'treeinfo-rhel5')
self.model1, files1 = DistSync.parse_treeinfo_file(path)
self.model2, files2 = DistSync.parse_treeinfo_file(path)
