github.com/mattermosttest/mattermost-server/v5@v5.0.0-20200917143240-9dfa12e121f9/store/localcachelayer/scheme_layer_test.go (about)

     1  // Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
     2  // See LICENSE.txt for license information.
     3  
     4  package localcachelayer
     5  
     6  import (
     7  	"testing"
     8  
     9  	"github.com/mattermost/mattermost-server/v5/model"
    10  	"github.com/mattermost/mattermost-server/v5/store/storetest"
    11  	"github.com/mattermost/mattermost-server/v5/store/storetest/mocks"
    12  	"github.com/stretchr/testify/assert"
    13  	"github.com/stretchr/testify/require"
    14  )
    15  
    16  func TestSchemeStore(t *testing.T) {
    17  	StoreTest(t, storetest.TestSchemeStore)
    18  }
    19  
    20  func TestSchemeStoreCache(t *testing.T) {
    21  	fakeScheme := model.Scheme{Id: "123", Name: "scheme-name"}
    22  
    23  	t.Run("first call not cached, second cached and returning same data", func(t *testing.T) {
    24  		mockStore := getMockStore()
    25  		mockCacheProvider := getMockCacheProvider()
    26  		cachedStore := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider)
    27  
    28  		scheme, err := cachedStore.Scheme().Get("123")
    29  		require.Nil(t, err)
    30  		assert.Equal(t, scheme, &fakeScheme)
    31  		mockStore.Scheme().(*mocks.SchemeStore).AssertNumberOfCalls(t, "Get", 1)
    32  		require.Nil(t, err)
    33  		assert.Equal(t, scheme, &fakeScheme)
    34  		cachedStore.Scheme().Get("123")
    35  		mockStore.Scheme().(*mocks.SchemeStore).AssertNumberOfCalls(t, "Get", 1)
    36  	})
    37  
    38  	t.Run("first call not cached, save, and then not cached again", func(t *testing.T) {
    39  		mockStore := getMockStore()
    40  		mockCacheProvider := getMockCacheProvider()
    41  		cachedStore := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider)
    42  
    43  		cachedStore.Scheme().Get("123")
    44  		mockStore.Scheme().(*mocks.SchemeStore).AssertNumberOfCalls(t, "Get", 1)
    45  		cachedStore.Scheme().Save(&fakeScheme)
    46  		cachedStore.Scheme().Get("123")
    47  		mockStore.Scheme().(*mocks.SchemeStore).AssertNumberOfCalls(t, "Get", 2)
    48  	})
    49  
    50  	t.Run("first call not cached, delete, and then not cached again", func(t *testing.T) {
    51  		mockStore := getMockStore()
    52  		mockCacheProvider := getMockCacheProvider()
    53  		cachedStore := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider)
    54  
    55  		cachedStore.Scheme().Get("123")
    56  		mockStore.Scheme().(*mocks.SchemeStore).AssertNumberOfCalls(t, "Get", 1)
    57  		cachedStore.Scheme().Delete("123")
    58  		cachedStore.Scheme().Get("123")
    59  		mockStore.Scheme().(*mocks.SchemeStore).AssertNumberOfCalls(t, "Get", 2)
    60  	})
    61  
    62  	t.Run("first call not cached, permanent delete all, and then not cached again", func(t *testing.T) {
    63  		mockStore := getMockStore()
    64  		mockCacheProvider := getMockCacheProvider()
    65  		cachedStore := NewLocalCacheLayer(mockStore, nil, nil, mockCacheProvider)
    66  
    67  		cachedStore.Scheme().Get("123")
    68  		mockStore.Scheme().(*mocks.SchemeStore).AssertNumberOfCalls(t, "Get", 1)
    69  		cachedStore.Scheme().PermanentDeleteAll()
    70  		cachedStore.Scheme().Get("123")
    71  		mockStore.Scheme().(*mocks.SchemeStore).AssertNumberOfCalls(t, "Get", 2)
    72  	})
    73  }