test_post.py added

app/routers/post.py

@@ -25,7 +25,7 @@ def get_posts(db: Session = Depends(get_db),current_user: User= Depends(Get_Curr
         post.votes = vote_count
         posts.append(post)
     return posts
-@router.post("/", response_model=PostResponse)
+@router.post("/", response_model=PostResponse,status_code=status.HTTP_201_CREATED)
 def create_post(post: PostCreate, db: Session = Depends(get_db), current_user: User= Depends(Get_Current_User)):
     new_post = Post(owner_id=current_user.id,**post.model_dump())
     db.add(new_post)

tests/{database.py → conftest.py}

@@ -5,6 +5,8 @@ from app.main import app
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 from app.config import setting
+from app.oauth2 import Create_Access_Token
+from app.models import Post
 
 
 SQLALCHEMY_DATABASE_URL = f"postgresql://{setting.database_username}:{setting.database_password}@{setting.database_hostname}:{setting.database_port}/{setting.database_name}_Test"
@@ -30,4 +32,45 @@ def client(session):
         finally:
             session.close()
     app.dependency_overrides[get_db] = override_get_db
-    yield TestClient(app)
+    yield TestClient(app)
+
+@pytest.fixture
+def create_test_user(client):
+    user_data={
+        "email": "abdullah@gmail.com",
+        "password": "abdullah1234"
+    }
+    response = client.post("/users/",json=user_data)
+    assert response.status_code == 201
+    new_user=response.json()
+    new_user["password"]="abdullah1234"
+    return new_user
+
+@pytest.fixture
+def token(create_test_user):
+    return Create_Access_Token({"id": create_test_user["id"]})
+
+@pytest.fixture
+def authorize_client(client,token):
+    client.headers = {
+        **client.headers,
+        "Authorization": f"Bearer {token}"
+    }
+    return client
+
+@pytest.fixture
+def test_posts(create_test_user,session):
+        posts_data = [
+            {"title": "first title", "content": "first content", "owner_id": create_test_user['id']},
+            {"title": "2nd title",   "content": "2nd content",   "owner_id": create_test_user['id']},
+            {"title": "3rd title",   "content": "3rd content",   "owner_id": create_test_user['id']},
+        ]
+        def create_post(post):
+            return Post(**post)
+        post=map(create_post,posts_data)
+        post=list(post)
+        session.add_all(post)
+        session.commit()
+        posts=session.query(Post).all()
+        return posts
+     

tests/test_posts.py

@@ -0,0 +1,48 @@
+from urllib import response
+
+import pytest
+from typing import List
+from pydantic import TypeAdapter
+from app.schemas import PostResponse
+from app.schemas.post import PostVote
+from tests.conftest import authorize_client
+
+def test_get_all_posts(authorize_client,test_posts):
+    response = authorize_client.get("/posts/")
+    posts = TypeAdapter(List[PostResponse]).validate_python(response.json())
+    assert response.status_code == 200
+    assert len(posts) == len(test_posts)
+
+def test_unauthorized_get_all_post(client,test_posts):
+    response=client.get("/posts/")
+    assert response.status_code == 401
+
+def test_unauthorized_get_one_post(client,test_posts):
+    response=client.get(f"/posts/{test_posts[0].id}")
+    assert response.status_code == 401
+
+def test_get_one_post_not_exit(authorize_client):
+    response=authorize_client.get(f"/posts/-9999")
+    assert response.status_code == 404
+
+def test_get_one_post(authorize_client,test_posts):
+    response=authorize_client.get(f"/posts/{test_posts[0].id}")
+    post=PostVote(**response.json())
+    assert post.id == test_posts[0].id
+    
+@pytest.mark.parametrize("title, content, published", [
+    ("MHA","Anime of the Year",True),
+    ("To be Hero X","Can be Considerd to be Anime of the Year",True),
+    ("Wind Breaker","Watched",False)])
+def test_create_post(authorize_client,title,content,published,create_test_user):
+    response =authorize_client.post("/posts/",json={"title": title, "content": content, "published": published})
+    created_post = PostResponse(**response.json())
+    assert response.status_code == 201
+    assert created_post.title == title
+    assert created_post.content == content
+    assert created_post.published == published
+    assert created_post.owner_id == create_test_user['id']
+    
+def test_unauthorized_user_create_post(client,test_posts):
+    response =client.post("/posts/",json={"title": "Anime", "content": "Worth Watching", "published": True})
+    assert response.status_code == 401

tests/test_users.py

@@ -2,19 +2,6 @@ import pytest
 from app.config import setting
 from app.schemas import user, token
 from jose import jwt
-from tests.database import client, session
-
-@pytest.fixture
-def create_test_user(client):
-    user_data={
-        "email": "abdullah@gmail.com",
-        "password": "abdullah1234"
-    }
-    response = client.post("/users/",json=user_data)
-    assert response.status_code == 201
-    new_user=response.json()
-    new_user["password"]="abdullah1234"
-    return new_user
 
 def test_create_user(client):
     response = client.post("/users/", json={"email": "abdullah@gmail.com", "password": "abdullah1234"})
@@ -30,3 +17,16 @@ def test_login_user(client,create_test_user):
     assert id == create_test_user["id"]
     assert login_response.token_type == "bearer"
     assert response.status_code==200
+
+@pytest.mark.parametrize("email, password, status_code", [
+    ("abdullah@gmail.com", "wrongpassword", 403),
+    ("wrong@gmail.com",    "abdullah1234",  403),
+    ("wrong@gmail.com",    "wrongpassword", 403),
+    (None,                 None,            422),
+    ("abdullah@gmail.com", None,            422),
+    (None,                 "abdullah1234",  422),
+    ("abdullah@gmail.com", "abdullah1234",  200),
+])
+def test_login_parametrized(client, create_test_user, email, password, status_code):
+    response = client.post("/login", data={"username": email, "password": password})
+    assert response.status_code == status_code