From ee5c042c9e3abbceb60417fe1ef71b585a25e9d2 Mon Sep 17 00:00:00 2001 From: "didericis (claude)" Date: Mon, 29 Jun 2026 12:55:25 -0400 Subject: [PATCH] feat: headless_prompt method on AgentProvider + --prompt arg --- tests/unit/test_contrib_pi_provider.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/tests/unit/test_contrib_pi_provider.py b/tests/unit/test_contrib_pi_provider.py index 0abe959..5f3c0a1 100644 --- a/tests/unit/test_contrib_pi_provider.py +++ b/tests/unit/test_contrib_pi_provider.py @@ -223,5 +223,14 @@ class TestPiDockerfile(unittest.TestCase): self.assertIn("chmod 1777 /tmp /var/tmp", dockerfile) +class TestPiHeadlessPrompt(unittest.TestCase): + def test_returns_p_flag_and_prompt(self): + self.assertEqual(["-p", "Do the task"], PiAgentProvider().headless_prompt("Do the task")) + + def test_preserves_prompt_text_verbatim(self): + text = "Fix issue #42: the widget breaks on empty input" + self.assertEqual(["-p", text], PiAgentProvider().headless_prompt(text)) + + if __name__ == "__main__": unittest.main()