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