bites 111

Author: [email protected]

.gitignore

@@ -49,3 +49,4 @@
 /191/README.md
 /113/README.md
 /119/README.md
+/111/README.md

111/ipinfo.py

@@ -0,0 +1,14 @@
+import json
+
+import requests
+
+IPINFO_URL = 'http://ipinfo.io/{ip}/json'
+
+
+def get_ip_country(ip_address):
+    """Receives ip address string, use IPINFO_URL to get geo data,
+       parse the json response returning the country code of the IP"""
+    r=requests.get(IPINFO_URL[0:16],ip_address)
+    data=r.text
+    jsondata=json.loads(data)
+    return jsondata['country']

111/test_ipinfo.py

@@ -0,0 +1,34 @@
+import json
+from unittest.mock import patch, Mock
+
+import requests
+
+from ipinfo import get_ip_country
+
+
+@patch.object(requests, 'get')
+def test_ipinfo_mexican_ip(mockget):
+    # hardcoding a requests response
+    content = (b'{\n  "ip": "187.190.38.36",\n  "hostname": "domain.net",\n'
+               b'  "city": "Mexico City",\n  "region": "Mexico City",\n  '
+               b'"country": "MX",\n ' b'"loc": "11.0000,-00.1111",\n  '
+               b'"postal": "12345",\n  "org": "some org"\n}')
+    mockget.return_value = Mock(content=content,
+                                text=content.decode("utf-8"),
+                                json=lambda: json.loads(content),
+                                status_code=200)
+    assert get_ip_country('187.190.38.36') == 'MX'
+
+
+@patch.object(requests, 'get')
+def test_ipinfo_japan_ip(mockget):
+    # and another IP in Japan
+    content = (b'{\n  "ip": "185.161.200.10",\n  "city": "Tokyo",\n  '
+               b'"region": "Tokyo",\n ' b'"country": "JP",\n  "loc": '
+               b'"00.1111,11.0000",\n  "postal": "123-4567",\n  '
+               b'"org": "some other org"\n}')
+    mockget.return_value = Mock(content=content,
+                                text=content.decode("utf-8"),
+                                json=lambda: json.loads(content),
+                                status_code=200)
+    assert get_ip_country('185.161.200.10') == 'JP'