def test_check_connection_backoff_on_limit_reached(requests_mock, config):
"""Error once, check that we retry and not fail"""
responses = [
{
"json": {
"error": "limit reached"
},
"status_code": 429,
"headers": {
"Retry-After": "0"
}
},
{
"json": [],
"status_code": 200
},
]
requests_mock.register_uri("GET", "/properties/v2/contact/properties",
responses)
source = SourceHubspot()
alive, error = source.check_connection(logger=logger, config=config)
assert alive
assert not error
def test_check_connection_backoff_on_server_error(requests_mock, config):
"""Error once, check that we retry and not fail"""
responses = [
{
"json": {
"error": "something bad"
},
"status_code": 500
},
{
"json": [],
"status_code": 200
},
]
requests_mock.register_uri("GET", "/properties/v2/contact/properties",
responses)
source = SourceHubspot()
alive, error = source.check_connection(logger=logger, config=config)
assert alive
assert not error