Added checks to ensure API responses are as expected (#7)

* Added checks to ensure API responses are as expected

* Add tests for broken IP/prefix API responses

* Add tests for bad IP/prefix status codes

app/ip_info.py

@@ -9,8 +9,13 @@ import requests
 def get_ip_information(ipv4_address: ipaddress.IPv4Address) -> dict:
     """Retrieves information about a given IPv4 address from IP-API.com."""
     api_endpoint = f"http://ip-api.com/json/{ipv4_address}"
-    resp = requests.get(api_endpoint).json()
-    return resp
+    try:
+        resp = requests.get(api_endpoint)
+        resp.raise_for_status()
+        ret = resp.json() if resp.json().get("status") == "success" else None
+    except (requests.exceptions.JSONDecodeError, requests.exceptions.HTTPError):
+        ret = None
+    return ret
 
 
 def get_autonomous_system_number(as_info: str) -> str:
@@ -22,7 +27,11 @@ def get_autonomous_system_number(as_info: str) -> str:
 def get_prefix_information(autonomous_system: int) -> list:
     """Retrieves prefix information about a given autonomous system."""
     api_endpoint = f"https://api.hackertarget.com/aslookup/?q={str(autonomous_system)}"
-    resp = requests.get(api_endpoint).text
-    prefixes = resp.split("\n")
+    try:
+        resp = requests.get(api_endpoint)
+        resp.raise_for_status()
+    except requests.exceptions.HTTPError:
+        return None
+    prefixes = resp.text.split("\n")
     prefixes.pop(0)
     return prefixes