Add tests for creating and reading blog posts and categories.

Author: drmacsika

blog/test_helpers.py

@@ -0,0 +1,28 @@
+from typing import Optional
+
+from accounts.test_helpers import create_random_user
+from core.test_utils import random_lower_string
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from blog.crud import category, post
+from blog.models import Category, Post
+from blog.schemas import CreateCategory, CreatePost
+
+
+def create_random_post(
+    db: AsyncSession, *,
+    author_id: Optional[int] = None) -> Post:
+    if author_id is None:
+        user = create_random_user(db)
+        author_id = user.id
+    title = random_lower_string()
+    description = random_lower_string()
+    obj_in = CreatePost(title=title, description=description, author_id=author_id)
+    return post.create(db=db, obj_in=obj_in)
+
+
+def create_random_category(db: AsyncSession) -> Category:
+    title = random_lower_string()
+    description = random_lower_string()
+    obj_in = CreateCategory(title=title, description=description, id=id)
+    return post.create(db=db, obj_in=obj_in)

blog/tests.py

@@ -0,0 +1,82 @@
+from core.settings import settings
+from core.utils import unique_slug_generator
+from fastapi.testclient import TestClient
+from sqlalchemy.ext.asyncio import AsyncSession
+
+from blog.test_helpers import create_random_category, create_random_post
+
+test_data = {
+        "title": "Archangel",
+        "description": "macsika"
+    }
+
+def test_create_post(
+        client: TestClient,
+        superuser_token_headers: dict,
+        db: AsyncSession
+    ) -> None:
+    response = client.post(
+        f"{settings.API_V1_STR}/blog/posts/",
+        headers=superuser_token_headers,
+        json=test_data
+    )
+    assert response.status_code == 201 or response.status_code == 200
+    content = response.json()
+    assert content["title"] == test_data["title"]
+    assert content["description"] == test_data["description"]
+    assert content["slug"] == unique_slug_generator(test_data["title"])
+    assert "id" in content
+    assert "owner_id" in content
+
+
+def test_create_category(
+        client: TestClient,
+        superuser_token_headers: dict,
+        db: AsyncSession
+    ) -> None:
+    response = client.post(
+        f"{settings.API_V1_STR}/blog/tags/",
+        headers=superuser_token_headers,
+        json=test_data
+    )
+    assert response.status_code == 201 or response.status_code == 200
+    content = response.json()
+    assert content["title"] == test_data["title"]
+    assert content["description"] == test_data["description"]
+    assert content["slug"] == unique_slug_generator(test_data["title"])
+    assert "id" in content
+
+
+def test_read_post(
+    client: TestClient,
+    superuser_token_headers: dict,
+    db: AsyncSession) -> None:
+    item = create_random_post(db)
+    response = client.get(
+        f"{settings.API_V1_STR}/blog/posts/{item.slug}",
+        headers=superuser_token_headers,
+    )
+    assert response.status_code == 200
+    content = response.json()
+    assert content["title"] == item.title
+    assert content["description"] == item.description
+    assert content["slug"] == unique_slug_generator(test_data["title"])
+    assert content["id"] == item.id
+    assert content["owner_id"] == item.owner_id
+    
+
+def test_read_category(
+    client: TestClient,
+    superuser_token_headers: dict,
+    db: AsyncSession) -> None:
+    item = create_random_post(db)
+    response = client.get(
+        f"{settings.API_V1_STR}/blog/tags/{item.slug}",
+        headers=superuser_token_headers,
+    )
+    assert response.status_code == 200
+    content = response.json()
+    assert content["title"] == item.title
+    assert content["description"] == item.description
+    assert content["slug"] == unique_slug_generator(test_data["title"])
+    assert content["id"] == item.id

core/test_utils.py

@@ -0,0 +1,27 @@
+import random
+import string
+from typing import Dict
+
+from fastapi.testclient import TestClient
+
+from core.settings import settings
+
+
+def random_lower_string() -> str:
+    return "".join(random.choices(string.ascii_lowercase, k=32))
+
+
+def random_email() -> str:
+    return f"{random_lower_string()}@{random_lower_string()}.com"
+
+
+def get_superuser_token_headers(client: TestClient) -> Dict[str, str]:
+    login_data = {
+        "username": settings.FIRST_SUPERUSER,
+        "password": settings.FIRST_SUPERUSER_PASSWORD,
+    }
+    r = client.post(f"{settings.API_V1_STR}/login/access-token", data=login_data)
+    tokens = r.json()
+    a_token = tokens["access_token"]
+    headers = {"Authorization": f"Bearer {a_token}"}
+    return headers