def test_registration_check_ok_reg(get_proxies, _init_session, _):
'''
Request completed OK, registered
Returns True
'''
config = Mock(legacy_upload=False, base_url='example.com')
conn = InsightsConnection(config)
res = requests.Response()
res._content = json.dumps({
"count":
1,
"page":
1,
"per_page":
50,
"results": [{
'insights_id': 'xxxxxx',
'id': 'yyyyyy'
}],
"total":
1
})
res.status_code = 200
conn.get = MagicMock(return_value=res)
assert conn.api_registration_check()
def test_registration_check_conn_error(get_proxies, _init_session, _):
'''
Connection error
Returns None
'''
config = Mock(legacy_upload=False, base_url='example.com')
conn = InsightsConnection(config)
conn.session.get.side_effect = requests.ConnectionError()
assert conn.api_registration_check() is None
def test_registration_check_legacy_registered(_, __):
'''
Ensure that connection function is called and data processed.
'''
config = Mock(base_url=None, legacy_upload=True)
conn = InsightsConnection(config)
conn.api_registration_check = Mock(return_value=True)
check = registration_check(conn)
assert isinstance(check, dict)
assert check['status'] is True
def test_registration_check_registered(write_unregistered_file,
write_registered_file, _, __):
'''
Ensure that connection function is called and files are written.
'''
config = Mock(base_url=None, legacy_upload=False)
conn = InsightsConnection(config)
conn.api_registration_check = Mock(return_value=True)
assert registration_check(conn) is True
conn.api_registration_check.assert_called_once()
write_registered_file.assert_called_once()
write_unregistered_file.assert_not_called()
def test_registration_check_conn_error(test_connection, get_proxies,
_init_session, _):
'''
Can't connect, run connection test
Returns False
'''
config = Mock(legacy_upload=True, base_url='example.com')
conn = InsightsConnection(config)
conn.get = MagicMock()
conn.get.side_effect = requests.ConnectionError()
assert conn.api_registration_check() is False
test_connection.assert_called_once()
def test_registration_check_bad_res(get_proxies, _init_session, _):
'''
Failure HTTP response
Returns None
'''
config = Mock(legacy_upload=False, base_url='example.com')
conn = InsightsConnection(config)
res = requests.Response()
res._content = 'wakannai'
res.status_code = 500
conn.get = MagicMock(return_value=res)
assert conn.api_registration_check() is None
def test_registration_check_parse_error(get_proxies, _init_session, _):
'''
Can't parse response
Returns None
'''
config = Mock(legacy_upload=False, base_url='example.com')
conn = InsightsConnection(config)
res = requests.Response()
res._content = 'zSDFasfghsRGH'
res.status_code = 200
conn.get = MagicMock(return_value=res)
assert conn.api_registration_check() is None
def test_registration_check_ok_reg(get_proxies, _init_session, _):
'''
Request completed OK, registered
Returns True
'''
config = Mock(legacy_upload=True, base_url='example.com')
conn = InsightsConnection(config)
res = requests.Response()
res._content = json.dumps({'unregistered_at': None})
res.status_code = 200
conn.get = MagicMock(return_value=res)
assert conn.api_registration_check()
def test_registration_check_ok_reg_then_unreg(get_proxies, _init_session, _):
'''
Request completed OK, was once registered but has been unregistered
Returns the date it was unregistered
'''
config = Mock(legacy_upload=True, base_url='example.com')
conn = InsightsConnection(config)
res = requests.Response()
res._content = json.dumps({'unregistered_at': '2019-04-10'})
res.status_code = 200
conn.get = MagicMock(return_value=res)
assert conn.api_registration_check() == '2019-04-10'