bgg_api/tests/test_main.py

from fastapi.testclient import TestClient
from datetime import date

from src.main import app

from src.classes import boardgame_classes

client = TestClient(app)

def test_read_main():
    response = client.get("/")
    assert response.status_code == 200
    assert response.json() == {"Hello": "World"}

def test_retrieve_boardgame():
    response = client.get("/boardgame/373167")
    assert response.status_code == 200

    returned_boardgame = boardgame_classes.BoardGame(**response.json())

    assert type(returned_boardgame.id) == int
    assert type(returned_boardgame.name) == str

def test_retrieve_collection():
    response = client.get("/collection")
    assert response.status_code == 200

    returned_boardgame = boardgame_classes.CollectionBoardGame(**response.json()[0])

    assert type(returned_boardgame.id) == int
    assert type(returned_boardgame.name) == str
    assert type(returned_boardgame.price_paid) == float
    assert type(returned_boardgame.acquisition_date) == date
    assert type(returned_boardgame.acquired_from) == str