78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
import datetime
|
|
|
|
import databases
|
|
import pytest
|
|
import sqlalchemy
|
|
from fastapi import FastAPI
|
|
from starlette.testclient import TestClient
|
|
|
|
from tests.settings import DATABASE_URL
|
|
from tests.test_inheritance_concrete import Category, Subject, metadata
|
|
|
|
app = FastAPI()
|
|
database = databases.Database(DATABASE_URL, force_rollback=True)
|
|
app.state.database = database
|
|
|
|
|
|
@app.on_event("startup")
|
|
async def startup() -> None:
|
|
database_ = app.state.database
|
|
if not database_.is_connected:
|
|
await database_.connect()
|
|
|
|
|
|
@app.on_event("shutdown")
|
|
async def shutdown() -> None:
|
|
database_ = app.state.database
|
|
if database_.is_connected:
|
|
await database_.disconnect()
|
|
|
|
|
|
@app.post("/subjects/", response_model=Subject)
|
|
async def create_item(item: Subject):
|
|
return item
|
|
|
|
|
|
@app.post("/categories/", response_model=Category)
|
|
async def create_category(category: Category):
|
|
await category.save()
|
|
return category
|
|
|
|
|
|
@pytest.fixture(autouse=True, scope="module")
|
|
def create_test_database():
|
|
engine = sqlalchemy.create_engine(DATABASE_URL)
|
|
metadata.create_all(engine)
|
|
yield
|
|
metadata.drop_all(engine)
|
|
|
|
|
|
def test_read_main():
|
|
client = TestClient(app)
|
|
with client as client:
|
|
test_category = dict(name="Foo", code=123, created_by="Sam", updated_by="Max")
|
|
test_subject = dict(name="Bar")
|
|
|
|
response = client.post(
|
|
"/categories/", json=test_category
|
|
)
|
|
assert response.status_code == 200
|
|
cat = Category(**response.json())
|
|
assert cat.name == 'Foo'
|
|
assert cat.created_by == 'Sam'
|
|
assert cat.created_date is not None
|
|
assert cat.id == 1
|
|
|
|
cat_dict = cat.dict()
|
|
cat_dict['updated_date'] = cat_dict['updated_date'].strftime("%Y-%m-%d %H:%M:%S.%f")
|
|
cat_dict['created_date'] = cat_dict['created_date'].strftime("%Y-%m-%d %H:%M:%S.%f")
|
|
test_subject['category'] = cat_dict
|
|
response = client.post(
|
|
"/subjects/", json=test_subject
|
|
)
|
|
assert response.status_code == 200
|
|
sub = Subject(**response.json())
|
|
assert sub.name == 'Bar'
|
|
assert sub.category.pk == cat.pk
|
|
assert isinstance(sub.updated_date, datetime.datetime)
|