merge: testing → main (reconcile 2-week divergence)

tests/geoip/test_parse.py

@@ -0,0 +1,66 @@
+"""Parser tests for RIR delegated-stats files."""
+from __future__ import annotations
+
+import ipaddress
+from pathlib import Path
+
+from decnet.geoip.rir.parse import parse_file
+
+
+_FIXTURE = """\
+2|ripencc|20260420|230000|19830101|20260419|+0000
+ripencc|*|asn|*|35000|summary
+ripencc|*|ipv4|*|25000|summary
+ripencc|DE|ipv4|85.214.0.0|65536|20060814|allocated|abc
+ripencc|GB|ipv4|46.101.0.0|65536|20120101|assigned|def
+ripencc|FR|ipv6|2001:db8::|32|20100101|allocated|ghi
+ripencc|*|ipv4|5.0.0.0|256|20200101|reserved|jkl
+ripencc|ZZ|ipv4|6.0.0.0|256|20200101|allocated|mno
+ripencc|ES|ipv4|*|0|20200101|allocated|pqr
+# comment line
+ripencc|IT|asn|12345|1|20100101|allocated|stu
+arin|US|ipv4|8.8.8.0|256|20000101|allocated|xyz
+"""
+
+
+def test_parse_skips_non_ipv4_and_sentinels(tmp_path: Path) -> None:
+    fixture = tmp_path / "ripe.txt"
+    fixture.write_text(_FIXTURE)
+    ranges = list(parse_file(fixture))
+    ccs = {r[2] for r in ranges}
+    # v4 allocated/assigned with real country codes only.
+    assert ccs == {"DE", "GB", "US"}
+
+
+def test_parse_range_boundaries(tmp_path: Path) -> None:
+    fixture = tmp_path / "arin.txt"
+    fixture.write_text(_FIXTURE)
+    ranges = [r for r in parse_file(fixture) if r[2] == "US"]
+    assert len(ranges) == 1
+    start, end, cc = ranges[0]
+    assert start == int(ipaddress.IPv4Address("8.8.8.0"))
+    assert end == int(ipaddress.IPv4Address("8.8.8.255"))
+    assert cc == "US"
+
+
+def test_parse_lowercase_cc_is_uppercased(tmp_path: Path) -> None:
+    fixture = tmp_path / "apnic.txt"
+    fixture.write_text("apnic|jp|ipv4|1.0.0.0|256|19990101|allocated|abc\n")
+    ranges = list(parse_file(fixture))
+    assert ranges == [(int(ipaddress.IPv4Address("1.0.0.0")),
+                       int(ipaddress.IPv4Address("1.0.0.255")),
+                       "JP")]
+
+
+def test_parse_malformed_lines_are_skipped(tmp_path: Path) -> None:
+    fixture = tmp_path / "broken.txt"
+    fixture.write_text(
+        "garbage\n"
+        "a|b|c\n"
+        "ripencc|DE|ipv4|not-an-ip|65536|20060814|allocated|abc\n"
+        "ripencc|DE|ipv4|85.214.0.0|not-a-count|20060814|allocated|abc\n"
+        "ripencc|DE|ipv4|85.214.0.0|65536|20060814|allocated|ok\n"
+    )
+    ranges = list(parse_file(fixture))
+    assert len(ranges) == 1
+    assert ranges[0][2] == "DE"