def setUp(self): super(TestRemoveOldVersions, self).setUp() self.rpms = model_factory.rpm_units(3, True) self.rpms.extend(model_factory.rpm_units(2, False)) self.srpms = model_factory.srpm_units(3, True) self.srpms.extend(model_factory.srpm_units(2, False)) self.drpms = model_factory.drpm_units(3, True) self.drpms.extend(model_factory.drpm_units(2, False))
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)