Function TestCacheTwo

flytestdlib/utils/auto_refresh_cache_test.go:37–92

(t *testing.T)

}

func TestCacheTwo(t *testing.T) {
	testResyncPeriod := time.Millisecond
	rateLimiter := NewRateLimiter("mockLimiter", 100, 1)

	t.Run("normal operation", func(t *testing.T) {
		// the size of the cache is at least as large as the number of items we're storing
		cache, err := NewAutoRefreshCache(syncFakeItem, rateLimiter, testResyncPeriod, 10, nil)
		assert.NoError(t, err)

		ctx, cancel := context.WithCancel(context.Background())
		cache.Start(ctx)

		// Create ten items in the cache
		for i := 1; i <= 10; i++ {
			_, err := cache.GetOrCreate(fakeCacheItem{
				id:  fmt.Sprintf("%d", i),
				val: 0,
			})
			assert.NoError(t, err)
		}

		// Wait half a second for all resync periods to complete
		time.Sleep(500 * time.Millisecond)
		for i := 1; i <= 10; i++ {
			item := cache.Get(fmt.Sprintf("%d", i))
			assert.Equal(t, 10, item.(fakeCacheItem).val)
		}
		cancel()
	})

	t.Run("deleting objects from cache", func(t *testing.T) {
		// the size of the cache is at least as large as the number of items we're storing
		cache, err := NewAutoRefreshCache(syncFakeItemAlwaysDelete, rateLimiter, testResyncPeriod, 10, nil)
		assert.NoError(t, err)

		ctx, cancel := context.WithCancel(context.Background())
		cache.Start(ctx)

		// Create ten items in the cache
		for i := 1; i <= 10; i++ {
			_, err = cache.GetOrCreate(fakeCacheItem{
				id:  fmt.Sprintf("%d", i),
				val: 0,
			})
			assert.NoError(t, err)
		}

		// Wait for all resync periods to complete
		time.Sleep(50 * time.Millisecond)
		for i := 1; i <= 10; i++ {
			obj := cache.Get(fmt.Sprintf("%d", i))
			assert.Nil(t, obj)
		}
		cancel()
	})
}