Function TestOnAnyErrorCode

core/router/status_test.go:20–83

(t *testing.T)

}

func TestOnAnyErrorCode(t *testing.T) {
	app := iris.New()
	app.Configure(iris.WithFireMethodNotAllowed)

	buff := &bytes.Buffer{}
	expectedPrintBeforeExecuteErr := "printed before error"

	// with a middleware
	app.OnAnyErrorCode(func(ctx *context.Context) {
		buff.WriteString(expectedPrintBeforeExecuteErr)
		ctx.Next()
	}, defaultErrHandler)

	expectedFoundResponse := "found"
	app.Get("/found", func(ctx *context.Context) {
		ctx.WriteString(expectedFoundResponse)
	})

	expected407 := "this should be sent, we manage the response response by ourselves"
	app.Get("/407", func(ctx *context.Context) {
		ctx.Record()
		ctx.WriteString(expected407)
		ctx.StatusCode(iris.StatusProxyAuthRequired)
	})

	e := httptest.New(t, app)

	e.GET("/found").Expect().Status(iris.StatusOK).
		Body().IsEqual(expectedFoundResponse)

	e.GET("/notfound").Expect().Status(iris.StatusNotFound).
		Body().IsEqual(http.StatusText(iris.StatusNotFound))

	checkAndClearBuf(t, buff, expectedPrintBeforeExecuteErr)

	e.POST("/found").Expect().Status(iris.StatusMethodNotAllowed).
		Body().IsEqual(http.StatusText(iris.StatusMethodNotAllowed))

	checkAndClearBuf(t, buff, expectedPrintBeforeExecuteErr)

	e.GET("/407").Expect().Status(iris.StatusProxyAuthRequired).
		Body().IsEqual(expected407)

	// Test Configuration.ResetOnFireErrorCode.
	app2 := iris.New()
	app2.Configure(iris.WithResetOnFireErrorCode)

	app2.OnAnyErrorCode(func(ctx *context.Context) {
		buff.WriteString(expectedPrintBeforeExecuteErr)
		ctx.Next()
	}, defaultErrHandler)

	app2.Get("/406", func(ctx *context.Context) {
		ctx.Record()
		ctx.WriteString("this should not be sent, only status text will be sent")
		ctx.WriteString("the handler can handle 'rollback' of the text when error code fired because of the recorder")
		ctx.StatusCode(iris.StatusNotAcceptable)
	})