def setUp(self):
super(TestRefreshContentSourcesCommand, self).setUp()
self.context = MagicMock(
config={'output': {
'poll_frequency_in_seconds': 1
}})
self.prompt = Mock()
self.context.prompt = self.prompt
self.renderer = PublishStepStatusRenderer(self.context)
self.command = RefreshContentSourcesCommand(self.context,
self.renderer)
step_details = {
'source_id': '1',
'succeeded': None,
'url': 'mock-url',
'added_count': '2',
'deleted_count': '0',
'errors': []
}
self.step = {
reporting_constants.PROGRESS_STEP_TYPE_KEY: u'foo_step',
reporting_constants.PROGRESS_STEP_UUID: u'abcde',
reporting_constants.PROGRESS_DESCRIPTION_KEY: u'foo description',
reporting_constants.PROGRESS_DETAILS_KEY: [step_details],
reporting_constants.PROGRESS_STATE_KEY:
reporting_constants.STATE_NOT_STARTED,
reporting_constants.PROGRESS_ITEMS_TOTAL_KEY: 1,
reporting_constants.PROGRESS_NUM_PROCESSED_KEY: 0,
reporting_constants.PROGRESS_ERROR_DETAILS_KEY: []
}
def setUp(self):
super(TestRefreshContentSourcesCommand, self).setUp()
self.context = MagicMock(config={"output": {"poll_frequency_in_seconds": 1}})
self.prompt = Mock()
self.context.prompt = self.prompt
self.renderer = PublishStepStatusRenderer(self.context)
self.command = RefreshContentSourcesCommand(self.context, self.renderer)
step_details = {
"source_id": "1",
"succeeded": None,
"url": "mock-url",
"added_count": "2",
"deleted_count": "0",
"errors": [],
}
self.step = {
reporting_constants.PROGRESS_STEP_TYPE_KEY: u"foo_step",
reporting_constants.PROGRESS_STEP_UUID: u"abcde",
reporting_constants.PROGRESS_DESCRIPTION_KEY: u"foo description",
reporting_constants.PROGRESS_DETAILS_KEY: [step_details],
reporting_constants.PROGRESS_STATE_KEY: reporting_constants.STATE_NOT_STARTED,
reporting_constants.PROGRESS_ITEMS_TOTAL_KEY: 1,
reporting_constants.PROGRESS_NUM_PROCESSED_KEY: 0,
reporting_constants.PROGRESS_ERROR_DETAILS_KEY: [],
}
class TestRefreshContentSourcesCommand(base.PulpClientTests):
def setUp(self):
super(TestRefreshContentSourcesCommand, self).setUp()
self.context = MagicMock(config={"output": {"poll_frequency_in_seconds": 1}})
self.prompt = Mock()
self.context.prompt = self.prompt
self.renderer = PublishStepStatusRenderer(self.context)
self.command = RefreshContentSourcesCommand(self.context, self.renderer)
step_details = {
"source_id": "1",
"succeeded": None,
"url": "mock-url",
"added_count": "2",
"deleted_count": "0",
"errors": [],
}
self.step = {
reporting_constants.PROGRESS_STEP_TYPE_KEY: u"foo_step",
reporting_constants.PROGRESS_STEP_UUID: u"abcde",
reporting_constants.PROGRESS_DESCRIPTION_KEY: u"foo description",
reporting_constants.PROGRESS_DETAILS_KEY: [step_details],
reporting_constants.PROGRESS_STATE_KEY: reporting_constants.STATE_NOT_STARTED,
reporting_constants.PROGRESS_ITEMS_TOTAL_KEY: 1,
reporting_constants.PROGRESS_NUM_PROCESSED_KEY: 0,
reporting_constants.PROGRESS_ERROR_DETAILS_KEY: [],
}
@patch("pulp.client.commands.polling.PollingCommand.poll")
def test_run_one(self, mock_poll):
"""
Test run() when there are no refresh content source tasks on the server.
"""
content_source_id = "test-content"
data = {options.OPTION_CONTENT_SOURCE_ID.keyword: content_source_id}
# No tasks are running
self.command.run(**data)
self.assertEqual(1, mock_poll.call_count)
@patch("pulp.client.commands.polling.PollingCommand.poll")
def test_run_all(self, mock_poll):
"""
Test run() when there are no refresh content source tasks on the server.
"""
data = {options.OPTION_CONTENT_SOURCE_ID.keyword: None}
# No tasks are running
self.command.run(**data)
self.assertEqual(1, mock_poll.call_count)
def test_progress(self):
"""
Test the progress() method with a progress_report.
"""
task = responses.Task({"progress_report": {"Refresh Content Sources": self.step}})
spinner = MagicMock()
self.renderer.display_report = MagicMock()
self.command.progress(task, spinner)
self.renderer.display_report.assert_called_once_with({"Refresh Content Sources": self.step})
class TestRefreshContentSourcesCommand(base.PulpClientTests):
def setUp(self):
super(TestRefreshContentSourcesCommand, self).setUp()
self.context = MagicMock(
config={'output': {
'poll_frequency_in_seconds': 1
}})
self.prompt = Mock()
self.context.prompt = self.prompt
self.renderer = PublishStepStatusRenderer(self.context)
self.command = RefreshContentSourcesCommand(self.context,
self.renderer)
step_details = {
'source_id': '1',
'succeeded': None,
'url': 'mock-url',
'added_count': '2',
'deleted_count': '0',
'errors': []
}
self.step = {
reporting_constants.PROGRESS_STEP_TYPE_KEY: u'foo_step',
reporting_constants.PROGRESS_STEP_UUID: u'abcde',
reporting_constants.PROGRESS_DESCRIPTION_KEY: u'foo description',
reporting_constants.PROGRESS_DETAILS_KEY: [step_details],
reporting_constants.PROGRESS_STATE_KEY:
reporting_constants.STATE_NOT_STARTED,
reporting_constants.PROGRESS_ITEMS_TOTAL_KEY: 1,
reporting_constants.PROGRESS_NUM_PROCESSED_KEY: 0,
reporting_constants.PROGRESS_ERROR_DETAILS_KEY: []
}
@patch('pulp.client.commands.polling.PollingCommand.poll')
def test_run_one(self, mock_poll):
"""
Test run() when there are no refresh content source tasks on the server.
"""
content_source_id = 'test-content'
data = {options.OPTION_CONTENT_SOURCE_ID.keyword: content_source_id}
# No tasks are running
self.command.run(**data)
self.assertEqual(1, mock_poll.call_count)
@patch('pulp.client.commands.polling.PollingCommand.poll')
def test_run_all(self, mock_poll):
"""
Test run() when there are no refresh content source tasks on the server.
"""
data = {options.OPTION_CONTENT_SOURCE_ID.keyword: None}
# No tasks are running
self.command.run(**data)
self.assertEqual(1, mock_poll.call_count)
def test_progress(self):
"""
Test the progress() method with a progress_report.
"""
task = responses.Task(
{'progress_report': {
'Refresh Content Sources': self.step
}})
spinner = MagicMock()
self.renderer.display_report = MagicMock()
self.command.progress(task, spinner)
self.renderer.display_report.assert_called_once_with(
{'Refresh Content Sources': self.step})
