pdate

tests/e2e/game/index.py

@@ -232,6 +232,8 @@ def test_prestige_resets_farm(alice):
 
     reset_farm("alice_test")
     set_farm_xp("alice_test", economy.xp_threshold(economy.PRESTIGE_MIN_LEVEL))
+    fee = economy.refactor_cost(0, 100000)
+    carried = economy.refactor_carryover(100000, fee, 0)
     open_game(page)
     prestige = page.locator("form[data-game-action='prestige'] button")
     prestige.wait_for(state="visible")
@@ -240,10 +242,49 @@ def test_prestige_resets_farm(alice):
     confirm_btn.wait_for(state="visible")
     confirm_btn.click()
     expect(page.locator("[data-hud-prestige]")).to_have_text("1")
-    expect(page.locator("[data-hud-coins]")).to_have_text("50")
+    expect(page.locator("[data-hud-coins]")).to_have_text(
+        str(economy.STARTING_COINS + carried)
+    )
     assert page.locator("form[data-game-action='prestige']").count() == 0
 
 
+def test_prestige_unaffordable_shows_cost_without_button(alice):
+    page, _ = alice
+    from devplacepy.services.game import economy
+
+    reset_farm("alice_test", coins=100)
+    set_farm_xp("alice_test", economy.xp_threshold(economy.PRESTIGE_MIN_LEVEL))
+    open_game(page)
+    assert page.locator("form[data-game-action='prestige']").count() == 0
+    assert page.is_visible("text=keep farming to afford it")
+
+
+def test_grant_section_claim(alice):
+    page, _ = alice
+    from devplacepy.database import get_table
+    from devplacepy.services.game import economy, store
+    from devplacepy.services.game.store import _iso_week, _now
+
+    reset_farm("alice_test", coins=100)
+    store.ensure_treasury()
+    get_table("game_treasury").update(
+        {"uid": "treasury-main", "balance": 100000}, ["uid"]
+    )
+    set_farm(
+        "alice_test",
+        harvests_week=economy.GRANT_MIN_WEEK_HARVESTS,
+        harvests_week_start=_iso_week(_now()),
+    )
+    open_game(page)
+    claim = page.locator("form[data-game-action='grant'] button")
+    claim.wait_for(state="visible")
+    claim.click()
+    expect(page.locator("[data-hud-coins]")).to_have_text(
+        str(100 + economy.GRANT_CAP)
+    )
+    assert page.locator("form[data-game-action='grant']").count() == 0
+
+
 def test_leaderboard_panel_populates(alice):
     page, _ = alice
     reset_farm("alice_test")