def test_remove_missing_drpms(self, mock_remove, mock_get_remote_units): purge.remove_missing_drpms(self.metadata_files, self.conduit) mock_get_remote_units.assert_called_once_with(ANY, presto.PACKAGE_TAG, presto.process_package_element) mock_remove.assert_called_once_with(self.conduit, models.DRPM, mock_get_remote_units.return_value)
def test_remove_missing_drpms(self, mock_remove, mock_get_remote_units): """ Test that the purge makes the appropriate calls and that it calls for both of the prestodelta files """ purge.remove_missing_drpms(self.metadata_files, self.conduit) mock_get_remote_units.assert_called_with(ANY, presto.PACKAGE_TAG, presto.process_package_element) self.assertEquals(2, mock_get_remote_units.call_count) mock_remove.assert_called_once_with(self.conduit, models.DRPM, set())