| from datetime import datetime |
|
|
| from fastapi.testclient import TestClient |
|
|
| from birdie.api import _counts, _status, create_app, mount_mobile_web |
| from birdie.models import Seat, SeatSnapshot, SeatState, Venue, VenueConfig, VenueSource, VenueType |
| from birdie.service import BirdieService |
| from birdie.storage import Repository |
|
|
|
|
| def test_public_api_reports_unknown_seats_as_not_free_and_hides_camera_polygons(clock): |
| venue = Venue( |
| venue_id="v", name="Venue", area="Doha", type=VenueType.cowork, |
| lat=0, lon=0, zones=["open"], source=VenueSource.recorded, |
| ) |
| seat = Seat( |
| seat_id="A", venue_id="v", zone="open", |
| polygon=[(0, 0), (1, 0), (1, 1)], |
| ) |
| svc = BirdieService( |
| repo=Repository(":memory:"), clock=clock, |
| venues={"v": VenueConfig(venue=venue, seats=[seat])}, |
| detector=lambda image, threshold: [], persist_venues=False, |
| ) |
| client = TestClient(create_app(svc)) |
|
|
| listed = client.get("/api/venues").json()[0] |
| detail = client.get("/api/venues/v").json() |
| assert listed["counts"]["free"] == 0 |
| assert listed["counts"]["unknown"] == 1 |
| assert listed["counts"]["taken"] == 0 |
| assert detail["seats"][0]["state"] == SeatState.UNKNOWN.value |
| assert detail["seats"][0]["status"] == "checking" |
| assert "polygon" not in detail["seats"][0] |
| assert client.get("/api/venues/missing").status_code == 404 |
|
|
|
|
| def test_public_availability_has_three_groups_and_a_checking_indicator(): |
| assert {state: _status(state) for state in SeatState} == { |
| SeatState.UNKNOWN: "checking", SeatState.FREE: "free", SeatState.OCCUPIED: "taken", |
| SeatState.HELD: "held", SeatState.REVIEW_REQUIRED: "held", SeatState.UNAVAILABLE: "taken", |
| } |
| now = datetime.fromisoformat("2026-11-07T12:00:00+00:00") |
| states = [SeatState.FREE, SeatState.OCCUPIED, SeatState.UNAVAILABLE, SeatState.HELD, |
| SeatState.REVIEW_REQUIRED, SeatState.UNKNOWN] |
| snapshots = [SeatSnapshot(venue_id="v", seat_id=str(i), zone="open", has_outlet=False, |
| state=state, since=now, updated_at=now) for i, state in enumerate(states)] |
| counts = _counts(snapshots) |
| assert (counts["free"], counts["taken"], counts["held_or_review"], counts["checking"]) == (1, 2, 2, 1) |
| assert counts["free"] + counts["taken"] + counts["held_or_review"] + counts["checking"] == counts["total"] |
|
|
|
|
| def test_mobile_web_is_served_after_api_routes(clock): |
| venue = Venue( |
| venue_id="v", name="Venue", area="Doha", type=VenueType.cowork, |
| lat=0, lon=0, zones=["open"], source=VenueSource.recorded, |
| ) |
| svc = BirdieService( |
| repo=Repository(":memory:"), clock=clock, |
| venues={"v": VenueConfig(venue=venue, seats=[])}, persist_venues=False, |
| detector=lambda image, threshold: [], |
| ) |
| app = create_app(svc) |
| mount_mobile_web(app) |
| client = TestClient(app) |
| assert "Birdie" in client.get("/").text |
| assert client.get("/api/venues").status_code == 200 |
|
|
|
|
| def test_staff_override_requires_token_and_changes_availability(clock, monkeypatch): |
| venue = Venue( |
| venue_id="v", name="Venue", area="Doha", type=VenueType.cowork, |
| lat=0, lon=0, zones=["open"], source=VenueSource.recorded, |
| ) |
| seat = Seat(seat_id="A", venue_id="v", zone="open") |
| svc = BirdieService( |
| repo=Repository(":memory:"), clock=clock, |
| venues={"v": VenueConfig(venue=venue, seats=[seat])}, |
| detector=lambda image, threshold: [], persist_venues=False, |
| ) |
| client = TestClient(create_app(svc)) |
| monkeypatch.setenv("BIRDIE_ADMIN_TOKEN", "test-token") |
|
|
| response = client.put("/api/venues/v/seats/A/unavailable", json={"unavailable": True}) |
| assert response.status_code == 401 |
| response = client.put( |
| "/api/venues/v/seats/A/unavailable", json={"unavailable": True}, |
| headers={"Authorization": "Bearer test-token"}, |
| ) |
| assert response.status_code == 200 |
| assert response.json()["state"] == SeatState.UNAVAILABLE.value |
| assert svc.counts("v")[SeatState.FREE] == 0 |
|
|
|
|
| def test_forecast_starts_from_the_live_count_and_covers_three_hours(clock): |
| venue = Venue( |
| venue_id="v", name="Venue", area="Doha", type=VenueType.library, |
| lat=0, lon=0, zones=["open"], source=VenueSource.simulated, drive_minutes=15, |
| ) |
| seats = [Seat(seat_id=f"A{i}", venue_id="v", zone="open") for i in range(1, 9)] |
| svc = BirdieService( |
| repo=Repository(":memory:"), clock=clock, |
| venues={"v": VenueConfig(venue=venue, seats=seats)}, |
| detector=lambda image, threshold: [], persist_venues=False, |
| ) |
| client = TestClient(create_app(svc)) |
|
|
| listed = client.get("/api/venues").json()[0]["forecast"] |
| detail = client.get("/api/venues/v").json() |
| timeline = detail["forecast"]["timeline"] |
| assert listed["arrival"]["minutes"] == 15 |
| assert "timeline" not in listed |
| assert [p["minutes"] for p in timeline] == [0, 30, 60, 90, 120, 150, 180] |
| assert timeline[0]["expected_free"] == detail["counts"]["usable_study_spaces"] |
| assert all(0 <= p["low"] <= p["expected_free"] <= p["high"] <= 8 for p in timeline) |
|
|
