Fixed boardgame retrieval bug

src/modules/data_connection.py

@@ -19,6 +19,7 @@ def get_boardgame(session: Session, boardgame_type: SQLModel, boardgame_id: int)
     else:
         to_return_boardgame = bgg_connection.get_boardgame(boardgame_id)
         db_connection.add_boardgame(session, to_return_boardgame)
+        to_return_boardgame = db_connection.get_boardgame(session, boardgame_type, boardgame_id)
 
     return to_return_boardgame
 

src/modules/db_connection.py

@@ -51,11 +51,14 @@ def get_boardgame(session: Session, boardgame_type: SQLModel, boardgame_id: int)
     statement = select(boardgame_type).where(boardgame_type.id == boardgame_id)
     results = session.exec(statement)
 
-    if len(results.all()) == 0:
+    returned_boardgames = results.all()
+
+    if len(returned_boardgames) == 0:
         boardgame = None
     else:
-        boardgame = results.all()[0]
+        boardgame = returned_boardgames[0]
 
+    print(boardgame)
     return boardgame
     
 

tests/test_main.py

@@ -78,7 +78,7 @@ def test_retrieve_players():
     response = client.get("/players?play_id=1")
     assert response.status_code == 200
 
-    returned_player = play_classes.PlayPlayerPublicWithPlay.model_validate(response.json()[0])
+    returned_player = play_classes.PlayPlayerPublic.model_validate(response.json()[0])
 
     assert type(returned_player.name) == str
     assert type(returned_player.username) == str