snapshot of https://github.com/mattermost/mattermost.git (master, 2026-09-16; blobs > 131072B omitted)

server/channels/store/retrylayer/retrylayer_test.go

@@ -0,0 +1,130 @@
+// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
+// See LICENSE.txt for license information.
+
+package retrylayer
+
+import (
+	"testing"
+
+	"github.com/lib/pq"
+	"github.com/pkg/errors"
+
+	"github.com/lib/pq/pqerror"
+	"github.com/mattermost/mattermost/server/public/model"
+	"github.com/mattermost/mattermost/server/v8/channels/store/storetest/mocks"
+)
+
+func genStore() *mocks.Store {
+	mock := &mocks.Store{}
+	mock.On("Audit").Return(&mocks.AuditStore{})
+	mock.On("Bot").Return(&mocks.BotStore{})
+	mock.On("Channel").Return(&mocks.ChannelStore{})
+	mock.On("ChannelGuard").Return(&mocks.ChannelGuardStore{})
+	mock.On("ChannelMemberHistory").Return(&mocks.ChannelMemberHistoryStore{})
+	mock.On("ChannelBookmark").Return(&mocks.ChannelBookmarkStore{})
+	mock.On("ClusterDiscovery").Return(&mocks.ClusterDiscoveryStore{})
+	mock.On("RemoteCluster").Return(&mocks.RemoteClusterStore{})
+	mock.On("Command").Return(&mocks.CommandStore{})
+	mock.On("CommandWebhook").Return(&mocks.CommandWebhookStore{})
+	mock.On("Compliance").Return(&mocks.ComplianceStore{})
+	mock.On("Emoji").Return(&mocks.EmojiStore{})
+	mock.On("FileInfo").Return(&mocks.FileInfoStore{})
+	mock.On("UploadSession").Return(&mocks.UploadSessionStore{})
+	mock.On("Group").Return(&mocks.GroupStore{})
+	mock.On("Job").Return(&mocks.JobStore{})
+	mock.On("License").Return(&mocks.LicenseStore{})
+	mock.On("LinkMetadata").Return(&mocks.LinkMetadataStore{})
+	mock.On("SharedChannel").Return(&mocks.SharedChannelStore{})
+	mock.On("OAuth").Return(&mocks.OAuthStore{})
+	mock.On("OutgoingOAuthConnection").Return(&mocks.OutgoingOAuthConnectionStore{})
+	mock.On("Plugin").Return(&mocks.PluginStore{})
+	mock.On("Post").Return(&mocks.PostStore{})
+	mock.On("Thread").Return(&mocks.ThreadStore{})
+	mock.On("Preference").Return(&mocks.PreferenceStore{})
+	mock.On("ProductNotices").Return(&mocks.ProductNoticesStore{})
+	mock.On("Reaction").Return(&mocks.ReactionStore{})
+	mock.On("RetentionPolicy").Return(&mocks.RetentionPolicyStore{})
+	mock.On("Role").Return(&mocks.RoleStore{})
+	mock.On("Scheme").Return(&mocks.SchemeStore{})
+	mock.On("Session").Return(&mocks.SessionStore{})
+	mock.On("SessionAttribute").Return(&mocks.SessionAttributeStore{})
+	mock.On("Status").Return(&mocks.StatusStore{})
+	mock.On("System").Return(&mocks.SystemStore{})
+	mock.On("Team").Return(&mocks.TeamStore{})
+	mock.On("TermsOfService").Return(&mocks.TermsOfServiceStore{})
+	mock.On("Token").Return(&mocks.TokenStore{})
+	mock.On("User").Return(&mocks.UserStore{})
+	mock.On("UserAccessToken").Return(&mocks.UserAccessTokenStore{})
+	mock.On("UserTermsOfService").Return(&mocks.UserTermsOfServiceStore{})
+	mock.On("Webhook").Return(&mocks.WebhookStore{})
+	mock.On("NotifyAdmin").Return(&mocks.NotifyAdminStore{})
+	mock.On("Draft").Return(&mocks.DraftStore{})
+	mock.On("PostPriority").Return(&mocks.PostPriorityStore{})
+	mock.On("PostAcknowledgement").Return(&mocks.PostAcknowledgementStore{})
+	mock.On("PostPersistentNotification").Return(&mocks.PostPersistentNotificationStore{})
+	mock.On("DesktopTokens").Return(&mocks.DesktopTokensStore{})
+	mock.On("ChannelBookmark").Return(&mocks.ChannelBookmarkStore{})
+	mock.On("ScheduledPost").Return(&mocks.ScheduledPostStore{})
+	mock.On("PropertyField").Return(&mocks.PropertyFieldStore{})
+	mock.On("PropertyGroup").Return(&mocks.PropertyGroupStore{})
+	mock.On("PropertyValue").Return(&mocks.PropertyValueStore{})
+	mock.On("AccessControlPolicy").Return(&mocks.AccessControlPolicyStore{})
+	mock.On("Attributes").Return(&mocks.AttributesStore{})
+	mock.On("AutoTranslation").Return(&mocks.AutoTranslationStore{})
+	mock.On("ContentFlagging").Return(&mocks.ContentFlaggingStore{})
+	mock.On("DeliveryTracking").Return(&mocks.DeliveryTrackingStore{})
+	mock.On("ReadReceipt").Return(&mocks.ReadReceiptStore{})
+	mock.On("Recap").Return(&mocks.RecapStore{})
+	mock.On("ScheduledRecap").Return(&mocks.ScheduledRecapStore{})
+	mock.On("TemporaryPost").Return(&mocks.TemporaryPostStore{})
+	mock.On("View").Return(&mocks.ViewStore{})
+	mock.On("ChannelJoinRequest").Return(&mocks.ChannelJoinRequestStore{})
+	return mock
+}
+
+func TestRetry(t *testing.T) {
+	t.Run("on regular error should not retry", func(t *testing.T) {
+		mock := genStore()
+		mockBotStore := mock.Bot().(*mocks.BotStore)
+		mockBotStore.On("Get", "test", false).Return(nil, errors.New("regular error")).Times(1)
+		mock.On("Bot").Return(&mockBotStore)
+		layer := New(mock)
+		layer.Bot().Get("test", false)
+		mockBotStore.AssertExpectations(t)
+	})
+	t.Run("on success should not retry", func(t *testing.T) {
+		mock := genStore()
+		mockBotStore := mock.Bot().(*mocks.BotStore)
+		mockBotStore.On("Get", "test", false).Return(&model.Bot{}, nil).Times(1)
+		mock.On("Bot").Return(&mockBotStore)
+		layer := New(mock)
+		layer.Bot().Get("test", false)
+		mockBotStore.AssertExpectations(t)
+	})
+
+	t.Run("on postgres repeatable error should retry", func(t *testing.T) {
+		for _, errCode := range []string{"40001", "40P01"} {
+			t.Run("error "+errCode, func(t *testing.T) {
+				mock := genStore()
+				mockBotStore := mock.Bot().(*mocks.BotStore)
+				pqErr := pq.Error{Code: pqerror.Code(errCode)}
+				mockBotStore.On("Get", "test", false).Return(nil, errors.Wrap(&pqErr, "test-error")).Times(3)
+				mock.On("Bot").Return(&mockBotStore)
+				layer := New(mock)
+				layer.Bot().Get("test", false)
+				mockBotStore.AssertExpectations(t)
+			})
+		}
+	})
+
+	t.Run("on postgres not repeatable error should not retry", func(t *testing.T) {
+		mock := genStore()
+		mockBotStore := mock.Bot().(*mocks.BotStore)
+		pqErr := pq.Error{Code: "20000"}
+		mockBotStore.On("Get", "test", false).Return(nil, errors.Wrap(&pqErr, "test-error")).Times(1)
+		mock.On("Bot").Return(&mockBotStore)
+		layer := New(mock)
+		layer.Bot().Get("test", false)
+		mockBotStore.AssertExpectations(t)
+	})
+}