diff --git a/tests/test_ip_info.py b/tests/test_ip_info.py index a1a8dcd..32906bf 100644 --- a/tests/test_ip_info.py +++ b/tests/test_ip_info.py @@ -18,7 +18,7 @@ def test_get_ip_information() -> None: assert ip_info.get("status") == "success" and ip_info.get("query") == test_query -def test_get_ip_information_broken_api() -> None: +def test_get_ip_information_broken_api_response() -> None: """TEST: ensure that None is returned if the IP API response is broken.""" test_query = "1.2.3.4" with requests_mock.Mocker() as mocker: @@ -27,6 +27,15 @@ def test_get_ip_information_broken_api() -> None: assert not resp +def test_get_ip_information_bad_response() -> None: + """TEST: ensure that None is returned if the IP API returns code 404.""" + test_query = "1.2.3.4" + with requests_mock.Mocker() as mocker: + mocker.get(f"http://ip-api.com/json/{test_query}", status_code=404) + resp = get_ip_information(test_query) + assert not resp + + def test_get_autonomous_system_number() -> None: """TEST: ensure that AS information is parsed into AS number correctly.""" as_info = "AS5089 Virgin Media Limited" @@ -41,7 +50,7 @@ def test_get_prefix_information() -> None: assert "144.254.0.0/16" in prefixes -def test_get_prefix_information_broken_api() -> None: +def test_get_prefix_information_broken_api_response() -> None: """TEST: ensure that None is returned if the prefix API response is broken.""" autonomous_system = "AS109" with requests_mock.Mocker() as mocker: @@ -51,3 +60,15 @@ def test_get_prefix_information_broken_api() -> None: ) resp = get_prefix_information(autonomous_system) assert not resp + + +def test_get_prefix_information_bad_response() -> None: + """TEST: ensure that None is returned if the prefix API returns code 404.""" + autonomous_system = "AS109" + with requests_mock.Mocker() as mocker: + mocker.get( + f"https://api.hackertarget.com/aslookup/?q={str(autonomous_system)}", + status_code=404, + ) + resp = get_prefix_information(autonomous_system) + assert not resp