def test_exit_ok(insights_config, insights_client):
"""
Support collection replaces the normal client run.
"""
with raises(SystemExit) as exc_info:
collect_and_output()
assert exc_info.value.code == 0
def test_exit_ok(insights_config, insights_client):
"""
Support collection replaces the normal client run.
"""
insights_config.return_value.load_all.return_value.compliance = False
with raises(SystemExit) as exc_info:
collect_and_output()
assert exc_info.value.code == 0
def test_no_upload(insights_config, insights_client):
'''
Copy collected data to a specified file and do not upload.
'''
insights_config.return_value.load_all.return_value.no_upload = True
try:
collect_and_output()
except SystemExit:
pass
insights_client.upload.assert_not_called()
def test_output_dir(insights_config, insights_client):
'''
Copy collected data to a specified directory and do not upload.
'''
insights_config.return_value.load_all.return_value.no_upload = True
insights_config.return_value.load_all.return_value.output_dir = '/var/tmp/test'
try:
collect_and_output()
except SystemExit:
pass
insights_client.return_value.copy_to_output_dir.assert_called_with(
insights_client.return_value.collect.return_value)
def test_collect_and_output_payload_off(insights_config, insights_client):
"""
Collection is done and uploaded in normal operation (no payload)
Archive deleted after upload
"""
insights_config.return_value.load_all.return_value.payload = False
try:
collect_and_output()
except SystemExit:
pass
insights_client.return_value.collect.assert_called_once()
insights_client.return_value.upload.assert_called_with(
payload=insights_client.return_value.collect.return_value,
content_type='application/vnd.redhat.advisor.collection+tgz')
insights_client.return_value.delete_cached_branch_info.assert_called_once()
def test_collect_and_output_payload_on(insights_config, insights_client):
"""
Collection is not done when a payload is uploaded
Payload is sent to upload function
Archive is not deleted
"""
insights_config.return_value.load_all.return_value.payload = 'testp'
insights_config.return_value.load_all.return_value.content_type = 'testct'
try:
collect_and_output()
except SystemExit:
pass
insights_client.return_value.collect.assert_not_called()
insights_client.return_value.upload.assert_called_with(payload='testp', content_type='testct')
insights_client.return_value.delete_archive.assert_not_called()
