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_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_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)