feat: add unit tests and make new schema for updation

Author: mdhishaamakhtar

main.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI
 
 from routes.posts import router
-from schemas.schemas import HealthResponse
+from schemas.models import HealthResponse
 
 app = FastAPI()
 

requirements.txt

@@ -3,6 +3,8 @@ aniso8601==7.0.0
 appdirs==1.4.4
 async-exit-stack==1.0.1
 async-generator==1.10
+atomicwrites==1.4.0
+attrs==20.3.0
 black==20.8b1
 certifi==2020.12.5
 chardet==4.0.0
@@ -17,15 +19,23 @@ graphql-relay==2.0.1
 greenlet==1.0.0
 h11==0.12.0
 idna==2.10
+iniconfig==1.1.1
+isort==5.8.0
 itsdangerous==1.1.0
 Jinja2==2.11.3
 MarkupSafe==1.1.1
 mypy-extensions==0.4.3
 orjson==3.5.2
+packaging==20.9
 pathspec==0.8.1
+pluggy==0.13.1
 promise==2.3
 psycopg2==2.8.6
+py==1.10.0
 pydantic==1.8.1
+pyparsing==2.4.7
+pytest==6.2.3
+pytest-dependency==0.5.1
 python-decouple==3.4
 python-dotenv==0.17.0
 python-multipart==0.0.5
@@ -34,7 +44,7 @@ regex==2021.4.4
 requests==2.25.1
 Rx==1.6.1
 six==1.15.0
-SQLAlchemy==1.4.10
+SQLAlchemy==1.4.11
 starlette==0.13.6
 toml==0.10.2
 typed-ast==1.4.3

routes/posts.py

@@ -4,7 +4,7 @@
 from sqlalchemy.orm import Session
 
 from database.connection import get_db
-from schemas.schemas import DeletePostResponse, Post
+from schemas.models import DeletePostResponse, Post, UpdatePost
 from utils.post_crud import (
     post_create,
     post_delete,
@@ -45,5 +45,5 @@ def delete_post(id, db: Session = Depends(get_db)):
 
 
 @router.patch("/update", status_code=status.HTTP_200_OK, response_model=Post)
-def update_post(post: Post, db: Session = Depends(get_db)):
+def update_post(post: UpdatePost, db: Session = Depends(get_db)):
     return post_update(db=db, post=post)

schemas/schemas.py schemas/models.pyschemas/schemas.py renamed to schemas/models.py

@@ -19,3 +19,12 @@ class Config:
 
 class DeletePostResponse(BaseModel):
     detail: str
+
+
+class UpdatePost(BaseModel):
+    id: UUID
+    title: str
+    description: str
+
+    class Config:
+        orm_mode = True

tests/__init__.py

@@ -0,0 +1,75 @@
+import pytest
+from fastapi import status
+from fastapi.testclient import TestClient
+
+from main import app
+
+client = TestClient(app)
+
+initial_post_title = "Hello"
+initial_post_description = "World"
+changed_post_description = "From the other side"
+
+
+@pytest.mark.dependency()
+def test_create_post(request):
+    response = client.post(
+        "/posts/create",
+        json={"title": initial_post_title, "description": initial_post_description},
+    )
+    assert response.status_code == status.HTTP_201_CREATED
+    assert response.json()["title"] == "Hello"
+    assert response.json()["description"] == "World"
+    request.config.cache.set("post_id", response.json()["id"])
+
+
+@pytest.mark.dependency(depends=["test_create_post"])
+def test_get_all_posts():
+    response = client.get("/posts/list/all")
+    assert response.status_code == status.HTTP_200_OK
+    assert response.json() is not None
+
+
+@pytest.mark.dependency(depends=["test_create_post"])
+def test_get_one_post(request):
+    post_id = request.config.cache.get("post_id", None)
+    response = client.get(f"/posts/get/{post_id}")
+    assert response.status_code == status.HTTP_200_OK
+    assert response.json()["id"] == post_id
+    assert response.json()["title"] == initial_post_title
+    assert (
+        response.json()["description"] == initial_post_description
+        or changed_post_description
+    )
+
+
+@pytest.mark.dependency(depends=["test_create_post", "test_get_one_post"])
+def test_patch_post(request):
+    post_id = request.config.cache.get("post_id", None)
+    response = client.patch(
+        "/posts/update",
+        json={
+            "id": post_id,
+            "title": initial_post_title,
+            "description": changed_post_description,
+        },
+    )
+    assert response.status_code == status.HTTP_200_OK
+    assert response.json()["id"] == post_id
+    assert response.json()["title"] == initial_post_title
+    assert response.json()["description"] == changed_post_description
+
+
+@pytest.mark.dependency(
+    depends=[
+        "test_create_post",
+        "test_get_one_post",
+        "test_patch_post",
+        "test_get_all_posts",
+    ]
+)
+def test_delete_post(request):
+    post_id = request.config.cache.get("post_id", None)
+    response = client.delete(f"/posts/delete/{post_id}")
+    assert response.status_code == status.HTTP_200_OK
+    assert response.json()["detail"] == "Post Deleted"

tests/test_posts.py

@@ -3,7 +3,7 @@
 from sqlalchemy.orm import Session
 
 from database.models import Posts
-from schemas.schemas import DeletePostResponse, Post
+from schemas.models import DeletePostResponse, Post, UpdatePost
 
 
 def post_create(db: Session, post: Post):
@@ -22,7 +22,7 @@ def post_get_one(db: Session, id: UUID):
     return db.query(Posts).filter_by(id=id).one()
 
 
-def post_update(db: Session, post: Post):
+def post_update(db: Session, post: UpdatePost):
     update_query = {Posts.title: post.title, Posts.description: post.description}
     db.query(Posts).filter_by(id=post.id).update(update_query)
     db.commit()