from services.rpg_state import affinity_prompt_block, stats_prompt_block


def test_affinity_mandatory_wording():
    block = affinity_prompt_block(5)
    assert "MANDATORY" in block
    assert "current player" in block


def test_stamina_2_exhausted_instruction():
    block = stats_prompt_block({"lust": 0, "stamina": 2, "tension": 0})
    assert "barely moves" in block.lower() or "exhausted" in block.lower()


def test_stamina_1_collapse_instruction():
    block = stats_prompt_block({"lust": 0, "stamina": 1, "tension": 0})
    assert "collapse" in block.lower() or "pass out" in block.lower()
