def test_yum_md_file(self, mock_copyfile):
mock_conduit = mock.MagicMock(spec_set=ImportUnitConduit('', '', '', ''))
model = model_factory.yum_md_file()
unit = Unit(model.TYPE, model.unit_key, model.metadata, '/foo/bar')
associate._associate_unit(self.repo, mock_conduit, unit)
expected_key = {'repo_id': self.repo.id, 'data_type': model.unit_key['data_type']}
self.assertEqual(mock_conduit.init_unit.call_args[0][0], model.TYPE)
self.assertEqual(mock_conduit.init_unit.call_args[0][1], expected_key)
self.assertEqual(mock_conduit.init_unit.call_args[0][2], model.metadata)
mock_conduit.save_unit.assert_called_once_with(mock_conduit.init_unit.return_value)
mock_copyfile.assert_called_once_with(unit.storage_path,
mock_conduit.init_unit.return_value.storage_path)
def test_distribution(self):
unit = model_factory.drpm_units(1)[0]
mock_conduit = mock.MagicMock(spec_set=ImportUnitConduit)
ret = associate._associate_unit('repo2', mock_conduit, unit)
self.assertTrue(ret is unit)
mock_conduit.associate_unit.assert_called_once_with(unit)
def test_distribution(self):
unit = model_factory.drpm_units(1)[0]
mock_conduit = mock.MagicMock(spec_set=ImportUnitConduit)
ret = associate._associate_unit('repo2', mock_conduit, unit)
self.assertTrue(ret is unit)
mock_conduit.associate_unit.assert_called_once_with(unit)
def test_rpm(self, mock_copyfile):
model = model_factory.rpm_models(1)[0]
unit = Unit(model.TYPE, model.unit_key, model.metadata, '/')
# passing "None" ensures that the importer isn't being called
ret = associate._associate_unit('', None, unit)
self.assertTrue(ret is unit)
self.assertEqual(mock_copyfile.call_count, 0)
def test_rpm(self, mock_copyfile):
model = model_factory.rpm_models(1)[0]
unit = Unit(model.TYPE, model.unit_key, model.metadata, '/')
# passing "None" ensures that the importer isn't being called
ret = associate._associate_unit('', None, unit)
self.assertTrue(ret is unit)
self.assertEqual(mock_copyfile.call_count, 0)
def test_group(self):
unit = model_factory.group_units(1)[0]
mock_conduit = mock.MagicMock()
ret = associate._associate_unit(self.repo, mock_conduit, unit)
saved_unit = mock_conduit.save_unit.call_args[0][0]
self.assertTrue(ret is mock_conduit.save_unit.return_value)
self.assertEqual(saved_unit.unit_key['repo_id'], self.repo.id)
self.assertEqual(saved_unit.unit_key['id'], unit.unit_key['id'])
def test_group(self):
unit = model_factory.group_units(1)[0]
mock_conduit = mock.MagicMock()
ret = associate._associate_unit(self.repo, mock_conduit, unit)
saved_unit = mock_conduit.save_unit.call_args[0][0]
self.assertTrue(ret is mock_conduit.save_unit.return_value)
self.assertEqual(saved_unit.unit_key['repo_id'], self.repo.id)
self.assertEqual(saved_unit.unit_key['id'], unit.unit_key['id'])
def test_yum_md_file(self, mock_copyfile):
mock_conduit = mock.MagicMock(
spec_set=ImportUnitConduit('', '', '', ''))
model = model_factory.yum_md_file()
unit = Unit(model.TYPE, model.unit_key, model.metadata, '/foo/bar')
associate._associate_unit(self.repo, mock_conduit, unit)
expected_key = {
'repo_id': self.repo.id,
'data_type': model.unit_key['data_type']
}
self.assertEqual(mock_conduit.init_unit.call_args[0][0], model.TYPE)
self.assertEqual(mock_conduit.init_unit.call_args[0][1], expected_key)
self.assertEqual(mock_conduit.init_unit.call_args[0][2],
model.metadata)
mock_conduit.save_unit.assert_called_once_with(
mock_conduit.init_unit.return_value)
mock_copyfile.assert_called_once_with(
unit.storage_path,
mock_conduit.init_unit.return_value.storage_path)
