anti
010568e558
Mirrors decnet/geoip/ end-to-end: paths/base/factory/lookup at the package level, iptoasn/ subpackage holds the data-source-specific fetch+parse+provider. AsnLookup is bisect-indexed over (start, end, AsnInfo) ranges with a pickled cache invalidated on raw-file mtime bump. Why iptoasn (and not bgp.tools / Team Cymru): public-domain dump, zero attribution, no UA mandate, daily refresh — keeps DECNET stealth intact (the geoip/rir module's "never identify as DECNET" comment applies the same way here). bgp.tools' ToS would have required an identifying UA, conflicting with feedback_stealth. Public surface: decnet.asn.enrich_ip(ip) -> (asn, name, source) or all-None on miss/disabled. Same shape as decnet.geoip.enrich_ip so the profiler can compose them in one call site.
96 lines
2.6 KiB
Python
96 lines
2.6 KiB
Python
"""IptoasnProvider + factory + public API tests."""
|
|
from __future__ import annotations
|
|
|
|
import gzip
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
|
|
def _seed_fixture(root: Path, content: str = "8.8.8.0\t8.8.8.255\t15169\tUS\tGOOGLE\n") -> None:
|
|
target = root / "ip2asn-v4.tsv.gz"
|
|
with gzip.open(target, "wt", encoding="utf-8") as fh:
|
|
fh.write(content)
|
|
|
|
|
|
def test_factory_returns_iptoasn_by_default() -> None:
|
|
from decnet.asn.factory import get_provider
|
|
|
|
provider = get_provider()
|
|
assert provider.name == "iptoasn"
|
|
|
|
|
|
def test_factory_rejects_unknown_provider(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
from decnet.asn import factory
|
|
|
|
monkeypatch.setenv("DECNET_ASN_PROVIDER", "nope")
|
|
factory.reset_cache()
|
|
with pytest.raises(ValueError):
|
|
factory.get_provider()
|
|
|
|
|
|
def test_provider_build_lookup_empty_when_no_files(tmp_path: Path) -> None:
|
|
from decnet.asn.iptoasn.provider import IptoasnProvider
|
|
|
|
p = IptoasnProvider()
|
|
lookup = p.build_lookup()
|
|
assert len(lookup) == 0
|
|
assert lookup.asn("8.8.8.8") is None
|
|
|
|
|
|
def test_provider_build_lookup_reads_present_file(tmp_path: Path) -> None:
|
|
from decnet.asn.iptoasn.provider import IptoasnProvider
|
|
|
|
_seed_fixture(tmp_path)
|
|
p = IptoasnProvider()
|
|
lookup = p.build_lookup()
|
|
info = lookup.asn("8.8.8.8")
|
|
assert info is not None
|
|
assert info.asn == 15169
|
|
assert info.name == "GOOGLE"
|
|
|
|
|
|
def test_provider_uses_cache_when_fresh(tmp_path: Path) -> None:
|
|
from decnet.asn.iptoasn.provider import IptoasnProvider
|
|
|
|
_seed_fixture(tmp_path)
|
|
p = IptoasnProvider()
|
|
a = p.build_lookup()
|
|
assert (tmp_path / ".iptoasn_index.pkl").exists()
|
|
|
|
p2 = IptoasnProvider()
|
|
b = p2.build_lookup()
|
|
assert len(b) == len(a)
|
|
|
|
|
|
def test_enrich_ip_short_circuits_when_disabled(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
import decnet.asn as asn
|
|
|
|
monkeypatch.setenv("DECNET_ASN_ENABLED", "false")
|
|
assert asn.enrich_ip("8.8.8.8") == (None, None, None)
|
|
|
|
|
|
def test_enrich_ip_returns_asn_and_source(tmp_path: Path) -> None:
|
|
from decnet.asn import enrich_ip
|
|
|
|
_seed_fixture(tmp_path)
|
|
asn, name, src = enrich_ip("8.8.8.8")
|
|
assert asn == 15169
|
|
assert name == "GOOGLE"
|
|
assert src == "iptoasn"
|
|
|
|
|
|
def test_enrich_ip_private_returns_none(tmp_path: Path) -> None:
|
|
from decnet.asn import enrich_ip
|
|
|
|
_seed_fixture(tmp_path)
|
|
assert enrich_ip("192.168.1.1") == (None, None, None)
|
|
|
|
|
|
def test_enrich_ip_unannounced_returns_none(tmp_path: Path) -> None:
|
|
from decnet.asn import enrich_ip
|
|
|
|
_seed_fixture(tmp_path)
|
|
# 9.0.0.0 isn't in our fixture range — no BGP announcement we know of.
|
|
assert enrich_ip("9.0.0.0") == (None, None, None)
|