cache_test.go

package gcache_test

import (
	"testing"
	"time"

	"sync"
	"sync/atomic"

	"github.com/bluele/gcache"
)

func TestLoaderFunc(t *testing.T) {
	size := 2
	var testCaches = []*gcache.CacheBuilder{
		gcache.New(size).Simple(),
		gcache.New(size).LRU(),
		gcache.New(size).LFU(),
		gcache.New(size).ARC(),
	}
	for _, builder := range testCaches {
		var testCounter int64
		counter := 1000
		cache := builder.
			LoaderFunc(func(key interface{}) (interface{}, error) {
				time.Sleep(10 * time.Millisecond)
				return atomic.AddInt64(&testCounter, 1), nil
			}).
			EvictedFunc(func(key, value interface{}) {
				panic(key)
			}).Build()

		var wg sync.WaitGroup
		for i := 0; i < counter; i++ {
			wg.Add(1)
			go func() {
				defer wg.Done()
				_, err := cache.Get(0)
				if err != nil {
					t.Error(err)
				}
			}()
		}
		wg.Wait()

		if testCounter != 1 {
			t.Errorf("testCounter != %v", testCounter)
		}
	}
}