def setUp(self): super(TestRemoveOldVersions, self).setUp() self.rpms = model_factory.rpm_models(3, True) self.rpms.extend(model_factory.rpm_models(2, False)) self.srpms = model_factory.srpm_models(3, True) self.srpms.extend(model_factory.srpm_models(2, False)) self.drpms = model_factory.drpm_models(3, True) self.drpms.extend(model_factory.drpm_models(2, False))
def test_srpms_check_all_and_associate_negative(self, mock_isfile, mock_save, mock_search_all_units): mock_search_all_units.return_value = [] mock_isfile.return_value = True units = model_factory.srpm_models(3) input_units = set([unit.as_named_tuple for unit in units]) result = check_all_and_associate(input_units, self.conduit) self.assertEqual(len(list(result)), 3)
def test_srpms_check_all_and_associate_positive(self, mock_isfile, mock_save, mock_search_all_units): units = model_factory.srpm_models(3) mock_search_all_units.return_value = units mock_isfile.return_value = True input_units = set([unit.as_named_tuple for unit in units]) for unit in units: unit.metadata['filename'] = 'test-filename' unit.storage_path = "existing_storage_path" result = check_all_and_associate(input_units, self.conduit) self.assertEqual(len(list(result)), 0)