From 86c3f54d03991539ada0382f1c1277642fead897 Mon Sep 17 00:00:00 2001 From: Yarne Coppens Date: Fri, 2 Aug 2024 10:54:41 +0200 Subject: [PATCH] Created wishlist test --- tests/test_main.py | 29 ++++++++++++++++++++++------- 1 file changed, 22 insertions(+), 7 deletions(-) diff --git a/tests/test_main.py b/tests/test_main.py index 1ce2920..bd6bec2 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -7,6 +7,12 @@ from src.classes import boardgame_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 + + + def test_read_main(): response = client.get("/") assert response.status_code == 200 @@ -18,17 +24,26 @@ def test_retrieve_boardgame(): returned_boardgame = boardgame_classes.BoardGame(**response.json()) - assert type(returned_boardgame.id) == int - assert type(returned_boardgame.name) == str + default_boardgame_test(returned_boardgame) -def test_retrieve_collection(): - response = client.get("/collection") +def test_retrieve_owned(): + response = client.get("/owned") assert response.status_code == 200 returned_boardgame = boardgame_classes.OwnedBoardGame(**response.json()[0]) - assert type(returned_boardgame.id) == int - assert type(returned_boardgame.name) == str + 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 \ No newline at end of file + 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 \ No newline at end of file