Added test case.

Author: chrislim2888

IP2Location.py

@@ -3,6 +3,7 @@
 import socket
 import re
 import json
+import os
 
 if sys.version < '3':
     import urllib, httplib
@@ -127,6 +128,10 @@ def __init__(self, filename=None,mode='FILE_IO'):
             
         '''
         self.mode = mode
+        
+        if os.path.isfile(filename) == False:
+            raise ValueError("The database file does not seem to exist.")
+        
         if filename:
             self.open(filename)
 
@@ -272,8 +277,8 @@ def find(self, addr):
             addr: IPv4 or IPv6 address as a string
      
             Returns IP2LocationRecord or None if address not found in file
-			
-			This function will be obselete in future.
+            
+            This function will be obselete in future.
         '''
         return self._get_record(addr)
 

data/IP2LOCATION-LITE-DB1.BIN

@@ -7,10 +7,10 @@
 import os
 import IP2Location
 
-'''
+"""
     Cache the database into memory to accelerate lookup speed.
     WARNING: Please make sure your system have sufficient RAM to use this feature.
-'''
+"""
 # database = IP2Location.IP2Location(os.path.join("data", "IPV6-COUNTRY.BIN"), "SHARED_MEMORY")
 
 # Default file I/O lookup
@@ -42,8 +42,8 @@
 
 # Web Service
 ws = IP2Location.IP2LocationWebService("demo","WS24",True)
-rec = ws.lookup('8.8.8.8', ['continent', 'country', 'region', 'city', 'geotargeting', 'country_groupings', 'time_zone_info'], 'en')
+rec = ws.lookup("8.8.8.8", ["continent", "country", "region", "city", "geotargeting", "country_groupings", "time_zone_info"], "en")
 print (rec)
-print ('\n')
-print ('Credit Remaining: {}\n'.format(ws.getcredit()))
+print ("\n")
+print ("Credit Remaining: {}\n".format(ws.getcredit()))
 

data/IP2LOCATION-LITE-DB1.IPV6.BIN

@@ -14,6 +14,7 @@
 	py_modules=['IP2Location'],
 	url="https://github.com/ip2location/ip2location-python",
 	packages=setuptools.find_packages(),
+	tests_require=[pytest>=4.6.11],
 	classifiers=[
 		"Development Status :: 5 - Production/Stable",
 		"Intended Audience :: Developers",

sample.py

@@ -0,0 +1,45 @@
+# -*- coding: utf-8 -*-
+
+import pytest
+
+import os, IP2Location
+
+ipv4database = os.path.join("data", "IP2LOCATION-LITE-DB1.BIN")
+
+ipv6database = os.path.join("data", "IP2LOCATION-LITE-DB1.IPV6.BIN")
+
+def testinvaliddatabase():
+    try:
+        database = IP2Location.IP2Location(os.path.join("data", "NULL.BIN"))
+    except ValueError as e:
+        assert "The database file does not seem to exist." == str(e)
+
+def testipv4countrycode():
+    database = IP2Location.IP2Location(ipv4database)
+    rec = database.get_all("8.8.8.8")
+    assert rec.country_short == "US", "Test failed because country code not same."
+
+def testipv4countryname():
+    database = IP2Location.IP2Location(ipv4database)
+    rec = database.get_all("8.8.8.8")
+    assert rec.country_long == "United States of America", "Test failed because country name not same."
+
+def testipv4unsupportedfield():
+    database = IP2Location.IP2Location(ipv4database)
+    rec = database.get_all("8.8.8.8")
+    assert rec.city == None, "Test failed because city name not same."
+
+def testipv6countrycode():
+    database = IP2Location.IP2Location(ipv6database)
+    rec = database.get_all("2001:4860:4860::8888")
+    assert rec.country_short == "US", "Test failed because country code not same."
+
+def testipv6countryname():
+    database = IP2Location.IP2Location(ipv6database)
+    rec = database.get_all("2001:4860:4860::8888")
+    assert rec.country_long == "United States of America", "Test failed because country name not same."
+
+def testipv6unsupportedfield():
+    database = IP2Location.IP2Location(ipv6database)
+    rec = database.get_all("2001:4860:4860::8888")
+    assert rec.city == None, "Test failed because city name not same."

setup.py

@@ -0,0 +1,22 @@
+# -*- coding: utf-8 -*-
+
+import pytest
+
+import IP2Location
+
+apikey = "demo"
+package = "WS24"
+usessl = True
+addons = ["continent", "country", "region", "city", "geotargeting", "country_groupings", "time_zone_info"]
+language = "en"
+
+ws = IP2Location.IP2LocationWebService(apikey,package,usessl)
+
+def testcountrycode():
+    # ws = IP2Location.IP2LocationWebService(apikey,package,usessl)
+    rec = ws.lookup("8.8.8.8", addons, language)
+    assert rec['country_code'] == "US", "Test failed because country code not same."
+
+def testgetcredit():
+    credit = ws.getcredit()
+    assert str(credit).isdigit() == True, "Test failed because it is no a digit value."