diff --git a/test_main.py b/test_main.py index 4d41a22..92e0e92 100644 --- a/test_main.py +++ b/test_main.py @@ -1,7 +1,8 @@ from fastapi.testclient import TestClient +from datetime import date from .main import app -from classes.boardgame import BoardGame +from classes.boardgame import BoardGame, CollectionBoardGame client = TestClient(app) @@ -23,7 +24,10 @@ def test_retrieve_collection(): response = client.get("/collection") assert response.status_code == 200 - returned_boardgame = BoardGame(**response.json()[0]) + returned_boardgame = CollectionBoardGame(**response.json()[0]) assert type(returned_boardgame.id) == int - assert type(returned_boardgame.name) == str \ No newline at end of file + 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 \ No newline at end of file