diff --git a/tests/test_main.py b/tests/test_main.py index a28c9f4..b89730d 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -1,6 +1,7 @@ import validators from fastapi.testclient import TestClient from datetime import date +from typing import Union from src.main import app @@ -71,4 +72,17 @@ def test_retrieve_plays(): 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 \ No newline at end of file + assert type(returned_play.location) == str + +def test_retrieve_players(): + response = client.get("/players?play_id=1") + assert response.status_code == 200 + + returned_player = play_classes.PlayPlayer(**response.json()[0]) + + assert type(returned_player.name) == str + assert type(returned_player.username) == str + assert type(returned_player.score) == float or returned_player.score == None + assert type(returned_player.first_play) == bool + assert type(returned_player.has_won) == bool + assert type(returned_player.play_id) == int \ No newline at end of file