Function TestCacheValidator

cache/cache_test.go:156–215

This works but we have issue on golog.SetLevel and get golog.Level on httptest.New when tests are running in parallel and the loggers are used. TODO: Fix it on golog repository or here, we'll see. func TestCacheHandlerParallel(t *testing.T) { t.Parallel() TestCache(t) }

(t *testing.T)

// }

func TestCacheValidator(t *testing.T) {
	app := iris.New()
	var n uint32

	h := func(ctx *context.Context) {
		atomic.AddUint32(&n, 1)
		ctx.Write([]byte(expectedBodyStr))
	}

	validCache := cache.Handler(cacheDuration)
	app.Get("/", validCache, h)

	managedCache := cache.Cache(cache.MaxAge(cacheDuration))
	managedCache.AddRule(rule.Validator([]rule.PreValidator{
		func(ctx *context.Context) bool {
			// should always invalid for cache, don't bother to go to try to get or set cache
			return ctx.Request().URL.Path != "/invalid"
		},
	}, nil))

	managedCache2 := cache.Cache(cache.MaxAge(cacheDuration))
	managedCache2.AddRule(rule.Validator(nil,
		[]rule.PostValidator{
			func(ctx *context.Context) bool {
				// it's passed the Claim and now Valid checks if the response contains a header of "DONT"
				return ctx.ResponseWriter().Header().Get("DONT") == ""
			},
		},
	))

	app.Get("/valid", validCache, h)

	app.Get("/invalid", managedCache.ServeHTTP, h)
	app.Get("/invalid2", managedCache2.ServeHTTP, func(ctx *context.Context) {
		atomic.AddUint32(&n, 1)
		ctx.Header("DONT", "DO not cache that response even if it was claimed")
		ctx.Write([]byte(expectedBodyStr))
	})

	e := httptest.New(t, app)

	// execute from cache the next time
	e.GET("/valid").Expect().Status(http.StatusOK).Body().IsEqual(expectedBodyStr)
	time.Sleep(cacheDuration / 5) // lets wait for a while, cache should be saved and ready
	e.GET("/valid").Expect().Status(http.StatusOK).Body().IsEqual(expectedBodyStr)
	counter := atomic.LoadUint32(&n)
	if counter > 1 {
		// n should be 1 because it doesn't changed after the first call
		t.Fatalf("%s: %v", t.Name(), &testError{1, counter})
	}
	// don't execute from cache, execute the original, counter should ++ here
	e.GET("/invalid").Expect().Status(http.StatusOK).Body().IsEqual(expectedBodyStr)  // counter = 2
	e.GET("/invalid2").Expect().Status(http.StatusOK).Body().IsEqual(expectedBodyStr) // counter = 3

	counter = atomic.LoadUint32(&n)
	if counter != 3 {
		// n should be 1 because it doesn't changed after the first call
		t.Fatalf("%s: %v", t.Name(), &testError{3, counter})