From 3132948ffc554d7a270984129383451f074acba2 Mon Sep 17 00:00:00 2001 From: Uman Shahzad Date: Fri, 6 Nov 2020 11:51:36 +0500 Subject: [PATCH] add batch test case for sync handler --- tests/handler_test.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/handler_test.py b/tests/handler_test.py index 58b4d8a..d771b77 100644 --- a/tests/handler_test.py +++ b/tests/handler_test.py @@ -79,6 +79,27 @@ def test_get_details(n): assert len(domains["domains"]) == 5 +@pytest.mark.parametrize("n", range(5)) +def test_get_batch_details(n): + token = os.environ.get("IPINFO_TOKEN", "") + handler = Handler(token) + ips = ["1.1.1.1", "8.8.8.8", "9.9.9.9"] + details = handler.getBatchDetails(ips) + + for ip in ips: + assert ip in details + d = details[ip] + assert d["ip"] == ip + assert d["country"] == "US" + assert d["country_name"] == "United States" + if token: + assert "asn" in d + assert "company" in d + assert "privacy" in d + assert "abuse" in d + assert "domains" in d + + def test_builtin_ip_types(): handler = Handler() fake_details = {"country": "US", "ip": "127.0.0.1", "loc": "12.34,56.78"}