add test

Author: Chise1

conftest.py

@@ -4,6 +4,7 @@
 from sqlalchemy.engine import ChunkedIteratorResult
 
 from fast_tmp.models import Base, User
+import tests.models
 import pytest
 from fast_tmp.db import engine, get_db_session
 
@@ -13,7 +14,7 @@ def prepare_database() -> Generator[None, None, None]:
     Base.metadata.create_all(engine)
     session = next(get_db_session())
     res: ChunkedIteratorResult = session.execute(select(User.username).where(User.username == "root"))
-    if len(res.fetchall())==0:
+    if len(res.fetchall()) == 0:
         user = User(username="root")
         user.set_password("root")
         session.add(user)

tests/test_many_to_one.py

@@ -0,0 +1,23 @@
+from fastapi import FastAPI
+from fastapi.testclient import TestClient
+
+from fast_tmp.admin.server import admin
+from fast_tmp.db import get_db_session
+
+from .models import Author, Book
+
+app = FastAPI()
+app.mount("/admin", admin, name="admin")
+
+
+def get_cookie(client: TestClient):
+    login_post_res = client.post("/admin/login", data={"username": "root", "password": "root"})
+    assert login_post_res.status_code == 307
+    return login_post_res.cookies.get("access_token")
+
+
+def write_base_data():
+    session_c = get_db_session()
+    session = next(session_c)
+
+    next(session_c)