133 lines
3.1 KiB
Python
133 lines
3.1 KiB
Python
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"}
|