]> git.giacomolli.uk - onechat/blobdiff - image_show_test.go
Require consent and review for autosquash
[onechat] / image_show_test.go
index 65e422c40bb328c954be8e578ccdced36c9fd48e..9317cbb896ad4d29126c49c6db69e760f4f2bae8 100644 (file)
@@ -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))