Separate multiline Go statements

[onechat] / prompt_test.go

diff --git a/prompt_test.go b/prompt_test.go
index 06fea49cf84315d734120c5d23363113b70fa4ed..158784b9a34d8e0e6d5c2e8198346d4d9d0a8bac 100644 (file)
--- a/prompt_test.go
+++ b/prompt_test.go
@@ -65,6 +65,7 @@ func TestPromptEndsWithTheInstructions(t *testing.T) {
        require.Contains(t, text, `Do the user's tasks yourself, with your tools, following the user's instructions
 at the end of this prompt: who they are, how their files are organized and how
 they want work done.`)
+
        require.True(t, strings.HasSuffix(text, "\n\n# The user's instructions\n\nI am Lena. Keep replies short."))
 }
 
@@ -166,3 +167,14 @@ func TestRootPromptDefinesChildRoutingAndConsent(t *testing.T) {
        require.NotContains(t, child, "Use agent_create")
        require.Contains(t, child, "Send nothing to WhatsApp. Draft replies in your report instead")
 }
+
+func TestRootPromptKeepsImageConsentAndChildForwardingRules(t *testing.T) {
+       text := prompt("Ada", "/AGENT/instructions.md", "")
+       require.Contains(t, text, "Use image_show or whatsapp_send with an image only on the user's request.")
+       require.Contains(t, text, "Pass an image to a child with an [image FILE] line in tell.")
+
+       child := promptChild("House", "Ada", "Keep house plans.", "/CHILD/instructions.md", "")
+       require.Contains(t, child, "Send nothing to WhatsApp. Draft replies in your report instead")
+       require.NotContains(t, promptToolNames(promptSubagentTools()), "image_show")
+       require.NotContains(t, promptToolNames(promptSubagentTools()), "whatsapp_send")
+}