From 8b6c0d89b7c264919ab659613e4353fd9699f330 Mon Sep 17 00:00:00 2001 From: louis Date: Sat, 21 Oct 2023 16:39:41 +0200 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20Fix=20loading=20Attachments=20fo?= =?UTF-8?q?r=20Messages?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- internal/api/timeline.go | 2 +- internal/api/timeline_test.go | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/internal/api/timeline.go b/internal/api/timeline.go index b40293dd..adac911c 100644 --- a/internal/api/timeline.go +++ b/internal/api/timeline.go @@ -37,7 +37,7 @@ func (h *handler) GetTimeline(c *gin.Context) { messages := make([]storage.Message, 0) if ticker.Active { pagination := pagination.NewPagination(c) - messages, err = h.storage.FindMessagesByTickerAndPagination(ticker, *pagination) + messages, err = h.storage.FindMessagesByTickerAndPagination(ticker, *pagination, storage.WithAttachments()) if err != nil { c.JSON(http.StatusOK, response.ErrorResponse(response.CodeDefault, response.MessageFetchError)) return diff --git a/internal/api/timeline_test.go b/internal/api/timeline_test.go index 1337f958..7ef071be 100644 --- a/internal/api/timeline_test.go +++ b/internal/api/timeline_test.go @@ -74,7 +74,7 @@ func TestGetTimelineMessageFetchError(t *testing.T) { c, _ := gin.CreateTestContext(w) c.Set("ticker", storage.Ticker{Active: true}) s := &storage.MockStorage{} - s.On("FindMessagesByTickerAndPagination", mock.Anything, mock.Anything).Return([]storage.Message{}, errors.New("storage error")) + s.On("FindMessagesByTickerAndPagination", mock.Anything, mock.Anything, mock.Anything).Return([]storage.Message{}, errors.New("storage error")) h := handler{ storage: s, @@ -92,7 +92,7 @@ func TestGetTimeline(t *testing.T) { c, _ := gin.CreateTestContext(w) c.Set("ticker", storage.Ticker{}) s := &storage.MockStorage{} - s.On("FindMessagesByTickerAndPagination", mock.Anything, mock.Anything).Return([]storage.Message{}, nil) + s.On("FindMessagesByTickerAndPagination", mock.Anything, mock.Anything, mock.Anything).Return([]storage.Message{}, nil) h := handler{ storage: s,