Function TestLRURevisionCacheEvictionMemoryBased

db/revision_cache_test.go:280–377

(t *testing.T)

}

func TestLRURevisionCacheEvictionMemoryBased(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) {
			dbcOptions := DatabaseContextOptions{
				RevisionCacheOptions: &RevisionCacheOptions{
					MaxBytes:     725,
					MaxItemCount: 10,
				},
			}
			db, ctx := SetupTestDBWithOptions(t, dbcOptions)
			defer db.Close(ctx)
			collection, ctx := GetSingleDatabaseCollectionWithUser(ctx, t, db)
			cacheStats := db.DbStats.Cache()

			smallBody := Body{
				"channels": "_default", // add channel for default sync func in default collection test runs
			}

			var currMem, expValue, revZeroSize int64
			var rev1Version *Version
			for i := 0; i < 10; i++ {
				currMem = cacheStats.RevisionCacheTotalMemory.Value()
				revSize, _, docVersion := createDocAndReturnSizeAndRev(t, ctx, fmt.Sprint(i), collection, smallBody)
				if i == 0 {
					revZeroSize = int64(revSize)
				}
				if i == 1 {
					rev1Version = docVersion
				}
				expValue = currMem + int64(revSize)
				assert.Equal(t, expValue, cacheStats.RevisionCacheTotalMemory.Value())
			}

			// test eviction by number of items (adding new doc from createDocAndReturnSizeAndRev shouldn't take memory over threshold defined as 730 bytes)
			expValue -= revZeroSize // for doc being evicted
			docSize, rev, _ := createDocAndReturnSizeAndRev(t, ctx, fmt.Sprint(11), collection, smallBody)
			expValue += int64(docSize)
			// assert doc 0 been evicted
			docRev, ok := db.revisionCache.Peek(ctx, "0", rev, collection.GetCollectionID())
			assert.False(t, ok)
			assert.Nil(t, docRev.BodyBytes)

			currMem = cacheStats.RevisionCacheTotalMemory.Value()
			// assert total memory is as expected
			assert.Equal(t, expValue, currMem)