2 files changed, 8 insertions(+), 3 deletions(-)

M fanboi2/tests/test_utils.py
M fanboi2/utils/geoip.py

@@ 92,11 92,16 @@ class TestGeoIP(unittest.TestCase):
         return geoip
 
     def _makeGeoIP2(self, country_code=None):
-        class MockGeoIP2Response(object):
+        class MockGeoIP2CountryResponse(object):
             @property
-            def country_code(self):
+            def iso_code(self):
                 return country_code
 
+        class MockGeoIP2Response(object):
+            @property
+            def country(self):
+                return MockGeoIP2CountryResponse()
+
         class MockGeoIP2(object):
             def country(self, ip_address):
                 return MockGeoIP2Response()

@@ 45,4 45,4 @@ class GeoIP(object):
             except AddressNotFoundError:
                 pass
         if response is not None:
-            return response.country_code
+            return response.country.iso_code