Function TestPartyOnErrorCode

core/router/status_test.go:95–160

(t *testing.T)

}

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

	globalNotFoundResponse := "custom not found"
	app.OnErrorCode(iris.StatusNotFound, func(ctx iris.Context) {
		ctx.WriteString(globalNotFoundResponse)
	})

	globalMethodNotAllowedResponse := "global: method not allowed"
	app.OnErrorCode(iris.StatusMethodNotAllowed, func(ctx iris.Context) {
		ctx.WriteString(globalMethodNotAllowedResponse)
	})

	app.Get("/path", h)

	usersResponse := "users: method allowed"
	users := app.Party("/users")
	users.OnErrorCode(iris.StatusMethodNotAllowed, func(ctx iris.Context) {
		ctx.WriteString(usersResponse)
	})
	users.Get("/", h)
	write400 := func(ctx iris.Context) { ctx.StatusCode(iris.StatusBadRequest) }
	// test setting the error code from a handler.
	users.Get("/badrequest", write400)

	usersuserResponse := "users:user: method allowed"
	user := users.Party("/{id:int}")
	user.OnErrorCode(iris.StatusMethodNotAllowed, func(ctx iris.Context) {
		ctx.WriteString(usersuserResponse)
	})
	usersuserNotFoundResponse := "users:user: not found"
	user.OnErrorCode(iris.StatusNotFound, func(ctx iris.Context) {
		ctx.WriteString(usersuserNotFoundResponse)
	})
	user.Get("/", h)
	user.Get("/ab/badrequest", write400)

	friends := users.Party("/friends")
	friends.Get("/{id:int}", h)

	e := httptest.New(t, app)

	e.GET("/notfound").Expect().Status(iris.StatusNotFound).Body().IsEqual(globalNotFoundResponse)
	e.POST("/path").Expect().Status(iris.StatusMethodNotAllowed).Body().IsEqual(globalMethodNotAllowedResponse)
	e.GET("/path").Expect().Status(iris.StatusOK).Body().IsEqual("/path")

	e.POST("/users").Expect().Status(iris.StatusMethodNotAllowed).
		Body().IsEqual(usersResponse)

	e.POST("/users/42").Expect().Status(iris.StatusMethodNotAllowed).
		Body().IsEqual(usersuserResponse)

	e.GET("/users/42").Expect().Status(iris.StatusOK).
		Body().IsEqual("/users/42")
	e.GET("/users/ab").Expect().Status(iris.StatusNotFound).Body().IsEqual(usersuserNotFoundResponse)
	// inherit the parent.
	e.GET("/users/42/friends/dsa").Expect().Status(iris.StatusNotFound).Body().IsEqual(usersuserNotFoundResponse)