bgg_api/tests/test_main.py

import validators
from fastapi.testclient import TestClient
from datetime import date

from src.main import app

from src.classes import boardgame_classes, play_classes

client = TestClient(app)

def default_boardgame_test(to_test_boardgame: boardgame_classes.BoardGame):
    assert type(to_test_boardgame.id) == int
    assert type(to_test_boardgame.name) == str
    assert type(to_test_boardgame.weight) == float
    assert type(to_test_boardgame.description) == str
    assert validators.url(str(to_test_boardgame.image_url))
    assert validators.url(str(to_test_boardgame.thumbnail_url))
    assert type(to_test_boardgame.year_published) == int
    assert type(to_test_boardgame.min_players) == int
    assert type(to_test_boardgame.max_players) == int
    assert type(to_test_boardgame.min_playing_time) == int
    assert type(to_test_boardgame.max_playing_time) == int
    assert type(to_test_boardgame.min_age) == int
    assert type(to_test_boardgame.type) == boardgame_classes.BoardgameType



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?id=373167")
    assert response.status_code == 200

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

    default_boardgame_test(returned_boardgame)

def test_retrieve_owned():
    response = client.get("/owned")
    assert response.status_code == 200

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

    default_boardgame_test(returned_boardgame)
    assert type(returned_boardgame.price_paid) == float
    assert type(returned_boardgame.acquisition_date) == date
    assert type(returned_boardgame.acquired_from) == str


def test_retrieve_wishlist():
    response = client.get("/wishlist")
    assert response.status_code == 200

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

    default_boardgame_test(returned_boardgame)
    assert type(returned_boardgame.wishlist_priority) == int
    assert returned_boardgame.wishlist_priority > 0


def test_retrieve_plays():
    response = client.get("/plays")
    assert response.status_code == 200

    returned_play = play_classes.Play(**response.json()[0])

    assert type(returned_play.boardgame_id) == int
    assert type(returned_play.play_date) == date
    assert type(returned_play.duration) == int
    assert type(returned_play.ignore_for_stats) == bool
    assert type(returned_play.location) == str