updating resp

Author: Samuel-Gill

ipinfo/handler.py

@@ -218,7 +218,7 @@ def getBatchDetails(
                 details = {}
                 details["ip"] = ip_address
                 details["bogon"] = True
-                result[ip_address] = details
+                result[ip_address] = Details(details)
             else:
                 lookup_addresses.append(ip_address)
 
@@ -339,7 +339,7 @@ def getBatchDetailsIter(
                 details = {}
                 details["ip"] = ip_address
                 details["bogon"] = True
-                result[ip_address] = details
+                yield Details(details)
             else:
                 lookup_addresses.append(ip_address)
 

tests/handler_test.py

@@ -196,4 +196,4 @@ def test_iterative_bogon_details():
     token = os.environ.get("IPINFO_TOKEN", "")
     handler = Handler(token)
     details = next(handler.getBatchDetailsIter(["127.0.0.1"]))
-    assert details == {"bogon": True, "ip": "127.0.0.1"}
+    assert details.all == {"bogon": True, "ip": "127.0.0.1"}