]> git.giacomolli.uk - onechat/blobdiff - telegram_image_test.go
Separate multiline Go statements
[onechat] / telegram_image_test.go
index 7f7b7c008ef979089e543b52fa05281f2b19dc84..94a014b9fdfbd6c942277c9a8d67311e9283afcd 100644 (file)
@@ -50,6 +50,7 @@ func TestTelegramSelectsLargestPhotoAtMost1568Pixels(t *testing.T) {
                UpdateID: 1,
                Message:  message,
        }))
                UpdateID: 1,
                Message:  message,
        }))
+
        test.events.WaitFor(t, "idle", 1)
 
        files, err := os.ReadDir(test.agent.Images().dir)
        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",
                },
        }
                        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})
        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, ""
        }
 
                return 0, ""
        }
+
        bot, logs := telegramStartTestBot(t, telegram)
        server := newModelFake(t)
        socket := daemonStartTestWithBot(t, server, bot)
        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,
        }}
                Width:  2,
                Height: 2,
        }}
+
        telegram.Add(TelegramUpdate{
                UpdateID: 7,
                Message:  message,
        })
        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)
        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)
        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,
        }}
                Width:  2,
                Height: 2,
        }}
+
        err := bot.handle(t.Context(), test.loop, nil, TelegramUpdate{
                UpdateID: 1,
                Message:  message,
        err := bot.handle(t.Context(), test.loop, nil, TelegramUpdate{
                UpdateID: 1,
                Message:  message,