Function TestRevCacheCapacityStat

db/revision_cache_test.go:1483–1622

(t *testing.T)

}

func TestRevCacheCapacityStat(t *testing.T) {
	testCases := []struct {
		name       string
		UseCVCache bool
	}{
		{
			name:       "Rev cache pathway",
			UseCVCache: false,
		},
		{
			name:       "CV cache pathway",
			UseCVCache: true,
		},
	}
	for _, testCase := range testCases {
		t.Run(testCase.name, func(t *testing.T) {
			cacheHitCounter, cacheMissCounter, getDocumentCounter, getRevisionCounter, cacheNumItems, cacheMemoryStat := base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}
			backingStoreMap := CreateTestSingleBackingStoreMap(&testBackingStore{[]string{"badDoc"}, &getDocumentCounter, &getRevisionCounter}, testCollectionID)
			cacheOptions := &RevisionCacheOptions{
				MaxItemCount: 4,
				MaxBytes:     0,
			}
			cache := NewLRURevisionCache(cacheOptions, backingStoreMap, &cacheHitCounter, &cacheMissCounter, &cacheNumItems, &cacheMemoryStat)
			ctx := base.TestCtx(t)
			var err error

			assert.Equal(t, int64(0), cacheNumItems.Value())
			assert.Equal(t, int64(len(cache.cache)), cacheNumItems.Value())

			// Create a new doc + asert num items increments
			cache.Put(ctx, documentRevisionForCacheTest("doc1", `{"test":"1234"}`), testCollectionID)
			assert.Equal(t, int64(1), cacheNumItems.Value())
			assert.Equal(t, int64(len(cache.cache)), cacheNumItems.Value())

			// test not found doc, assert that the stat isn't incremented
			if testCase.UseCVCache {
				cv := Version{SourceID: "test", Value: 123}
				_, err = cache.GetWithCV(ctx, "badDoc", &cv, testCollectionID, false, false)
				require.Error(t, err)
			} else {
				_, err = cache.GetWithRev(ctx, "badDoc", "1-abc", testCollectionID, false)
				require.Error(t, err)
			}
			assert.Equal(t, int64(1), cacheNumItems.Value())
			assert.Equal(t, int64(len(cache.cache)), cacheNumItems.Value())

			// Get on a doc that doesn't exist in cache, assert num items increments
			var docRev DocumentRevision
			if testCase.UseCVCache {
				cv := Version{SourceID: "test", Value: 123}
				docRev, err = cache.GetWithCV(ctx, "doc2", &cv, testCollectionID, false, false)
				require.NoError(t, err)
			} else {
				docRev, err = cache.GetWithRev(ctx, "doc2", "1-abc", testCollectionID, false)
				require.NoError(t, err)
			}
			assert.Equal(t, "doc2", docRev.DocID)
			assert.Equal(t, int64(2), cacheNumItems.Value())