X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/b1be0ee8530df5db8973bfe355f2f6d3149383a0..HEAD:/image_show_test.go?ds=sidebyside

diff --git a/image_show_test.go b/image_show_test.go
index 65e422c..9317cbb 100644
--- a/image_show_test.go
+++ b/image_show_test.go
@@ -22,6 +22,7 @@ func TestImageShowUsesStoredImageForTelegramAndTerminal(t *testing.T) {
 		store: store,
 		bot:   bot,
 	}}
+
 	args, err := json.Marshal(ToolImageShowArguments{
 		Image:   name,
 		Caption: "A red point.",
@@ -38,7 +39,7 @@ func TestImageShowUsesStoredImageForTelegramAndTerminal(t *testing.T) {
 	require.Equal(t, "A red point.", calls[0].params["caption"])
 	require.Equal(t, name, calls[0].params["photo_name"])
 	require.Equal(t, picture, calls[0].params["photo"])
-	require.Empty(t, logs.String())
+	require.Equal(t, "<6>image sent\n", logs.String())
 }
 
 func TestImageShowRefusesUnknownNamesBeforeSending(t *testing.T) {
@@ -79,6 +80,7 @@ func TestImageShowFallsBackOnlyWhenTelegramRefusesPhoto(t *testing.T) {
 			failure: true,
 		},
 	}
+
 	for _, tc := range testCases {
 		t.Run(tc.name, func(t *testing.T) {
 			telegram := newTelegramFake(t)
@@ -89,6 +91,7 @@ func TestImageShowFallsBackOnlyWhenTelegramRefusesPhoto(t *testing.T) {
 
 				return 0, ""
 			}
+
 			store := &ImageStore{dir: t.TempDir()}
 			picture := imageTestPNG(t, 3, 3)
 			name, err := store.Save(picture)
@@ -99,6 +102,7 @@ func TestImageShowFallsBackOnlyWhenTelegramRefusesPhoto(t *testing.T) {
 				store: store,
 				bot:   bot,
 			}
+
 			_, err = show.Show(t.Context(), name, "Caption")
 			if tc.failure {
 				require.Error(t, err)
@@ -194,6 +198,7 @@ func TestAgentShowsTelegramReceivedImageToItsOwner(t *testing.T) {
 
 		modelWriteEvents(w, modelCompletedEvent([]any{modelMessageItem("Shown.")}, 100, 0, 5))
 	}
+
 	socket := daemonStartTestWithBot(t, server, bot)
 	terminal := terminalStartTestClient(t, socket)
 	terminal.WaitFor(t, "[connected to OneChat]")
@@ -205,10 +210,12 @@ func TestAgentShowsTelegramReceivedImageToItsOwner(t *testing.T) {
 		Width:  6,
 		Height: 4,
 	}}
+
 	telegram.Add(TelegramUpdate{
 		UpdateID: 1,
 		Message:  message,
 	})
+
 	terminal.WaitFor(t, "Shown.\n[done]")
 	require.Contains(t, terminal.Output(), "[result] [image "+name+"]")
 	require.Contains(t, terminal.Output(), filepath.Join(imageDirName, name))