tests: login, articles, like

Author: kayprogrammer

app/routes.py

@@ -110,7 +110,7 @@ async def single_article_view(
     summary="Like an article",
     description="""
         ****
-        This endpoint allows authenticated users to like an article
+        This endpoint allows authenticated users to like or unlike an article
         Set token generated from the login endpoint in the authorize dialog field. 
     """,
     status_code=200,

app/schemas.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 from typing import List
 from pydantic import BaseModel, EmailStr, Field
 
@@ -28,6 +29,8 @@ class ArticleSchema(BaseModel):
     slug: str
     desc: str
     likes_count: int
+    created_at: datetime
+    updated_at: datetime
 
 
 class ArticlesResponseSchema(ResponseSchema):

app/tests/__init__.py

@@ -0,0 +1,92 @@
+from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
+
+from app.main import app
+from app.database import get_db, Base
+from pytest_postgresql import factories
+from pytest_postgresql.janitor import DatabaseJanitor
+from httpx import AsyncClient
+import pytest, asyncio
+
+from app.models import Article, User
+from app.utils import get_password_hash
+
+test_db = factories.postgresql_proc(port=None, dbname="test_db")
+
+
+@pytest.fixture(scope="session")
+def event_loop():
+    """Overrides pytest default function scoped event loop"""
+    loop = asyncio.get_event_loop_policy().new_event_loop()
+    yield loop
+    loop.close()
+
+
+@pytest.fixture(scope="session")
+async def engine(test_db):
+    pg_host = test_db.host
+    pg_port = test_db.port
+    pg_user = test_db.user
+    pg_db = test_db.dbname
+    pg_password = test_db.password
+
+    with DatabaseJanitor(
+        user=pg_user,
+        host=pg_host,
+        port=pg_port,
+        dbname=pg_db,
+        version=test_db.version,
+        password=pg_password,
+    ):
+        connection_str = f"postgresql+psycopg://{pg_user}:@{pg_host}:{pg_port}/{pg_db}"
+        engine = create_async_engine(connection_str)
+        yield engine
+        engine.dispose()
+
+
+@pytest.fixture
+async def database(engine):
+    async with engine.begin() as conn:
+        await conn.run_sync(Base.metadata.drop_all)
+        await conn.run_sync(Base.metadata.create_all)
+
+    TestSessionLocal = async_sessionmaker(
+        bind=engine,
+        expire_on_commit=False,
+    )
+    async with TestSessionLocal() as db:
+        yield db
+
+
+@pytest.fixture
+async def client(database):
+    async def overide_get_db():
+        try:
+            yield database
+        finally:
+            await database.close()
+
+    app.dependency_overrides[get_db] = overide_get_db
+    async with AsyncClient(app=app, base_url="http://test/api/v1") as client:
+        yield client
+
+
+@pytest.fixture
+async def test_user(database):
+    user = User(
+        name="Test User",
+        email="testuser@example.com",
+        password=get_password_hash("testpassword"),
+    )
+    database.add(user)
+    await database.commit()
+    return user
+
+
+@pytest.fixture
+async def test_article(database):
+    article = Article(
+        title="Test Article", slug="test-article", desc="This is my test article"
+    )
+    database.add(article)
+    await database.commit()
+    return article

app/tests/conftest.py

@@ -0,0 +1,94 @@
+from app.utils import create_auth_token
+
+
+async def test_login(mocker, client, test_user):
+    # Test for error response with invalid credentials
+    response = await client.post(
+        "/auth/login",
+        json={"email": "invalid@email.com", "password": "invalidpassword"},
+    )
+    assert response.status_code == 401
+    assert response.json() == {
+        "status": "failure",
+        "message": "Invalid credentials",
+    }
+
+    # Test for success response valid credentials
+    response = await client.post(
+        "/auth/login",
+        json={"email": test_user.email, "password": "testpassword"},
+    )
+    assert response.status_code == 201
+    assert response.json() == {
+        "status": "success",
+        "message": "Login successful",
+        "data": {"token": mocker.ANY},
+    }
+
+
+async def test_retrieve_all_articles(client, test_article):
+    # Verify that all articles are retrieved successfully
+    response = await client.get("/articles")
+    assert response.status_code == 200
+    result = response.json()
+    assert result["status"] == "success"
+    assert result["message"] == "Articles fetched successfully"
+    data = result["data"]
+    assert len(data) > 0
+    assert any(isinstance(obj["title"], str) for obj in data)
+
+
+async def test_retrieve_article_detail(client, test_article):
+    # Verify that a single article detail was received successfully
+    response = await client.get(f"/articles/{test_article.slug}")
+    assert response.status_code == 200
+    assert response.json() == {
+        "status": "success",
+        "message": "Article details fetched successfully",
+        "data": {
+            "title": test_article.title,
+            "slug": test_article.slug,
+            "desc": test_article.desc,
+            "likes_count": test_article.likes_count,
+            "created_at": test_article.created_at.isoformat(),
+            "updated_at": test_article.updated_at.isoformat(),
+        },
+    }
+
+    # Verify that an error returns for invalid slug
+    response = await client.get("/articles/invalid_slug")
+    assert response.status_code == 404
+    assert response.json() == {
+        "status": "failure",
+        "message": "Article does not exist!",
+    }
+
+
+async def test_like_article(client, test_article, test_user):
+    # Verify that an error returns for unauthorized user
+    response = await client.get(f"/articles/{test_article.slug}/like")
+    assert response.status_code == 401
+    assert response.json() == {
+        "status": "failure",
+        "message": "Unauthorized User!",
+    }
+
+    # Set authorization for client
+    token = create_auth_token(test_user.id)
+    client.headers = {**client.headers, "Authorization": f"Bearer {token}"}
+
+    # Verify that an error returns for invalid slug
+    response = await client.get("/articles/invalid_slug/like")
+    assert response.status_code == 404
+    assert response.json() == {
+        "status": "failure",
+        "message": "Article does not exist!",
+    }
+
+    # Verify that the article was liked or unliked successfully
+    response = await client.get(f"/articles/{test_article.slug}/like")
+    assert response.status_code == 200
+    assert response.json() == {
+        "status": "success",
+        "message": "Like added successfully",
+    }

app/tests/test_routes.py

@@ -8,7 +8,7 @@
 from app.conf import settings
 from app.models import User
 
-pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")
+pwd_context = CryptContext(schemes=["pbkdf2_sha256"], deprecated="auto")
 
 ALGORITHM = "HS256"
 

app/utils.py

@@ -14,23 +14,34 @@ httpcore==1.0.6
 httptools==0.6.4
 httpx==0.27.2
 idna==3.10
+iniconfig==2.0.0
 Jinja2==3.1.4
 Mako==1.3.6
 markdown-it-py==3.0.0
 MarkupSafe==3.0.2
 mdurl==0.1.2
+mirakuru==2.5.3
+packaging==24.1
 passlib==1.7.4
+pluggy==1.5.0
+port-for==0.7.4
+psutil==6.1.0
 psycopg==3.2.3
 psycopg-binary==3.2.3
 pydantic==2.9.2
 pydantic-settings==2.6.1
 pydantic_core==2.23.4
 Pygments==2.18.0
 PyJWT==2.9.0
+pytest==8.3.3
+pytest-asyncio==0.24.0
+pytest-mock==3.14.0
+pytest-postgresql==6.1.1
 python-dotenv==1.0.1
 python-multipart==0.0.16
 PyYAML==6.0.2
 rich==13.9.3
+setuptools==75.3.0
 shellingham==1.5.4
 sniffio==1.3.1
 SQLAlchemy==2.0.36