from typing import Mapping import httpx from fastapi.testclient import TestClient from app import app client = TestClient(app) def test_add_and_retrieve_player(test_client: TestClient, cleanups): response = _json_get(test_client, "/player/1") assert response.status_code == 404 create_player_response = _json_post(client, "/player", {"name": "jason"}) assert create_player_response.status_code == 201 response_1 = _json_get(client, "/player/1") assert response_1.json()["name"] == "jason" def cleanup(): delete_response = _json_delete(client, "/player/1") assert delete_response.status_code == 204 cleanups.add_success(cleanup) def test_add_and_retrieve_deck(test_client: TestClient, cleanups): not_found_response = _json_get(test_client, "/deck/1") assert not_found_response.status_code == 404 # Try (and fail) to create a deck owned by a non-existent player invalid_owner_response = _json_post( test_client, "/deck", {"name": "Baby's First Deck", "owner_id": 1} ) assert invalid_owner_response.status_code == 400 assert invalid_owner_response.json()["detail"] == "Owner id 1 not found" create_jim_response = _json_post(test_client, "/player", {"name": "jim"}) assert create_jim_response.status_code == 201 jim_id = create_jim_response.json()["id"] create_deck_response = _json_post( test_client, "/deck", {"name": "Baby's First Deck", "owner_id": str(jim_id)} ) assert create_deck_response.status_code == 201 # _Should_ always be 1, since we expect to start with an empty database, but why risk it? deck_id = create_deck_response.json()["id"] get_deck_response = _json_get(test_client, f"/deck/{deck_id}") assert get_deck_response.status_code == 200 assert get_deck_response.json()["name"] == "Baby's First Deck" # Very basic HTML testing html_response = test_client.get(f"/deck/{deck_id}") assert """owned by jim""" in html_response.text def success_cleanup(): delete_response = _json_delete(test_client, f"/deck/{deck_id}") assert delete_response.status_code == 204 cleanups.add_success(success_cleanup) def test_incremental_add_of_games(test_client: TestClient, cleanups): latest_deck_response = _json_get(test_client, "/game/latest_game") assert latest_deck_response.status_code == 404 # https://github.com/tiangolo/fastapi/issues/1536#issuecomment-640781718 with open("seed-data/all-in-one.csv", "rb") as f: test_client.post( "/api/seed/all_in_one", files={"file": ("fake_all_in_one_filename.csv", f, "text/csv")}, ) latest_deck_response = _json_get(test_client, "/game/latest_game") assert latest_deck_response.status_code == 200 print(latest_deck_response.json()) assert latest_deck_response.json()["date"] == "2024-07-05T00:00:00" # then seed again, and check that it successfully gets the expected latest with open( "seed-data/all-in-one-updated-for-incremental-add-testing.csv", "rb" ) as f: test_client.post( "/api/seed/all_in_one", files={"file": ("fake_all_in_one_filename.csv", f, "text/csv")}, ) latest_deck_response = _json_get(test_client, "/game/latest_game") assert latest_deck_response.status_code == 200 assert latest_deck_response.json()["date"] == "2024-07-25T00:00:00" def _json_get(c: TestClient, path: str) -> httpx.Response: return c.get(f"/api{path}", headers={"Content-Type": "application/json"}) def _json_post(c: TestClient, path: str, body: Mapping) -> httpx.Response: return c.post( f"/api{path}", headers={"Content-Type": "application/json"}, json=body ) def _json_delete(c: TestClient, path: str) -> httpx.Response: return c.delete(f"/api{path}", headers={"Content-Type": "application/json"})