From 724a40c0a6598ffe530092625fd4577bf60b0089 Mon Sep 17 00:00:00 2001 From: Yarne Coppens Date: Thu, 1 Aug 2024 14:44:24 +0200 Subject: [PATCH] Test now checks properly for collection board game --- test_main.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) 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