def test_registration_check_called_for_support(registration_check,
InsightsConnection):
'''
Check registration when doing --support
'''
support = InsightsSupport(InsightsConfig(offline=False))
support.collect_support_info()
registration_check.assert_called_once()
def pre_update(client, config):
if config.version:
logger.info(constants.version)
sys.exit(constants.sig_kill_ok)
# validate the remove file
if config.validate:
if validate_remove_file(config.remove_file):
sys.exit(constants.sig_kill_ok)
else:
sys.exit(constants.sig_kill_bad)
# handle cron stuff
if config.enable_schedule:
# enable automatic scheduling
logger.debug('Updating config...')
updated = get_scheduler(config).set_daily()
if updated:
logger.info('Automatic scheduling for Insights has been enabled.')
sys.exit(constants.sig_kill_ok)
if config.disable_schedule:
# disable automatic schedling
updated = get_scheduler(config).remove_scheduling()
if updated:
logger.info('Automatic scheduling for Insights has been disabled.')
if not config.register:
sys.exit(constants.sig_kill_ok)
# delete someday
if config.analyze_container:
logger.debug('Not scanning host.')
logger.debug('Scanning image ID, tar file, or mountpoint.')
# test the insights connection
if config.test_connection:
logger.info("Running Connection Tests...")
rc = client.test_connection()
if rc == 0:
sys.exit(constants.sig_kill_ok)
else:
sys.exit(constants.sig_kill_bad)
if config.support:
support = InsightsSupport(config)
support.collect_support_info()
sys.exit(constants.sig_kill_ok)
if config.diagnosis:
remediation_id = None
if config.diagnosis is not True:
remediation_id = config.diagnosis
resp = client.get_diagnosis(remediation_id)
if not resp:
sys.exit(constants.sig_kill_bad)
print(json.dumps(resp))
sys.exit(constants.sig_kill_ok)
def test_collect_support_info_support_diag_dump(support_diag_dump, subprocess):
"""
collect_suppport_info_method callse _support_diag_dump method.
"""
config = Mock()
support = InsightsSupport(config)
support.collect_support_info()
support_diag_dump.assert_called_once_with()
def test_support_diag_dump_offline(test_connection, registration_check,
InsightsConnection):
'''
Check registration when doing --support
'''
support = InsightsSupport(InsightsConfig(offline=True))
support.collect_support_info()
InsightsConnection.assert_not_called()
registration_check.assert_not_called()
test_connection.assert_not_called()
def test_skip_registration_check_legacy_upload_off(registration_check,
InsightsConnection):
'''
Don't check registration when legacy_upload=False
'''
config = InsightsConfig(legacy_upload=False)
support = InsightsSupport(config)
support.collect_support_info()
registration_check.assert_not_called()
def test_registration_check_legacy_upload_on(registration_check,
InsightsConnection):
'''
Check registration when legacy_upload=True
'''
config = InsightsConfig(legacy_upload=True)
support = InsightsSupport(config)
support.collect_support_info()
registration_check.assert_called_once()
def test_support_diag_dump_insights_connection(insights_connection,
registration_check, isfile,
popen, statvfs):
"""
_support_diag_dump method instantiates InsightsConnection with InsightsConfig.
"""
config = Mock(**{"return_value.auto_config": "", "proxy": ""})
support = InsightsSupport(config)
support._support_diag_dump()
insights_connection.assert_called_once_with(config)
def pre_update():
if config['version']:
logger.info(constants.version)
sys.exit(constants.sig_kill_ok)
# validate the remove file
if config['validate']:
if validate_remove_file():
sys.exit(constants.sig_kill_ok)
else:
sys.exit(constants.sig_kill_bad)
# handle cron stuff
if config['enable_schedule'] and config['disable_schedule']:
logger.error(
'Conflicting options: --enable-schedule and --disable-schedule')
sys.exit(constants.sig_kill_bad)
if config['enable_schedule']:
# enable automatic scheduling
logger.debug('Updating config...')
updated = get_scheduler().set_daily()
if updated:
logger.info('Automatic scheduling for Insights has been enabled.')
sys.exit(constants.sig_kill_ok)
if config['disable_schedule']:
# disable automatic schedling
updated = get_scheduler().remove_scheduling()
if updated:
logger.info('Automatic scheduling for Insights has been disabled.')
if not config['register']:
sys.exit(constants.sig_kill_ok)
if config['container_mode']:
logger.debug('Not scanning host.')
logger.debug('Scanning image ID, tar file, or mountpoint.')
# test the insights connection
if config['test_connection']:
logger.info("Running Connection Tests...")
pconn = client.get_connection()
rc = pconn.test_connection()
if rc == 0:
sys.exit(constants.sig_kill_ok)
else:
sys.exit(constants.sig_kill_bad)
if config['support']:
support = InsightsSupport()
support.collect_support_info()
sys.exit(constants.sig_kill_ok)
def pre_update(client, config):
if config.version:
logger.info(constants.version)
sys.exit(constants.sig_kill_ok)
# validate the remove file
if config.validate:
try:
validate_remove_file(config)
sys.exit(constants.sig_kill_ok)
except RuntimeError as e:
logger.error(e)
sys.exit(constants.sig_kill_bad)
# handle cron stuff
if config.enable_schedule:
# enable automatic scheduling
logger.debug('Updating config...')
updated = get_scheduler(config).set_daily()
if updated:
logger.info('Automatic scheduling for Insights has been enabled.')
sys.exit(constants.sig_kill_ok)
if config.disable_schedule:
# disable automatic schedling
updated = get_scheduler(config).remove_scheduling()
if updated:
logger.info('Automatic scheduling for Insights has been disabled.')
if not config.register:
sys.exit(constants.sig_kill_ok)
# test the insights connection
if config.test_connection:
logger.info("Running Connection Tests...")
rc = client.test_connection()
if rc == 0:
sys.exit(constants.sig_kill_ok)
else:
sys.exit(constants.sig_kill_bad)
if config.support:
support = InsightsSupport(config)
support.collect_support_info()
sys.exit(constants.sig_kill_ok)
if config.diagnosis:
remediation_id = None
if config.diagnosis is not True:
remediation_id = config.diagnosis
resp = client.get_diagnosis(remediation_id)
if not resp:
sys.exit(constants.sig_kill_bad)
print(json.dumps(resp))
sys.exit(constants.sig_kill_ok)
if config.checkin:
checkin_success = client.checkin()
if checkin_success:
sys.exit(constants.sig_kill_ok)
else:
sys.exit(constants.sig_kill_bad)
