fix: Fix geoip not containing postal_code for UK (#19307)

PostHog · Dec 13, 2023 · cab8aea · cab8aea
1 parent 5583607
commit cab8aea
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/posthog/api/test/test_geoip.py b/posthog/api/test/test_geoip.py
@@ -16,15 +16,16 @@
 @pytest.mark.parametrize(
     "test_input,expected",
     [
-        (australia_ip, "Australia"),
-        (uk_ip, "United Kingdom"),
-        (us_ip_v6, "United States"),
+        (australia_ip, ("Australia", 6)),
+        (uk_ip, ("United Kingdom", 5)),
+        (us_ip_v6, ("United States", 6)),
     ],
 )
 def test_geoip_results(test_input, expected):
     properties = get_geoip_properties(test_input)
-    assert properties["$geoip_country_name"] == expected
-    assert len(properties) == 6
+    name, length = expected
+    assert properties["$geoip_country_name"] == name
+    assert len(properties) == length
 
 
 class TestGeoIPDBError(TestCase):