diff --git a/sedr/util.py b/sedr/util.py index 2dc77ae..74a151a 100644 --- a/sedr/util.py +++ b/sedr/util.py @@ -131,7 +131,7 @@ def parse_locations(jsondata) -> None: # ) -def test_conformance_links(jsondata) -> tuple[bool, str]: +def test_conformance_links(jsondata: dict, timeout: int) -> tuple[bool, str]: """Test that all conformance links are valid and resolves.""" msg = "" for link in jsondata["conformsTo"]: @@ -143,21 +143,21 @@ def test_conformance_links(jsondata) -> tuple[bool, str]: # TODO: These links are part of the standard but doesn't work, so skipping for now. msg += f"test_conformance_links Link {link} doesn't resolv, but that is a known issue. " continue - resp = None + response = requests.Response() try: - resp = requests.head(url=link, timeout=10) + response = requests.head(url=link, timeout=timeout) except requests.exceptions.MissingSchema as error: msg += f"test_conformance_links Link <{link}> from /conformance is malformed: {error}). " - if not resp.status_code < 400: - msg += f"test_conformance_links Link {link} from /conformance is broken (gives status code {resp.status_code}). " + if not response.status_code < 400: + msg += f"test_conformance_links Link {link} from /conformance is broken (gives status code {response.status_code}). " if msg: return False, msg return True, "" -def locate_openapi_url(url: str) -> str: +def locate_openapi_url(url: str, timeout: int) -> str: """Locate the OpenAPI URL based on main URL.""" - request = requests.get(url, timeout=10) + request = requests.get(url, timeout=timeout) # Json # See https://github.com/metno/sedr/issues/6