monkeypatch probe_geoip to provide static responses

Author: aagbsn

ooniapi/services/ooniprobe/tests/integ/test_geolookup.py

@@ -1,12 +1,28 @@
-def test_geolookup(client):
+from typing import Dict, Tuple
+import ooniprobe.routers.v1.probe_services as ps
+from ooniprobe.dependencies import CCReaderDep, ASNReaderDep
+
+def fake_probe_geoip(
+    ipaddr: str,
+    probe_cc: str,
+    asn: str,
+    cc_reader: CCReaderDep,
+    asn_reader: ASNReaderDep,
+) -> Tuple[Dict, str, int]:
+    return {"probe_cc": "US", "probe_asn": "AS4242", "probe_network_name": "Testing Networks"}, "US", 4242
+
+
+def test_geolookup(client, monkeypatch):
+    monkeypatch.setattr(ps, "probe_geoip", fake_probe_geoip)
     j = dict(
         addresses=["192.33.4.12", "170.247.170.2", "2801:1b8:10::b", "2001:500:2::c"]
     )
     c = client.post("/api/v1/geolookup", json=j).json()
     assert "geolocation" in c
     assert "v" in c
     g = c["geolocation"]
+
     for ip in j["addresses"]:
-        assert g[ip]["cc"] != "ZZ"
-        assert g[ip]["asn"] != "AS0"
-        assert g[ip]["as_name"] is not None
+        assert g[ip]["cc"] == "US"
+        assert g[ip]["asn"] == "AS4242"
+        assert g[ip]["as_name"] == "Testing Networks"