Function TestLoadingCache

cache_test.go:11–64

(t *testing.T)

)

func TestLoadingCache(t *testing.T) {
	c := NewLoadingCache(
		func(key interface{}) (interface{}, error) {
			switch v := key.(type) {
			case int:
				return strconv.Itoa(v), nil
			default:
				return nil, errors.New("invalid type")
			}
		},
		2) // cap = 2
	assertEqual(t, 0, len(c.m))
	v, err := c.get(1)
	assertNoErr(t, err)
	assertEqual(t, "1", v)
	assertEqual(t, 1, len(c.m))

	v, err = c.get(1)
	assertNoErr(t, err)
	assertEqual(t, "1", v)
	assertEqual(t, 1, len(c.m))

	v, err = c.get(2)
	assertNoErr(t, err)
	assertEqual(t, "2", v)
	assertEqual(t, 2, len(c.m))

	// over capacity, m is reset
	v, err = c.get(3)
	assertNoErr(t, err)
	assertEqual(t, "3", v)
	assertEqual(t, 1, len(c.m))

	// Invalid capacity
	assertPanic(t, func() {
		NewLoadingCache(func(key interface{}) (interface{}, error) { return key, nil }, -1)
	})

	// Loading failure
	c = NewLoadingCache(
		func(key interface{}) (interface{}, error) {
			if key.(int)%2 == 0 {
				return key, nil
			} else {
				return nil, fmt.Errorf("artificial error: %d", key.(int))
			}
		}, 0)
	v, err = c.get(12)
	assertNoErr(t, err)
	assertEqual(t, 12, v)
	_, err = c.get(21)
	assertErr(t, err)
	assertEqual(t, "artificial error: 21", err.Error())
}

const (
	benchLoadingCacheRandSeed    = 12345