add airlines crud

tests/test_airline.py

@@ -0,0 +1,132 @@
+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_airline(client: FlaskClient, tenant: dbm.Tenant):
+    response = client.post(
+        "/api/v1/airlines",
+        json={
+            "airline": {
+                "name": "Test Airline",
+                "iata": "TA",
+                "icao": "TST",
+            }
+        },
+    )
+
+    assert response.status_code == 201
+    assert response.json == {
+        "airline": {
+            "id": 1,
+            "name": "Test Airline",
+            "iata": "TA",
+            "icao": "TST",
+        }
+    }
+
+
+def test_get_airline(client: FlaskClient, tenant: dbm.Tenant):
+    airline = dbm.Airline(
+        tenant_id=tenant.id,
+        name="Test Airline",
+        iata="TA",
+        icao="TST",
+    )
+    db.session.add(airline)
+    db.session.commit()
+
+    response = client.get(f"/api/v1/airlines/{airline.id}")
+
+    assert response.status_code == 200
+    assert response.json == {
+        "airline": {
+            "id": airline.id,
+            "name": "Test Airline",
+            "iata": "TA",
+            "icao": "TST",
+        }
+    }
+
+
+def test_get_airline_not_found(client: FlaskClient):
+    response = client.get("/api/v1/airlines/1")
+
+    assert response.status_code == 404
+    assert response.json == {"message": "Airline not found"}
+
+
+def test_update_airline(client: FlaskClient, tenant: dbm.Tenant):
+    airline = dbm.Airline(
+        tenant_id=tenant.id,
+        name="Test Airline",
+        iata="TA",
+        icao="TST",
+    )
+    db.session.add(airline)
+    db.session.commit()
+
+    response = client.put(
+        f"/api/v1/airlines/{airline.id}",
+        json={
+            "airline": {
+                "name": "Updated Airline",
+                "iata": "UA",
+                "icao": "UPD",
+            }
+        },
+    )
+
+    assert response.status_code == 200
+    assert response.json == {
+        "airline": {
+            "id": airline.id,
+            "name": "Updated Airline",
+            "iata": "UA",
+            "icao": "UPD",
+        }
+    }
+
+
+def test_update_airline_not_found(client: FlaskClient):
+    response = client.put(
+        "/api/v1/airlines/1",
+        json={
+            "airline": {
+                "name": "Updated Airline",
+                "iata": "UA",
+                "icao": "UPD",
+            }
+        },
+    )
+
+    assert response.status_code == 404
+    assert response.json == {"message": "Airline not found"}
+
+
+def test_delete_airline(client: FlaskClient, tenant: dbm.Tenant):
+    airline = dbm.Airline(
+        tenant_id=tenant.id,
+        name="Test Airline",
+        iata="TA",
+        icao="TST",
+    )
+    db.session.add(airline)
+    db.session.commit()
+
+    response = client.delete(f"/api/v1/airlines/{airline.id}")
+
+    assert response.status_code == 204
+    assert response.data == b""
+
+    with db.session.begin():
+        assert db.session.scalar(select(func.count(dbm.Airline.id))) == 0
+
+
+def test_delete_airline_not_found(client: FlaskClient):
+    response = client.delete("/api/v1/airlines/1")
+
+    assert response.status_code == 404
+    assert response.json == {"message": "Airline not found"}