78 lines
3.0 KiB
Python
78 lines
3.0 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
|
|
from tft.api import app as app_mod
|
|
from tft.constants.loader import load_constants
|
|
from tft.model import artifact as artifact_mod
|
|
from tests.test_sim import make_static
|
|
|
|
|
|
@pytest.fixture
|
|
def client(monkeypatch):
|
|
static = make_static()
|
|
static["meta"]["patch"] = "test"
|
|
art = artifact_mod.build(static)
|
|
cfg = load_constants(17)
|
|
monkeypatch.setattr(app_mod, "_load_game_deps", lambda: (art, cfg))
|
|
return TestClient(app_mod.app)
|
|
|
|
|
|
def test_full_game_via_api(client):
|
|
state = client.post("/api/game", params={"seed": 3}).json()
|
|
game_id = state["game_id"]
|
|
assert state["round"]["label"] == "1-1"
|
|
assert state["player"]["hp"] == 100
|
|
assert len(state["opponents"]) == 7
|
|
|
|
steps = 0
|
|
while not state["over"]:
|
|
if state["augment_offer"]:
|
|
client.post(f"/api/game/{game_id}/action",
|
|
json={"action": "pick_augment", "choice": 0})
|
|
affordable = [i for i, s in enumerate(state["shop"])
|
|
if s and s["cost"] <= state["player"]["gold"]]
|
|
if affordable:
|
|
client.post(f"/api/game/{game_id}/action",
|
|
json={"action": "buy", "slot": affordable[0]})
|
|
resp = client.post(f"/api/game/{game_id}/action", json={"action": "next"})
|
|
assert resp.status_code == 200
|
|
state = resp.json()
|
|
steps += 1
|
|
assert steps < 100
|
|
assert 1 <= state["placement"] <= 8
|
|
|
|
|
|
def test_invalid_action_is_400(client):
|
|
game_id = client.post("/api/game", params={"seed": 1}).json()["game_id"]
|
|
resp = client.post(f"/api/game/{game_id}/action",
|
|
json={"action": "sell", "where": "bench", "idx": 99})
|
|
assert resp.status_code == 400
|
|
resp = client.post(f"/api/game/{game_id}/action", json={"action": "dance"})
|
|
assert resp.status_code == 400
|
|
assert client.post("/api/game/nope/action", json={"action": "next"}).status_code == 404
|
|
|
|
|
|
def test_state_shape(client):
|
|
state = client.post("/api/game", params={"seed": 5}).json()
|
|
for key in ("round", "player", "board", "bench", "shop", "traits",
|
|
"opponents", "items", "over", "log"):
|
|
assert key in state
|
|
assert state["player"]["score"] == 0.0
|
|
assert all(o["board"] is not None for o in state["opponents"])
|
|
|
|
|
|
def test_upgrade_hint():
|
|
from tft.api.app import upgrade_hint
|
|
from tft.sim.game import Game
|
|
static = make_static()
|
|
static["meta"]["patch"] = "t"
|
|
game = Game(artifact_mod.build(static), load_constants(17), seed=1)
|
|
assert upgrade_hint(game, "U1_0") == 0
|
|
game.bench = [{"api_name": "U1_0", "stars": 1, "items": []} for _ in range(2)]
|
|
assert upgrade_hint(game, "U1_0") == 2
|
|
game.bench = [{"api_name": "U1_0", "stars": 2, "items": []} for _ in range(2)] + \
|
|
[{"api_name": "U1_0", "stars": 1, "items": []} for _ in range(2)]
|
|
assert upgrade_hint(game, "U1_0") == 3
|
|
game.bench = [{"api_name": "U1_0", "stars": 3, "items": []}]
|
|
assert upgrade_hint(game, "U1_0") == 0
|