X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/48d7bce28b18ecef4c52c98821675605464cecb9..c781f79f3c8cb9a7577ee1574d196988da2719b3:/telegram_image_test.go?ds=sidebyside diff --git a/telegram_image_test.go b/telegram_image_test.go index 7f7b7c0..94a014b 100644 --- a/telegram_image_test.go +++ b/telegram_image_test.go @@ -50,6 +50,7 @@ func TestTelegramSelectsLargestPhotoAtMost1568Pixels(t *testing.T) { UpdateID: 1, Message: message, })) + test.events.WaitFor(t, "idle", 1) files, err := os.ReadDir(test.agent.Images().dir) @@ -163,6 +164,7 @@ func TestTelegramImageDocumentsUseContentFormatAndRejectOtherMedia(t *testing.T) failure: "exceeds 20 MB", }, } + for _, tc := range testCases { t.Run(tc.name, func(t *testing.T) { text, err := telegramImageText(t.Context(), api, store, &TelegramMessage{Document: &tc.document}) @@ -193,6 +195,7 @@ func TestTelegramRetriesImageDownloadBeforeAdvancingOffset(t *testing.T) { return 0, "" } + bot, logs := telegramStartTestBot(t, telegram) server := newModelFake(t) socket := daemonStartTestWithBot(t, server, bot) @@ -205,15 +208,18 @@ func TestTelegramRetriesImageDownloadBeforeAdvancingOffset(t *testing.T) { Width: 2, Height: 2, }} + telegram.Add(TelegramUpdate{ UpdateID: 7, Message: message, }) + terminal.WaitFor(t, "[done]") require.Eventually(t, func() bool { offset, err := telegramLoadOffset(bot.offsetPath) return err == nil && offset == 8 }, 10*time.Second, time.Millisecond) + require.Contains(t, logs.String(), "receive failed") require.Contains(t, logs.String(), "receive back") require.NotContains(t, logs.String(), telegramTestToken) @@ -245,6 +251,7 @@ func TestTelegramStorageFailureDoesNotAcceptTheImage(t *testing.T) { Width: 2, Height: 2, }} + err := bot.handle(t.Context(), test.loop, nil, TelegramUpdate{ UpdateID: 1, Message: message,