bgg_api/test_main.py

from fastapi.testclient import TestClient

from .main import app
from classes.boardgame import BoardGame

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(**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(**response.json()[0])

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