def setUp(self):
super(PuppetStatusRendererTests, self).setUp()
self.renderer = PuppetStatusRenderer(self.context)
self.sync_report = SyncProgressReport.from_progress_dict(
IMPORTER_REPORT)
self.publish_report = PublishProgressReport.from_progress_dict(
DISTRIBUTOR_REPORT)
def setUp(self):
super(PuppetStatusRendererTests, self).setUp()
self.renderer = PuppetStatusRenderer(self.context)
self.config['logging'] = {'filename': 'test-extension-status.log'}
self.sync_report = SyncProgressReport.from_progress_dict(
IMPORTER_REPORT)
self.publish_report = PublishProgressReport.from_progress_dict(
DISTRIBUTOR_REPORT)
def setUp(self):
super(PuppetStatusRendererTests, self).setUp()
self.renderer = PuppetStatusRenderer(self.context)
self.config['logging'] = {'filename' : 'test-extension-status.log'}
self.sync_report = SyncProgressReport.from_progress_dict(IMPORTER_REPORT)
self.publish_report = PublishProgressReport.from_progress_dict(DISTRIBUTOR_REPORT)
class PuppetStatusRendererTests(base_cli.ExtensionTests):
def setUp(self):
super(PuppetStatusRendererTests, self).setUp()
self.renderer = PuppetStatusRenderer(self.context)
self.sync_report = SyncProgressReport.from_progress_dict(
IMPORTER_REPORT)
self.publish_report = PublishProgressReport.from_progress_dict(
DISTRIBUTOR_REPORT)
def test_display_sync_metadata(self):
# Test
self.renderer._display_sync_metadata_step(self.sync_report)
# Verify
expected_tags = ['download-metadata', TAG_PROGRESS_BAR]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.sync_metadata_last_state,
constants.STATE_SUCCESS)
def test_display_sync_metadata_not_started(self):
# Setup
self.sync_report.metadata_state = constants.STATE_NOT_STARTED
# Test
self.renderer._display_sync_metadata_step(self.sync_report)
# Verify
self.assertEqual(0, len(self.prompt.get_write_tags()))
self.assertEqual(self.renderer.sync_metadata_last_state,
constants.STATE_NOT_STARTED)
def test_display_sync_metadata_failed(self):
# Setup
self.sync_report.metadata_state = constants.STATE_FAILED
# Test
self.renderer._display_sync_metadata_step(self.sync_report)
# Verify
expected_tags = [
'download-metadata', TAG_FAILURE, TAG_FAILURE, TAG_FAILURE
]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.sync_metadata_last_state,
constants.STATE_FAILED)
def test_display_sync_modules(self):
# Test
self.renderer._display_sync_modules_step(self.sync_report)
# Verify
expected_tags = ['downloading', TAG_PROGRESS_BAR]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.sync_modules_last_state,
constants.STATE_SUCCESS)
def test_display_sync_modules_not_started(self):
# Setup
self.sync_report.modules_state = constants.STATE_NOT_STARTED
# Test
self.renderer._display_sync_modules_step(self.sync_report)
# Verify
expected_tags = []
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.sync_modules_last_state,
constants.STATE_NOT_STARTED)
def test_display_sync_modules_failed(self):
# Setup
self.sync_report.modules_state = constants.STATE_FAILED
# Test
self.renderer._display_sync_modules_step(self.sync_report)
# Verify
expected_tags = ['downloading', TAG_FAILURE, TAG_FAILURE, TAG_FAILURE]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.sync_modules_last_state,
constants.STATE_FAILED)
def test_display_publish_modules(self):
# Test
self.renderer._display_publish_modules_step(self.publish_report)
# Verify
expected_tags = ['publishing', TAG_PROGRESS_BAR]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_modules_last_state,
constants.STATE_SUCCESS)
def test_display_publish_modules_not_started(self):
# Setup
self.publish_report.modules_state = constants.STATE_NOT_STARTED
# Test
self.renderer._display_publish_modules_step(self.publish_report)
# Verify
expected_tags = []
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_modules_last_state,
constants.STATE_NOT_STARTED)
def test_display_publish_modules_failed(self):
# Setup
self.publish_report.modules_state = constants.STATE_FAILED
# Test
self.renderer._display_publish_modules_step(self.publish_report)
# Verify
expected_tags = ['publishing', TAG_FAILURE, TAG_FAILURE, TAG_FAILURE]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_modules_last_state,
constants.STATE_FAILED)
def test_display_publish_metadata(self):
# Test
self.renderer._display_publish_metadata_step(self.publish_report)
# Verify
expected_tags = ['generating', TAG_SPINNER, 'completed']
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_metadata_last_state,
constants.STATE_SUCCESS)
def test_display_publish_metadata_not_started(self):
# Setup
self.publish_report.metadata_state = constants.STATE_NOT_STARTED
# Test
self.renderer._display_publish_metadata_step(self.publish_report)
# Verify
expected_tags = []
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_metadata_last_state,
constants.STATE_NOT_STARTED)
def test_display_publish_metadata_in_progress(self):
# Setup
self.renderer.publish_metadata_last_state = constants.STATE_RUNNING
self.publish_report.metadata_state = constants.STATE_RUNNING
# Test
self.renderer._display_publish_metadata_step(self.publish_report)
# Verify
expected_tags = [TAG_SPINNER]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_metadata_last_state,
constants.STATE_RUNNING)
def test_display_publish_metadata_complete(self):
# Setup
self.renderer.publish_metadata_last_state = constants.STATE_RUNNING
self.publish_report.metadata_state = constants.STATE_SUCCESS
# Test
self.renderer._display_publish_metadata_step(self.publish_report)
# Verify
expected_tags = [TAG_SPINNER, 'completed']
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_metadata_last_state,
constants.STATE_SUCCESS)
def test_display_publish_metadata_failed(self):
# Setup
self.renderer.publish_metadata_last_state = constants.STATE_RUNNING
self.publish_report.metadata_state = constants.STATE_FAILED
# Test
self.renderer._display_publish_metadata_step(self.publish_report)
# Verify
expected_tags = [TAG_SPINNER, TAG_FAILURE, TAG_FAILURE, TAG_FAILURE]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_metadata_last_state,
constants.STATE_FAILED)
def test_publsh_http_https(self):
# Test
self.renderer._display_publish_http_https_step(self.publish_report)
# Verify
expected_tags = ['http-completed', 'https-completed']
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_http_last_state,
constants.STATE_SUCCESS)
self.assertEqual(self.renderer.publish_https_last_state,
constants.STATE_SUCCESS)
def test_publish_http_https_not_started(self):
# Setup
self.publish_report.publish_http = constants.STATE_NOT_STARTED
self.publish_report.publish_https = constants.STATE_NOT_STARTED
# Test
self.renderer._display_publish_http_https_step(self.publish_report)
# Verify
expected_tags = []
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_http_last_state,
constants.STATE_NOT_STARTED)
self.assertEqual(self.renderer.publish_https_last_state,
constants.STATE_NOT_STARTED)
def test_publish_http_https_skipped(self):
# Setup
self.publish_report.publish_http = constants.STATE_SKIPPED
self.publish_report.publish_https = constants.STATE_SKIPPED
# Test
self.renderer._display_publish_http_https_step(self.publish_report)
# Verify
expected_tags = ['http-skipped', 'https-skipped']
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_http_last_state,
constants.STATE_SKIPPED)
self.assertEqual(self.renderer.publish_https_last_state,
constants.STATE_SKIPPED)
def test_publish_http_https_unknown(self):
# Setup
self.publish_report.publish_http = 'unknown'
self.publish_report.publish_https = 'unknown'
# Test
self.renderer._display_publish_http_https_step(self.publish_report)
# Verify
expected_tags = ['http-unknown', 'https-unknown']
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_http_last_state, 'unknown')
self.assertEqual(self.renderer.publish_https_last_state, 'unknown')
def test_render_module_errors(self):
# Setup
# Need a valid traceback instance to be formatted and this is the best
# I could come up with to make one :)
tb = None
try:
raise Exception()
except Exception:
tb = sys.exc_info()[2]
tb = traceback.extract_tb(tb)
individual_errors = []
for i in range(0, 10):
individual_errors.append({
'module': 'mod_%s' % i,
'author': 'some author',
'exception': 'e_%s' % i,
'traceback': tb,
})
# Test
self.renderer._render_module_errors(individual_errors)
# Verify
expected_tags = [TAG_FAILURE]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
# grab every printed line except the first message, the blank line after
# it, and the blank line at the end
printed_module_names = set(x.strip()
for x in self.prompt.output.lines[2:-1])
# verify that only module names were printed, and not tracebacks
self.assertEqual(
printed_module_names,
set([
'%s: %s' % (e['module'], e['exception'])
for e in individual_errors
]))
def test_display_report(self):
# Test
self.renderer.display_report(FULL_REPORT)
# Verify
expected_tags = [
'download-metadata', 'progress_bar', 'downloading', 'progress_bar',
'publishing', 'progress_bar', 'generating', 'spinner', 'completed',
'http-completed', 'https-completed'
]
tags = self.prompt.get_write_tags()
self.assertEqual(expected_tags, tags)
class PuppetStatusRendererTests(base_cli.ExtensionTests):
def setUp(self):
super(PuppetStatusRendererTests, self).setUp()
self.renderer = PuppetStatusRenderer(self.context)
self.config['logging'] = {'filename' : 'test-extension-status.log'}
self.sync_report = SyncProgressReport.from_progress_dict(IMPORTER_REPORT)
self.publish_report = PublishProgressReport.from_progress_dict(DISTRIBUTOR_REPORT)
def test_display_sync_metadata(self):
# Test
self.renderer._display_sync_metadata_step(self.sync_report)
# Verify
expected_tags = ['download-metadata', TAG_PROGRESS_BAR]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.sync_metadata_last_state, constants.STATE_SUCCESS)
def test_display_sync_metadata_not_started(self):
# Setup
self.sync_report.metadata_state = constants.STATE_NOT_STARTED
# Test
self.renderer._display_sync_metadata_step(self.sync_report)
# Verify
self.assertEqual(0, len(self.prompt.get_write_tags()))
self.assertEqual(self.renderer.sync_metadata_last_state, constants.STATE_NOT_STARTED)
def test_display_sync_metadata_failed(self):
# Setup
self.sync_report.metadata_state = constants.STATE_FAILED
# Test
self.renderer._display_sync_metadata_step(self.sync_report)
# Verify
expected_tags = ['download-metadata', TAG_FAILURE, TAG_FAILURE, TAG_FAILURE]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.sync_metadata_last_state, constants.STATE_FAILED)
def test_display_sync_modules(self):
# Test
self.renderer._display_sync_modules_step(self.sync_report)
# Verify
expected_tags = ['downloading', TAG_PROGRESS_BAR]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.sync_modules_last_state, constants.STATE_SUCCESS)
def test_display_sync_modules_not_started(self):
# Setup
self.sync_report.modules_state = constants.STATE_NOT_STARTED
# Test
self.renderer._display_sync_modules_step(self.sync_report)
# Verify
expected_tags = []
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.sync_modules_last_state, constants.STATE_NOT_STARTED)
def test_display_sync_modules_failed(self):
# Setup
self.sync_report.modules_state = constants.STATE_FAILED
# Test
self.renderer._display_sync_modules_step(self.sync_report)
# Verify
expected_tags = ['downloading', TAG_FAILURE, TAG_FAILURE, TAG_FAILURE]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.sync_modules_last_state, constants.STATE_FAILED)
def test_display_publish_modules(self):
# Test
self.renderer._display_publish_modules_step(self.publish_report)
# Verify
expected_tags = ['publishing', TAG_PROGRESS_BAR]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_modules_last_state, constants.STATE_SUCCESS)
def test_display_publish_modules_not_started(self):
# Setup
self.publish_report.modules_state = constants.STATE_NOT_STARTED
# Test
self.renderer._display_publish_modules_step(self.publish_report)
# Verify
expected_tags = []
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_modules_last_state, constants.STATE_NOT_STARTED)
def test_display_publish_modules_failed(self):
# Setup
self.publish_report.modules_state = constants.STATE_FAILED
# Test
self.renderer._display_publish_modules_step(self.publish_report)
# Verify
expected_tags = ['publishing', TAG_FAILURE, TAG_FAILURE, TAG_FAILURE]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_modules_last_state, constants.STATE_FAILED)
def test_display_publish_metadata(self):
# Test
self.renderer._display_publish_metadata_step(self.publish_report)
# Verify
expected_tags = ['generating', TAG_SPINNER, 'completed']
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_metadata_last_state, constants.STATE_SUCCESS)
def test_display_publish_metadata_not_started(self):
# Setup
self.publish_report.metadata_state = constants.STATE_NOT_STARTED
# Test
self.renderer._display_publish_metadata_step(self.publish_report)
# Verify
expected_tags = []
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_metadata_last_state, constants.STATE_NOT_STARTED)
def test_display_publish_metadata_in_progress(self):
# Setup
self.renderer.publish_metadata_last_state = constants.STATE_RUNNING
self.publish_report.metadata_state = constants.STATE_RUNNING
# Test
self.renderer._display_publish_metadata_step(self.publish_report)
# Verify
expected_tags = [TAG_SPINNER]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_metadata_last_state, constants.STATE_RUNNING)
def test_display_publish_metadata_complete(self):
# Setup
self.renderer.publish_metadata_last_state = constants.STATE_RUNNING
self.publish_report.metadata_state = constants.STATE_SUCCESS
# Test
self.renderer._display_publish_metadata_step(self.publish_report)
# Verify
expected_tags = [TAG_SPINNER, 'completed']
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_metadata_last_state, constants.STATE_SUCCESS)
def test_display_publish_metadata_failed(self):
# Setup
self.renderer.publish_metadata_last_state = constants.STATE_RUNNING
self.publish_report.metadata_state = constants.STATE_FAILED
# Test
self.renderer._display_publish_metadata_step(self.publish_report)
# Verify
expected_tags = [TAG_SPINNER, TAG_FAILURE, TAG_FAILURE, TAG_FAILURE]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_metadata_last_state, constants.STATE_FAILED)
def test_publsh_http_https(self):
# Test
self.renderer._display_publish_http_https_step(self.publish_report)
# Verify
expected_tags = ['http-completed', 'https-completed']
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_http_last_state, constants.STATE_SUCCESS)
self.assertEqual(self.renderer.publish_https_last_state, constants.STATE_SUCCESS)
def test_publish_http_https_not_started(self):
# Setup
self.publish_report.publish_http = constants.STATE_NOT_STARTED
self.publish_report.publish_https = constants.STATE_NOT_STARTED
# Test
self.renderer._display_publish_http_https_step(self.publish_report)
# Verify
expected_tags = []
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_http_last_state, constants.STATE_NOT_STARTED)
self.assertEqual(self.renderer.publish_https_last_state, constants.STATE_NOT_STARTED)
def test_publish_http_https_skipped(self):
# Setup
self.publish_report.publish_http = constants.STATE_SKIPPED
self.publish_report.publish_https = constants.STATE_SKIPPED
# Test
self.renderer._display_publish_http_https_step(self.publish_report)
# Verify
expected_tags = ['http-skipped', 'https-skipped']
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_http_last_state, constants.STATE_SKIPPED)
self.assertEqual(self.renderer.publish_https_last_state, constants.STATE_SKIPPED)
def test_publish_http_https_unknown(self):
# Setup
self.publish_report.publish_http = 'unknown'
self.publish_report.publish_https = 'unknown'
# Test
self.renderer._display_publish_http_https_step(self.publish_report)
# Verify
expected_tags = ['http-unknown', 'https-unknown']
self.assertEqual(expected_tags, self.prompt.get_write_tags())
self.assertEqual(self.renderer.publish_http_last_state, 'unknown')
self.assertEqual(self.renderer.publish_https_last_state, 'unknown')
def test_render_module_errors(self):
# Setup
# Need a valid traceback instance to be formatted and this is the best
# I could come up with to make one :)
tb = None
try:
raise Exception()
except Exception:
tb = sys.exc_info()[2]
tb = traceback.extract_tb(tb)
individual_errors = {}
for i in range(0, 10):
individual_errors['mod_%s' % i] = {
'exception' : 'e_%s' % i,
'traceback' : tb,
}
# Test
self.renderer._render_module_errors(individual_errors)
# Verify
expected_tags = [TAG_FAILURE]
self.assertEqual(expected_tags, self.prompt.get_write_tags())
# grab every printed line except the first message, the blank line after
# it, and the blank line at the end
printed_module_names = set(x.strip() for x in self.prompt.output.lines[2:-1])
# verify that only module names were printed, and not tracebacks
self.assertEqual(printed_module_names, set(individual_errors.keys()))
def test_display_report(self):
# Test
self.renderer.display_report(FULL_REPORT)
# Verify
expected_tags = ['download-metadata', 'progress_bar',
'downloading', 'progress_bar',
'publishing', 'progress_bar',
'generating', 'spinner',
'completed',
'http-completed', 'https-completed']
tags = self.prompt.get_write_tags()
self.assertEqual(expected_tags, tags)
def setUp(self):
super(PuppetStatusRendererTests, self).setUp()
self.renderer = PuppetStatusRenderer(self.context)
self.sync_report = SyncProgressReport.from_progress_dict(IMPORTER_REPORT)
self.publish_report = PublishProgressReport.from_progress_dict(DISTRIBUTOR_REPORT)
