Function TestUpdateDeltaRevCacheMemoryStat

db/revision_cache_test.go:825–882

(t *testing.T)

}

func TestUpdateDeltaRevCacheMemoryStat(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, memoryBytesCounted := base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}, base.SgwIntStat{}
			backingStoreMap := CreateTestSingleBackingStoreMap(&testBackingStore{nil, &getDocumentCounter, &getRevisionCounter}, testCollectionID)
			cacheOptions := &RevisionCacheOptions{
				MaxItemCount: 10,
				MaxBytes:     140,
			}
			cache := NewLRURevisionCache(cacheOptions, backingStoreMap, &cacheHitCounter, &cacheMissCounter, &cacheNumItems, &memoryBytesCounted)

			firstDelta := []byte("delta")
			secondDelta := []byte("modified delta")
			thirdDelta := []byte("another delta further modified")
			ctx := base.TestCtx(t)

			// Trigger load into cache
			docRev, err := cache.GetWithRev(ctx, "doc1", "1-abc", testCollectionID, RevCacheIncludeDelta)
			assert.NoError(t, err, "Error adding to cache")

			revCacheMem := memoryBytesCounted.Value()
			revCacheDelta := newRevCacheDelta(firstDelta, "1-abc", docRev, false, nil)
			cache.UpdateDelta(ctx, "doc1", "1-abc", testCollectionID, revCacheDelta)
			// assert that rev cache memory increases by expected amount
			newMem := revCacheMem + revCacheDelta.totalDeltaBytes
			assert.Equal(t, newMem, memoryBytesCounted.Value())
			oldDeltaSize := revCacheDelta.totalDeltaBytes

			newMem = memoryBytesCounted.Value()
			revCacheDelta = newRevCacheDelta(secondDelta, "1-abc", docRev, false, nil)
			cache.UpdateDelta(ctx, "doc1", "1-abc", testCollectionID, revCacheDelta)

			// assert the overall memory stat is correctly updated (by the diff between the old delta and the new delta)
			newMem += revCacheDelta.totalDeltaBytes - oldDeltaSize
			assert.Equal(t, newMem, memoryBytesCounted.Value())

			revCacheDelta = newRevCacheDelta(thirdDelta, "1-abc", docRev, false, nil)
			cache.UpdateDelta(ctx, "doc1", "1-abc", testCollectionID, revCacheDelta)

			// assert that eviction took place and as result stat is now 0 (only item in cache was doc1)
			assert.Equal(t, int64(0), memoryBytesCounted.Value())
			assert.Equal(t, 0, cache.lruList.Len())
		})
	}
}