stub out v1 api

tests/conftest.py

@@ -1,9 +1,31 @@
 import pytest
+from flask import Flask, g
+from flask.testing import FlaskClient
 
+from teufa import db as dbm
 from teufa.app import create_app
+from teufa.ext import db
 
 
 @pytest.fixture
 def app():
     app = create_app()
-    yield app
+    with app.app_context():
+        dbm.Base.metadata.create_all(db.engine)
+
+        tenant = dbm.Tenant(
+            name="Default",
+            hostname="localhost",
+        )
+        db.session.add(tenant)
+        db.session.commit()
+        g.tenant = tenant
+
+        yield app
+
+        dbm.Base.metadata.drop_all(db.engine)
+
+
+@pytest.fixture
+def client(app: Flask) -> FlaskClient:
+    return app.test_client()

tests/test_flight.py

@@ -0,0 +1,173 @@
+import json
+
+from flask import g
+from flask.testing import FlaskClient
+from sqlalchemy.sql import func, select
+
+from teufa import db as dbm
+from teufa.ext import db
+
+
+def test_create_flight(client: FlaskClient):
+    aircraft = dbm.Aircraft(
+        tenant_id=g.tenant.id,
+        icao="B737",
+        tail_number="N12345",
+        range_nm=3000,
+    )
+    db.session.add(aircraft)
+    db.session.commit()
+
+    response = client.post(
+        "/api/v1/flights",
+        json={
+            "flight": {
+                "departure_icao": "KDEN",
+                "arrival_icao": "KLGA",
+                "aircraft_id": aircraft.id,
+            }
+        },
+    )
+
+    assert response.status_code == 201
+    assert json.loads(response.json) == {
+        "flight": {
+            "id": 1,
+            "departure_icao": "KDEN",
+            "arrival_icao": "KLGA",
+            "aircraft_id": 1,
+        }
+    }
+
+
+def test_get_flight(client: FlaskClient):
+    aircraft = dbm.Aircraft(
+        tenant_id=g.tenant.id,
+        icao="B737",
+        tail_number="N12345",
+        range_nm=3000,
+    )
+    db.session.add(aircraft)
+    db.session.commit()
+    db.session.add(
+        dbm.Flight(
+            id=1,
+            tenant_id=g.tenant.id,
+            departure_icao="KDEN",
+            arrival_icao="KLGA",
+            aircraft_id=aircraft.id,
+        )
+    )
+    db.session.commit()
+
+    response = client.get("/api/v1/flights/1")
+
+    assert response.status_code == 200
+    assert json.loads(response.json) == {
+        "flight": {
+            "id": 1,
+            "departure_icao": "KDEN",
+            "arrival_icao": "KLGA",
+            "aircraft_id": 1,
+        }
+    }
+
+
+def test_get_flight_not_found(client: FlaskClient):
+    response = client.get("/api/v1/flights/1")
+
+    assert response.status_code == 404
+    assert json.loads(response.json) == {"message": "Flight not found"}
+
+
+def test_update_flight(client: FlaskClient):
+    aircraft = dbm.Aircraft(
+        tenant_id=g.tenant.id,
+        icao="B737",
+        tail_number="N12345",
+        range_nm=3000,
+    )
+    db.session.add(aircraft)
+    db.session.commit()
+    db.session.add(
+        dbm.Flight(
+            id=1,
+            tenant_id=g.tenant.id,
+            departure_icao="KDEN",
+            arrival_icao="KLGA",
+            aircraft_id=aircraft.id,
+        )
+    )
+    db.session.commit()
+
+    response = client.put(
+        "/api/v1/flights/1",
+        json={
+            "flight": {
+                "departure_icao": "KJFK",
+                "arrival_icao": "KLAX",
+                "aircraft_id": 1,
+            }
+        },
+    )
+
+    assert response.status_code == 200
+    assert json.loads(response.json) == {
+        "flight": {
+            "id": 1,
+            "departure_icao": "KJFK",
+            "arrival_icao": "KLAX",
+            "aircraft_id": 1,
+        }
+    }
+
+
+def test_update_flight_not_found(client: FlaskClient):
+    response = client.put(
+        "/api/v1/flights/1",
+        json={
+            "flight": {
+                "departure_icao": "KJFK",
+                "arrival_icao": "KLAX",
+            }
+        },
+    )
+
+    assert response.status_code == 404
+    assert json.loads(response.json) == {"message": "Flight not found"}
+
+
+def test_delete_flight(client: FlaskClient):
+    aircraft = dbm.Aircraft(
+        tenant_id=g.tenant.id,
+        icao="B737",
+        tail_number="N12345",
+        range_nm=3000,
+    )
+    db.session.add(aircraft)
+    db.session.commit()
+    db.session.add(
+        dbm.Flight(
+            id=1,
+            tenant_id=g.tenant.id,
+            departure_icao="KDEN",
+            arrival_icao="KLGA",
+            aircraft_id=aircraft.id,
+        )
+    )
+    db.session.commit()
+
+    response = client.delete("/api/v1/flights/1")
+
+    assert response.status_code == 204
+    assert response.data == b""
+
+    with db.session.begin():
+        assert db.session.scalar(select(func.count(dbm.Flight.id))) == 0
+
+
+def test_delete_flight_not_found(client: FlaskClient):
+    response = client.delete("/api/v1/flights/1")
+
+    assert response.status_code == 404
+    assert json.loads(response.json) == {"message": "Flight not found"}