def test_without_export_dir(self):
"""
Test that when there is no export directory in the config, the repo working directory is used
"""
# Setup
config = PluginCallConfiguration({}, self.repo_config)
# Test
result = export_utils.retrieve_repo_config(self.repo, config)
result_working_dir, result_date_filter = result
self.assertEqual(result_working_dir, os.path.join(self.repo.working_dir, 'relative/path'))
self.assertEqual({}, result_date_filter)
export_utils.get_repo_relative_url.assert_called_once_with(self.repo.id)
export_utils.create_date_range_filter.assert_called_once_with(config)
def test_with_export_dir(self):
"""
Test that when an export directory is in the configuration, it is used as the working directory
"""
# Setup
self.repo_config[constants.EXPORT_DIRECTORY_KEYWORD] = '/some/export/dir'
expected_working_dir = '/some/export/dir/relative/path'
config = PluginCallConfiguration({}, self.repo_config)
# Test
result = export_utils.retrieve_repo_config(self.repo, config)
result_working_dir, result_date_filter = result
self.assertEqual(result_working_dir, expected_working_dir)
self.assertEqual({}, result_date_filter)
export_utils.get_repo_relative_url.assert_called_once_with(self.repo.id)
export_utils.create_date_range_filter.assert_called_once_with(config)
def publish_repo(self, repo, publish_conduit, config):
"""
Export a yum repository to a given directory, or to ISO
:param repo: metadata describing the repository
:type repo: pulp.plugins.model.Repository
:param publish_conduit: provides access to relevant Pulp functionality
:type publish_conduit: pulp.plugins.conduits.repo_publish.RepoPublishConduit
:param config: plugin configuration
:type config: pulp.plugins.config.PluginConfiguration
:return: report describing the publish run
:rtype: pulp.plugins.model.PublishReport
"""
# First, validate the configuration because there may be override config options, and
# currently, validate_config is not called prior to publishing by the manager.
valid_config, msg = export_utils.validate_export_config(config)
if not valid_config:
raise PulpDataException(msg)
_logger.info('Starting export of [%s]' % repo.id)
progress_status = {
ids.TYPE_ID_RPM: {'state': constants.STATE_NOT_STARTED},
ids.TYPE_ID_ERRATA: {'state': constants.STATE_NOT_STARTED},
ids.TYPE_ID_DISTRO: {'state': constants.STATE_NOT_STARTED},
ids.TYPE_ID_PKG_CATEGORY: {'state': constants.STATE_NOT_STARTED},
ids.TYPE_ID_PKG_GROUP: {'state': constants.STATE_NOT_STARTED},
'metadata': {'state': constants.STATE_NOT_STARTED},
'isos': {'state': constants.STATE_NOT_STARTED},
'publish_http': {'state': constants.STATE_NOT_STARTED},
'publish_https': {'state': constants.STATE_NOT_STARTED},
}
def progress_callback(type_id, status):
progress_status[type_id] = status
publish_conduit.set_progress(progress_status)
# Retrieve a config tuple and unpack it for use
config_settings = export_utils.retrieve_repo_config(repo, config)
self.working_dir, self.date_filter = config_settings
# Before starting, clean out the working directory. Done to remove last published ISOs
shutil.rmtree(repo.working_dir, ignore_errors=True)
os.makedirs(repo.working_dir)
# If a date filter is not present, do a complete export. If it is, do an incremental export.
if self.date_filter:
result = export_utils.export_incremental_content(self.working_dir, publish_conduit,
self.date_filter, progress_callback)
else:
result = export_utils.export_complete_repo(repo.id, self.working_dir, publish_conduit,
config, progress_callback)
self.summary = result[0]
self.details = result[1]
if not config.get(constants.EXPORT_DIRECTORY_KEYWORD):
util.generate_listing_files(repo.working_dir, self.working_dir)
# build iso and publish via HTTPS
self._publish_isos(repo, config, progress_callback)
else:
export_dir = config.get(constants.EXPORT_DIRECTORY_KEYWORD)
util.generate_listing_files(export_dir, self.working_dir)
if len(self.details['errors']) != 0:
return publish_conduit.build_failure_report(self.summary, self.details)
return publish_conduit.build_success_report(self.summary, self.details)
