def test_legacy_register(handle_registration):
'''
handle_unregistration called when legacy upload
'''
config = InsightsConfig(legacy_upload=True)
client = InsightsClient(config)
client.register()
handle_registration.assert_called_once()
def test_register_upload(handle_registration):
'''
handle_registration called when upload
'''
config = InsightsConfig(legacy_upload=False)
client = InsightsClient(config)
client.register()
handle_registration.assert_called_once()
def test_reg_check_registered_unreachable():
# register the machine first
config = InsightsConfig(register=True)
client = InsightsClient(config)
client.connection = FakeConnection(registered=None)
client.session = True
assert client.register() is True
# reset config and try to check registration
config.register = False
client.connection = FakeConnection(registered=False)
assert client.get_registation_status()['unreachable'] is True
assert client.register() is None
for r in constants.registered_files:
assert os.path.isfile(r) is True
for u in constants.unregistered_files:
assert os.path.isfile(u) is False
def test_platform_register_skip(handle_registration):
'''
handle_registration not called when platform upload
'''
config = InsightsConfig(legacy_upload=False)
client = InsightsClient(config)
assert client.register() # short circuits to True
handle_registration.assert_not_called()
def test_register():
config = InsightsConfig(register=True)
client = InsightsClient(config)
client.connection = FakeConnection()
client.session = True
assert client.register() is True
for r in constants.registered_files:
assert os.path.isfile(r) is True
for u in constants.unregistered_files:
assert os.path.isfile(u) is False
def test_reg_check_unregistered():
# unregister the machine first
config = InsightsConfig()
client = InsightsClient(config)
client.connection = FakeConnection(registered='unregistered')
client.session = True
# test function and integration in .register()
assert client.get_registation_status()['status'] is False
assert client.register() is False
for r in constants.registered_files:
assert os.path.isfile(r) is False
for u in constants.unregistered_files:
assert os.path.isfile(u) is True
def test_force_reregister():
config = InsightsConfig(reregister=True)
client = InsightsClient(config)
client.connection = FakeConnection(registered=None)
client.session = True
# initialize comparisons
old_machine_id = None
new_machine_id = None
# register first
assert client.register() is True
for r in constants.registered_files:
assert os.path.isfile(r) is True
# get modified time of .registered to ensure it's regenerated
old_reg_file1_ts = os.path.getmtime(constants.registered_files[0])
old_reg_file2_ts = os.path.getmtime(constants.registered_files[1])
old_machine_id = generate_machine_id()
# wait to allow for timestamp difference
time.sleep(3)
# reregister with new machine-id
client.connection = FakeConnection(registered=True)
config.reregister = True
assert client.register() is True
new_machine_id = generate_machine_id()
new_reg_file1_ts = os.path.getmtime(constants.registered_files[0])
new_reg_file2_ts = os.path.getmtime(constants.registered_files[1])
assert old_machine_id != new_machine_id
assert old_reg_file1_ts != new_reg_file1_ts
assert old_reg_file2_ts != new_reg_file2_ts
