Add tests.

Author: jhtimmins

setup.py

@@ -1,7 +1,7 @@
 from setuptools import setup
 
 setup(name='ipinfo_wrapper',
-      version='0.1.4',
+      version='0.1.5',
       description='Official Python library for IPInfo',
       url='https://github.com/ipinfo/python',
       author='James Timmins',
@@ -11,6 +11,7 @@
       install_requires=[
         'requests',
         'cachetools',
+        'pytest',
       ],
       include_package_data=True,
       zip_safe=False)

tests/__init__.py

@@ -0,0 +1,21 @@
+from ipinfo_wrapper.cache.default import DefaultCache
+
+
+def _get_new_cache():
+    maxsize = 4
+    ttl = 8
+    return DefaultCache(maxsize, ttl)
+
+
+def test_contains():
+    cache = _get_new_cache()
+    cache['foo'] = 'bar'
+
+    assert 'foo' in cache
+
+
+def test_get():
+    cache = _get_new_cache()
+    cache['foo'] = 'bar'
+
+    assert cache['foo'] == 'bar'

tests/default_cache_test.py

@@ -0,0 +1,27 @@
+import pytest
+from ipinfo_wrapper.details import Details
+
+
+def test_init():
+    data = {'foo': 'bar'}
+    details = Details(data)
+    assert details.details == data
+
+
+def test_getattr_success():
+    data = {'foo': 'bar'}
+    details = Details(data)
+    assert details.foo == data['foo']
+
+
+def test_getattr_fail():
+    data = {'foo': 'bar'}
+    details = Details(data)
+    with pytest.raises(Exception):
+        details.blah
+
+
+def test_all():
+    data = {'foo': 'bar', 'ham': 'eggs'}
+    details = Details(data)
+    assert details.all == data

tests/details_test.py

@@ -0,0 +1,43 @@
+from ipinfo_wrapper.cache.default import DefaultCache
+from ipinfo_wrapper.details import Details
+from ipinfo_wrapper.handler import Handler
+import ipaddress
+
+
+def test_init():
+    token = 'mytesttoken'
+    handler = Handler(token)
+    assert handler.access_token == token
+    assert isinstance(handler.cache, DefaultCache)
+    assert 'US' in handler.countries
+
+
+def test_headers():
+    token = 'mytesttoken'
+    handler = Handler(token)
+    headers = handler._get_headers()
+
+    assert 'user-agent' in headers
+    assert 'accept' in headers
+    assert 'authorization' in headers
+
+
+def test_get_details():
+    handler = Handler()
+    fake_details = {
+        'country': 'US',
+        'ip': '127.0.0.1',
+        'loc': '12.34,56.78'
+    }
+
+    handler._requestDetails = lambda x: fake_details
+
+    details = handler.getDetails(fake_details['ip'])
+    assert isinstance(details, Details)
+    assert details.country == fake_details['country']
+    assert details.country_name == 'United States'
+    assert details.ip == fake_details['ip']
+    assert isinstance(details.ip_address, ipaddress.IPv4Address)
+    assert details.loc == fake_details['loc']
+    assert details.longitude == '56.78'
+    assert details.latitude == '12.34'

tests/handler_test.py

@@ -0,0 +1,7 @@
+import ipinfo_wrapper
+from ipinfo_wrapper.handler import Handler
+
+
+def test_get_handler():
+    handler = ipinfo_wrapper.getHandler()
+    assert(isinstance(handler, Handler))

tests/init_test.py

@@ -0,0 +1,6 @@
+def f(x):
+    return x + 1
+
+
+def test_answer():
+    assert f(1) == 2