240db4e2a7
Some checks failed
Server CI / Postgres (shard 0) (push) Has been skipped
Server CI / Postgres (shard 3) (push) Has been skipped
BuildEnv Docker Image / build-image (push) Failing after 4s
BuildEnv Docker Image / build-image-fips (push) Failing after 4s
Migration Review & Release Notes / Generate migration review & release notes (push) Failing after 3s
Server CI / Build Environment (push) Failing after 3s
Server CI / Postgres (shard 1) (push) Has been skipped
Tools CI / check-style (mattermost-govet) (push) Failing after 4s
Server CI / Postgres (shard 2) (push) Has been skipped
Tools CI / check-style (mmgotool) (push) Failing after 3s
Server CI / Elasticsearch v8 Compatibility (push) Has been skipped
Server CI / OpenSearch v2 Compatibility (push) Has been skipped
Server CI / Postgres FIPS (shard 0) (push) Has been skipped
Tools CI / Test (mattermost-govet) (push) Failing after 3s
Server CI / Postgres FIPS (shard 1) (push) Has been skipped
Server CI / Postgres FIPS (shard 2) (push) Has been skipped
Tools CI / Test (mmgotool) (push) Failing after 3s
Server CI / Postgres FIPS (shard 3) (push) Has been skipped
Server CI / Run mmctl tests (push) Has been skipped
Server CI / Run mmctl tests (FIPS) (push) Has been skipped
Server CI / Merge Postgres Test Results (push) Has been skipped
Server CI / Merge Postgres FIPS Test Results (push) Has been skipped
Server CI / Check backport migrations (push) Has been skipped
Server CI / Check generated files (push) Has been skipped
Server CI / check-style (push) Has been skipped
Server CI / Vet API (push) Has been skipped
Server CI / Build mattermost server app (push) Has been skipped
Server CI / Server CI Complete (push) Failing after 2s
Server CI / Server CI Artifacts (push) Has been skipped
Server CI / Sentry Upload (push) Has been skipped
Server CI / Server CI Report (push) Failing after 5m1s
Server CI Nightly Race / Race Detector (push) Has been skipped
CodeQL / Analyze (go) (push) Has been cancelled
CodeQL / Analyze (javascript) (push) Has been cancelled
Webapp CI Cache Warm / Warm Webapp node_modules Cache (push) Has been cancelled
E2E CI Cache Warm / Warm E2E npm Registry Cache (push) Has been cancelled
API / build (push) Has been cancelled
Web App CI / check-types (push) Has been cancelled
Web App CI / check-lint (push) Has been cancelled
Web App CI / check-i18n (push) Has been cancelled
Web App CI / check-external-links (push) Has been cancelled
Web App CI / test (platform) (push) Has been cancelled
Web App CI / test (mattermost-redux) (push) Has been cancelled
Web App CI / test (channels shard 1/4) (push) Has been cancelled
Web App CI / test (channels shard 2/4) (push) Has been cancelled
Web App CI / test (channels shard 3/4) (push) Has been cancelled
Web App CI / test (channels shard 4/4) (push) Has been cancelled
Web App CI / upload-coverage (push) Has been cancelled
Web App CI / build (push) Has been cancelled
YAML Lint / yamllint (push) Has been cancelled
Push mirrored docker images / cd/Push mirrored docker images (push) Has been cancelled
docs-cd / Build and deploy docs (push) Has been cancelled
Docs CI / build (push) Has been cancelled
E2E Cache Warm: Cypress deps / Warm Cypress deps cache (push) Has been cancelled
Scorecards supply-chain security / Scorecard analysis (push) Has been cancelled
Server CI Weekly / Postgres FIPS (push) Has been skipped
Server CI Weekly / Run mmctl tests (FIPS) (push) Has been skipped
Server CI Weekly / Postgres with binary parameters (push) Has been skipped
Server CI Weekly / Build Environment (push) Failing after 1s
Server CI Cache Warm / Warm Go Module and Build Cache (push) Failing after 2s
Server CI Nightly Race / Build Environment (push) Failing after 1s
539 lines
23 KiB
Go
539 lines
23 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package localcachelayer
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/plugin/plugintest/mock"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
"github.com/mattermost/mattermost/server/public/shared/request"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store/storetest"
|
|
"github.com/mattermost/mattermost/server/v8/channels/store/storetest/mocks"
|
|
cmocks "github.com/mattermost/mattermost/server/v8/platform/services/cache/mocks"
|
|
)
|
|
|
|
func TestChannelStore(t *testing.T) {
|
|
StoreTestWithSqlStore(t, storetest.TestReactionStore)
|
|
}
|
|
|
|
func TestChannelStoreChannelMemberCountsCache(t *testing.T) {
|
|
countResult := int64(10)
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
t.Run("first call not cached, second cached and returning same data", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
count, err := cachedStore.Channel().GetMemberCount("id", true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, count, countResult)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 1)
|
|
count, err = cachedStore.Channel().GetMemberCount("id", true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, count, countResult)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 1)
|
|
})
|
|
|
|
t.Run("first call not cached, second force not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetMemberCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 1)
|
|
cachedStore.Channel().GetMemberCount("id", false)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 2)
|
|
})
|
|
|
|
t.Run("first call force not cached, second not cached, third cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetMemberCount("id", false)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 1)
|
|
cachedStore.Channel().GetMemberCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 2)
|
|
cachedStore.Channel().GetMemberCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 2)
|
|
})
|
|
|
|
t.Run("first call with GetMemberCountFromCache not cached, second cached and returning same data", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
count := cachedStore.Channel().GetMemberCountFromCache("id")
|
|
assert.Equal(t, count, countResult)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 1)
|
|
count = cachedStore.Channel().GetMemberCountFromCache("id")
|
|
assert.Equal(t, count, countResult)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 1)
|
|
})
|
|
|
|
t.Run("first call not cached, clear cache, second call not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetMemberCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 1)
|
|
cachedStore.Channel().ClearCaches()
|
|
cachedStore.Channel().GetMemberCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 2)
|
|
})
|
|
|
|
t.Run("first call not cached, invalidate cache, second call not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetMemberCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 1)
|
|
cachedStore.Channel().InvalidateMemberCount("id")
|
|
cachedStore.Channel().GetMemberCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMemberCount", 2)
|
|
})
|
|
}
|
|
|
|
func TestChannelStoreChannelsMemberCountCache(t *testing.T) {
|
|
channelsCountResult := map[string]int64{
|
|
"channel1": 10,
|
|
"channel2": 20,
|
|
}
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
t.Run("first call not cached, second cached and returning same data", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
channelsCount, err := cachedStore.Channel().GetChannelsMemberCount([]string{"channel1", "channel2"})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, channelsCount, channelsCountResult)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetChannelsMemberCount", 1)
|
|
channelsCount, err = cachedStore.Channel().GetChannelsMemberCount([]string{"channel1", "channel2"})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, channelsCount, channelsCountResult)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetChannelsMemberCount", 1)
|
|
})
|
|
|
|
t.Run("first call not cached, invalidate cache, second call not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetChannelsMemberCount([]string{"channel1", "channel2"})
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetChannelsMemberCount", 1)
|
|
cachedStore.Channel().InvalidateMemberCount("channel1")
|
|
cachedStore.Channel().InvalidateMemberCount("channel2")
|
|
cachedStore.Channel().GetChannelsMemberCount([]string{"channel1", "channel2"})
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetChannelsMemberCount", 2)
|
|
})
|
|
}
|
|
|
|
func TestChannelStoreChannelPinnedPostsCountsCache(t *testing.T) {
|
|
countResult := int64(10)
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
t.Run("first call not cached, second cached and returning same data", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
count, err := cachedStore.Channel().GetPinnedPostCount("id", true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, count, countResult)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 1)
|
|
count, err = cachedStore.Channel().GetPinnedPostCount("id", true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, count, countResult)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 1)
|
|
})
|
|
|
|
t.Run("first call not cached, second force not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetPinnedPostCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 1)
|
|
cachedStore.Channel().GetPinnedPostCount("id", false)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 2)
|
|
})
|
|
|
|
t.Run("first call force not cached, second not cached, third cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetPinnedPostCount("id", false)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 1)
|
|
cachedStore.Channel().GetPinnedPostCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 2)
|
|
cachedStore.Channel().GetPinnedPostCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 2)
|
|
})
|
|
|
|
t.Run("first call not cached, clear cache, second call not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetPinnedPostCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 1)
|
|
cachedStore.Channel().ClearCaches()
|
|
cachedStore.Channel().GetPinnedPostCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 2)
|
|
})
|
|
|
|
t.Run("first call not cached, invalidate cache, second call not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetPinnedPostCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 1)
|
|
cachedStore.Channel().InvalidatePinnedPostCount("id")
|
|
cachedStore.Channel().GetPinnedPostCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetPinnedPostCount", 2)
|
|
})
|
|
}
|
|
|
|
func TestChannelStoreGuestCountCache(t *testing.T) {
|
|
countResult := int64(12)
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
t.Run("first call not cached, second cached and returning same data", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
count, err := cachedStore.Channel().GetGuestCount("id", true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, count, countResult)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 1)
|
|
count, err = cachedStore.Channel().GetGuestCount("id", true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, count, countResult)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 1)
|
|
})
|
|
|
|
t.Run("first call not cached, second force not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetGuestCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 1)
|
|
cachedStore.Channel().GetGuestCount("id", false)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 2)
|
|
})
|
|
|
|
t.Run("first call force not cached, second not cached, third cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetGuestCount("id", false)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 1)
|
|
cachedStore.Channel().GetGuestCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 2)
|
|
cachedStore.Channel().GetGuestCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 2)
|
|
})
|
|
|
|
t.Run("first call not cached, clear cache, second call not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetGuestCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 1)
|
|
cachedStore.Channel().ClearCaches()
|
|
cachedStore.Channel().GetGuestCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 2)
|
|
})
|
|
|
|
t.Run("first call not cached, invalidate cache, second call not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().GetGuestCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 1)
|
|
cachedStore.Channel().InvalidateGuestCount("id")
|
|
cachedStore.Channel().GetGuestCount("id", true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetGuestCount", 2)
|
|
})
|
|
}
|
|
|
|
func TestChannelStoreChannel(t *testing.T) {
|
|
channelId := "channel1"
|
|
fakeChannel := model.Channel{Id: channelId, Name: "channel1-name"}
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
t.Run("first call by id not cached, second cached and returning same data", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
channel, err := cachedStore.Channel().Get(channelId, true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, channel, &fakeChannel)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 1)
|
|
channel, err = cachedStore.Channel().Get(channelId, true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, channel, &fakeChannel)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 1)
|
|
})
|
|
|
|
t.Run("first call not cached, second force no cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().Get(channelId, true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 1)
|
|
cachedStore.Channel().Get(channelId, false)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 2)
|
|
})
|
|
|
|
t.Run("first call force no cached, second not cached, third cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
cachedStore.Channel().Get(channelId, false)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 1)
|
|
cachedStore.Channel().Get(channelId, true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 2)
|
|
cachedStore.Channel().Get(channelId, true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 2)
|
|
})
|
|
|
|
t.Run("first call not cached, clear cache, second call not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().Get(channelId, true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 1)
|
|
cachedStore.Channel().ClearCaches()
|
|
cachedStore.Channel().Get(channelId, true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 2)
|
|
})
|
|
|
|
t.Run("first call not cached, invalidate cache, second call not cached", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
cachedStore.Channel().Get(channelId, true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 1)
|
|
cachedStore.Channel().InvalidateChannel(channelId)
|
|
cachedStore.Channel().Get(channelId, true)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 2)
|
|
})
|
|
}
|
|
|
|
func TestChannelStoreSavePopulatesCache(t *testing.T) {
|
|
channelId := "channel1"
|
|
fakeChannel := model.Channel{Id: channelId, Name: "channel1-name"}
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
t.Run("save populates cache so subsequent get does not hit store", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
_, err = cachedStore.Channel().Save(request.TestContext(t), &fakeChannel, int64(0))
|
|
require.NoError(t, err)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Save", 1)
|
|
|
|
// Get should be served from cache without hitting the underlying store
|
|
channel, err := cachedStore.Channel().Get(channelId, true)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, &fakeChannel, channel)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 0)
|
|
})
|
|
|
|
t.Run("save populates cache, invalidate removes it, get hits store", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
_, err = cachedStore.Channel().Save(request.TestContext(t), &fakeChannel, int64(0))
|
|
require.NoError(t, err)
|
|
|
|
cachedStore.Channel().InvalidateChannel(channelId)
|
|
|
|
_, err = cachedStore.Channel().Get(channelId, true)
|
|
require.NoError(t, err)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 1)
|
|
})
|
|
}
|
|
|
|
func TestChannelStoreUpdateInvalidatesCache(t *testing.T) {
|
|
channelId := "channel1"
|
|
fakeChannel := model.Channel{Id: channelId, Name: "channel1-name"}
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
t.Run("update after save invalidates cache, get hits store", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
// Save populates cache
|
|
_, err = cachedStore.Channel().Save(request.TestContext(t), &fakeChannel, int64(0))
|
|
require.NoError(t, err)
|
|
|
|
// Update invalidates cache
|
|
_, err = cachedStore.Channel().Update(request.TestContext(t), &fakeChannel)
|
|
require.NoError(t, err)
|
|
|
|
// Get should miss cache and hit the store
|
|
_, err = cachedStore.Channel().Get(channelId, true)
|
|
require.NoError(t, err)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 1)
|
|
})
|
|
|
|
t.Run("update after get invalidates cache, next get hits store", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
// First Get populates cache
|
|
_, err = cachedStore.Channel().Get(channelId, true)
|
|
require.NoError(t, err)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 1)
|
|
|
|
// Update invalidates cache
|
|
_, err = cachedStore.Channel().Update(request.TestContext(t), &fakeChannel)
|
|
require.NoError(t, err)
|
|
|
|
// Second Get should miss cache and hit the store again
|
|
_, err = cachedStore.Channel().Get(channelId, true)
|
|
require.NoError(t, err)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "Get", 2)
|
|
})
|
|
}
|
|
|
|
func TestChannelStoreGetManyCache(t *testing.T) {
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
t.Run("first call not cached, second cached and returning same data", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
fakeChannel := model.Channel{Id: "channel1", Name: "channel1-name"}
|
|
fakeChannel2 := model.Channel{Id: "channel2", Name: "channel2-name"}
|
|
channels, err := cachedStore.Channel().GetMany([]string{fakeChannel.Id}, true)
|
|
require.NoError(t, err)
|
|
assert.ElementsMatch(t, model.ChannelList{&fakeChannel}, channels)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMany", 1)
|
|
|
|
channels, err = cachedStore.Channel().GetMany([]string{fakeChannel.Id}, true)
|
|
require.NoError(t, err)
|
|
assert.ElementsMatch(t, model.ChannelList{&fakeChannel}, channels)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMany", 1)
|
|
|
|
channels, err = cachedStore.Channel().GetMany([]string{fakeChannel.Id, fakeChannel2.Id}, true)
|
|
require.NoError(t, err)
|
|
assert.ElementsMatch(t, model.ChannelList{&fakeChannel, &fakeChannel2}, channels)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMany", 2)
|
|
})
|
|
|
|
t.Run("passing allowCache=false should bypass cache", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
fakeChannel := model.Channel{Id: "channel1", Name: "channel1-name"}
|
|
channels, err := cachedStore.Channel().GetMany([]string{fakeChannel.Id}, false)
|
|
require.NoError(t, err)
|
|
assert.ElementsMatch(t, model.ChannelList{&fakeChannel}, channels)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetMany", 1)
|
|
})
|
|
}
|
|
|
|
func TestChannelStoreGetByNamesCache(t *testing.T) {
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
t.Run("first call not cached, second cached and returning same data", func(t *testing.T) {
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
fakeChannel := model.Channel{Id: "channel1", Name: "channel1-name"}
|
|
fakeChannel2 := model.Channel{Id: "channel2", Name: "channel2-name"}
|
|
channels, err := cachedStore.Channel().GetByNames("team1", []string{fakeChannel.Name}, true)
|
|
require.NoError(t, err)
|
|
assert.ElementsMatch(t, []*model.Channel{&fakeChannel}, channels)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetByNames", 1)
|
|
|
|
channels, err = cachedStore.Channel().GetByNames("team1", []string{fakeChannel.Name}, true)
|
|
require.NoError(t, err)
|
|
assert.ElementsMatch(t, []*model.Channel{&fakeChannel}, channels)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetByNames", 1)
|
|
|
|
channels, err = cachedStore.Channel().GetByNames("team1", []string{fakeChannel.Name, fakeChannel2.Name}, true)
|
|
require.NoError(t, err)
|
|
assert.ElementsMatch(t, []*model.Channel{&fakeChannel, &fakeChannel2}, channels)
|
|
mockStore.Channel().(*mocks.ChannelStore).AssertNumberOfCalls(t, "GetByNames", 2)
|
|
})
|
|
}
|
|
|
|
func TestChannelStoreGetAllChannelMembersForUser(t *testing.T) {
|
|
logger := mlog.CreateConsoleTestLogger(t)
|
|
|
|
mockStore := getMockStore(t)
|
|
mockCacheProvider := getMockCacheProvider()
|
|
cachedStore, err := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider, logger)
|
|
require.NoError(t, err)
|
|
|
|
cmock := cmocks.NewCache(t)
|
|
cmock.On("Get", "u1", mock.AnythingOfType("*model.StringMap")).Return(nil)
|
|
|
|
cachedStore.channel.rootStore.channelMembersForUserCache = cmock
|
|
|
|
_, err = cachedStore.Channel().GetAllChannelMembersForUser(request.TestContext(t), "u1", true, false)
|
|
require.NoError(t, err)
|
|
}
|