kenzuya/unshackle
1
0
Fork 2
mirror of https://github.com/unshackle-dl/unshackle.git synced 2026-11-14 13:05:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(ip-info): return timezone and add a required-field filter

Browse Source 
get_ip_info(require=...) skips a geolocation API whose result lacks a required field, so ipinfo Lite falls through to ipinfo.io or ip-api.in.
This commit is contained in:
imSp4rky
2026-10-15 10:30:21 -06:00
parent 8a1fbf2de1
commit c231114cda
2 changed files with 26 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/core/test_verify_proxy_exit.py
+15
View File
@@ -80,3 +80,18 @@ def test_get_ip_info_collects_connection_errors_only(monkeypatch):
errors: list[Exception] = []
assert ip_info.get_ip_info(SimpleNamespace(proxies={"all": "http://127.0.0.1:9"}), errors=errors) is None
assert [str(e) for e in errors] == ["refused"]
def test_require_skips_provider_missing_field(monkeypatch):
lite = {"country": "ca", "timezone": ""}
full = {"country": "ca", "timezone": "America/Edmonton"}
monkeypatch.setattr(
ip_info, "build_providers", lambda: [("lite", lambda s: dict(lite)), ("full", lambda s: dict(full))]
)
monkeypatch.setattr(
ip_info,
"Cacher",
lambda *a: SimpleNamespace(get=lambda k: SimpleNamespace(data=None, expired=True, set=lambda *a, **k: None)),
)
assert ip_info.get_ip_info(object())["_provider"] == "lite"
assert ip_info.get_ip_info(object(), require=("timezone",))["timezone"] == "America/Edmonton"