from ai_reviewer.prompt import build_prompt


def test_prompt_includes_diff_chunk_and_schema():
    chunk = "diff --git a/x.py b/x.py\n@@ -1 +1 @@\n-print(1)\n+print(2)\n"
    prompt = build_prompt(chunk)

    assert "Output JSON schema" in prompt
    assert "Diff chunk:" in prompt
    assert chunk.strip() in prompt