Tests for the API

Signed-off-by: Abdulazeez Abdulazeez Adeshina <youngestdev@gmail.com>

Author: Youngestdev

ch08/planner/tests/conftest.py

@@ -0,0 +1,33 @@
+import asyncio
+import httpx
+import pytest
+
+from main import app
+from database.database import Settings
+from models.events import Event
+from models.users import User
+
+
+@pytest.fixture(scope="session")
+def event_loop():
+    loop = asyncio.get_event_loop()
+    yield loop
+    loop.close()
+
+
+async def init_db():
+    test_settings = Settings()
+    test_settings.DATABASE_URL = "mongodb://localhost:27017/testdb"
+
+    await test_settings.initialize_database()
+
+
+@pytest.fixture(scope="session")
+async def default_client():
+    await init_db()
+    async with httpx.AsyncClient(app=app, base_url="http://app") as client:
+        yield client
+
+        # Clean up resources
+        await Event.find_all().delete()
+        await User.find_all().delete()

ch08/planner/tests/test_arthmetic_operations.py

@@ -0,0 +1,30 @@
+def add(a: int, b: int) -> int:
+    return a + b
+
+
+def subtract(a: int, b: int) -> int:
+    return b - a
+
+
+def multiply(a: int, b: int) -> int:
+    return a * b
+
+
+def divide(a: int, b: int) -> int:
+    return b // a
+
+
+def test_add() -> None:
+    assert add(1, 1) == 2
+
+
+def test_subtract() -> None:
+    assert subtract(2, 5) == 3
+
+
+def test_multiply() -> None:
+    assert multiply(10, 10) == 100
+
+
+def test_divide() -> None:
+    assert divide(25, 100) == 4

ch08/planner/tests/test_fixture.py

@@ -0,0 +1,20 @@
+import pytest 
+
+
+# Fixture is defined.
+from models.events import EventUpdate
+
+
+@pytest.fixture
+def event() -> EventUpdate:
+    return EventUpdate(
+        title="FastAPI Book Launch",
+        image="https://packt.com/fastapi.png",
+        description="We will be discussing the contents of the FastAPI book in this event.Ensure to come with your own copy to win gifts!",
+        tags=["python", "fastapi", "book", "launch"],
+        location="Google Meet"
+    )
+
+
+def test_event_name(event: EventUpdate) -> None:
+    assert event.title == "FastAPI Book Launch"

ch08/planner/tests/test_login.py

@@ -0,0 +1,42 @@
+import httpx
+import pytest
+
+
+@pytest.mark.asyncio
+async def test_sign_new_user(default_client: httpx.AsyncClient) -> None:
+    payload = {
+        "email": "testuser@packt.com",
+        "password": "testpassword",
+    }
+
+    headers = {
+        "accept": "application/json",
+        "Content-Type": "application/json"
+    }
+
+    test_response = {
+        "message": "User created successfully"
+    }
+
+    response = await default_client.post("/user/signup", json=payload, headers=headers)
+
+    assert response.status_code == 200
+    assert response.json() == test_response
+
+
+@pytest.mark.asyncio
+async def test_sign_user_in(default_client: httpx.AsyncClient) -> None:
+    payload = {
+        "username": "testuser@packt.com",
+        "password": "testpassword"
+    }
+
+    headers = {
+        "accept": "application/json",
+        "Content-Type": "application/x-www-form-urlencoded"
+    }
+
+    response = await default_client.post("/user/signin", data=payload, headers=headers)
+
+    assert response.status_code == 200
+    assert response.json()["token_type"] == "Bearer"

ch08/planner/tests/test_routes.py

@@ -0,0 +1,131 @@
+import httpx
+import pytest
+
+from auth.jwt_handler import create_access_token
+from models.events import Event
+
+
+@pytest.fixture(scope="module")
+async def access_token() -> str:
+    return create_access_token("testuser@packt.com")
+
+
+@pytest.fixture(scope="module")
+async def mock_event() -> Event:
+    new_event = Event(
+        creator="testuser@packt.com",
+        title="FastAPI Book Launch",
+        image="https://linktomyimage.com/image.png",
+        description="We will be discussing the contents of the FastAPI book in this event.Ensure to come with your own copy to win gifts!",
+        tags=["python", "fastapi", "book", "launch"],
+        location="Google Meet"
+    )
+
+    await Event.insert_one(new_event)
+
+    yield new_event
+
+
+@pytest.mark.asyncio
+async def test_get_events(default_client: httpx.AsyncClient, mock_event: Event) -> None:
+    response = await default_client.get("/event/")
+
+    assert response.status_code == 200
+    assert response.json()[0]["_id"] == str(mock_event.id)
+
+
+@pytest.mark.asyncio
+async def test_get_event(default_client: httpx.AsyncClient, mock_event: Event) -> None:
+    url = f"/event/{str(mock_event.id)}"
+    response = await default_client.get(url)
+
+    assert response.status_code == 200
+    assert response.json()["creator"] == mock_event.creator
+    assert response.json()["_id"] == str(mock_event.id)
+
+
+@pytest.mark.asyncio
+async def test_post_event(default_client: httpx.AsyncClient, access_token: str) -> None:
+    payload = {
+        "title": "FastAPI Book Launch",
+        "image": "https://linktomyimage.com/image.png",
+        "description": "We will be discussing the contents of the FastAPI book in this event.Ensure to come with your own copy to win gifts!",
+        "tags": [
+            "python",
+            "fastapi",
+            "book",
+            "launch"
+        ],
+        "location": "Google Meet",
+    }
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {access_token}"
+    }
+
+    test_response = {
+        "message": "Event created successfully"
+    }
+
+    response = await default_client.post("/event/new", json=payload, headers=headers)
+
+    assert response.status_code == 200
+    assert response.json() == test_response
+
+
+@pytest.mark.asyncio
+async def test_get_events_count(default_client: httpx.AsyncClient) -> None:
+    response = await default_client.get("/event/")
+
+    events = response.json()
+
+    assert response.status_code == 200
+    assert len(events) == 2
+
+
+@pytest.mark.asyncio
+async def test_update_event(default_client: httpx.AsyncClient, mock_event: Event, access_token: str) -> None:
+    test_payload = {
+        "title": "Updated FastAPI event"
+    }
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {access_token}"
+    }
+
+    url = f"/event/{str(mock_event.id)}"
+
+    response = await default_client.put(url, json=test_payload, headers=headers)
+
+    assert response.status_code == 200
+    assert response.json()["title"] == test_payload["title"]
+
+
+@pytest.mark.asyncio
+async def test_delete_event(default_client: httpx.AsyncClient, mock_event: Event, access_token: str) -> None:
+    test_response = {
+        "message": "Event deleted successfully."
+    }
+
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {access_token}"
+    }
+
+    url = f"/event/{mock_event.id}"
+
+    response = await default_client.delete(url, headers=headers)
+
+    assert response.status_code == 200
+    assert response.json() == test_response
+
+
+@pytest.mark.asyncio
+async def test_get_event_again(default_client: httpx.AsyncClient, mock_event: Event) -> None:
+    url = f"/event/{str(mock_event.id)}"
+    response = await default_client.get(url)
+
+    assert response.status_code == 404
+