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'