"""Tests for school autosuggest (spec 2026-08-26).""" from backend import data_loader class _FakeDocs: def __init__(self, hits, explode=False): self._hits = hits self._explode = explode self.last_params = None def search(self, params): self.last_params = params if self._explode: raise RuntimeError("typesense is down") return {"hits": [{"document": d} for d in self._hits]} class _FakeClient: def __init__(self, hits, explode=False): self.docs = _FakeDocs(hits, explode) self.collections = {"schools": type("C", (), {"documents": self.docs})()} _HIT = { "urn": 100010, "school_name": "Brecknock Primary School", "local_authority": "Camden", "postcode": "NW1 1AA", "phase": "Primary", "school_type": "Community school", } def _use(monkeypatch, client): monkeypatch.setattr(data_loader, "_get_typesense_client", lambda: client) def test_returns_the_fields_a_suggestion_needs(monkeypatch): # Local authority is not decoration: there are many schools called # "St Mary's", and a list without it cannot be chosen between. _use(monkeypatch, _FakeClient([_HIT])) out = data_loader.suggest_schools_typesense("breck") assert out == [{ "urn": 100010, "school_name": "Brecknock Primary School", "local_authority": "Camden", "postcode": "NW1 1AA", "phase": "Primary", "school_type": "Community school", }] def test_a_missing_optional_field_becomes_an_empty_string(monkeypatch): # phase and school_type are optional in the Typesense schema. A missing # key must not KeyError in the keystroke path. _use(monkeypatch, _FakeClient([{"urn": 1, "school_name": "X", "local_authority": "Y", "postcode": "Z"}])) out = data_loader.suggest_schools_typesense("x") assert out[0]["phase"] == "" and out[0]["school_type"] == "" def test_typesense_unavailable_gives_no_suggestions_rather_than_raising(monkeypatch): _use(monkeypatch, None) assert data_loader.suggest_schools_typesense("anything") == [] def test_a_typesense_error_gives_no_suggestions_rather_than_raising(monkeypatch): _use(monkeypatch, _FakeClient([], explode=True)) assert data_loader.suggest_schools_typesense("anything") == [] def test_the_limit_is_passed_through_and_clamped(monkeypatch): client = _FakeClient([]) _use(monkeypatch, client) data_loader.suggest_schools_typesense("x", limit=500) assert client.docs.last_params["per_page"] == 20