Function TestVirtualHostHandler

vhost_test.go:13–117

(t *testing.T)

)

func TestVirtualHostHandler(t *testing.T) {
	okHandler := func(c *Context) error { return c.String(http.StatusOK, http.StatusText(http.StatusOK)) }
	teapotHandler := func(c *Context) error { return c.String(http.StatusTeapot, http.StatusText(http.StatusTeapot)) }
	acceptHandler := func(c *Context) error { return c.String(http.StatusAccepted, http.StatusText(http.StatusAccepted)) }
	teapotMiddleware := MiddlewareFunc(func(next HandlerFunc) HandlerFunc { return teapotHandler })

	ok := New()
	ok.GET("/", okHandler)
	ok.GET("/foo", okHandler)

	teapot := New()
	teapot.GET("/", teapotHandler)
	teapot.GET("/foo", teapotHandler)

	middle := New()
	middle.Use(teapotMiddleware)
	middle.GET("/", okHandler)
	middle.GET("/foo", okHandler)

	virtualHosts := NewVirtualHostHandler(map[string]*Echo{
		"ok.com":         ok,
		"teapot.com":     teapot,
		"middleware.com": middle,
	})
	virtualHosts.GET("/", acceptHandler)
	virtualHosts.GET("/foo", acceptHandler)

	var testCases = []struct {
		name         string
		whenHost     string
		whenPath     string
		expectBody   string
		expectStatus int
	}{
		{
			name:         "No Host Root",
			whenHost:     "",
			whenPath:     "/",
			expectBody:   http.StatusText(http.StatusAccepted),
			expectStatus: http.StatusAccepted,
		},
		{
			name:         "No Host Foo",
			whenHost:     "",
			whenPath:     "/foo",
			expectBody:   http.StatusText(http.StatusAccepted),
			expectStatus: http.StatusAccepted,
		},
		{
			name:         "OK Host Root",
			whenHost:     "ok.com",
			whenPath:     "/",
			expectBody:   http.StatusText(http.StatusOK),
			expectStatus: http.StatusOK,
		},
		{
			name:         "OK Host Foo",
			whenHost:     "ok.com",