Function TestUseModuleCombine

srvutil/servlet_test.go:133–185

(t *testing.T)

}

func TestUseModuleCombine(t *testing.T) {
	s1 := HandlerServlet("/ok", h)
	s1 = UseServlet(s1, func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
			res.Header().Set("foo", "bar")
			next.ServeHTTP(res, req)
		})
	})

	s2 := HandlerServlet("/ok2", h)
	s2 = UseServlet(s2, func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
			res.Header().Set("foox", "barx")
			next.ServeHTTP(res, req)
		})
	})

	s2 = UseServlet(s2, func(next http.Handler) http.Handler {
		return http.HandlerFunc(func(res http.ResponseWriter, req *http.Request) {
			res.Header().Set("fooz", "barz")
			next.ServeHTTP(res, req)
		})
	})
	s2 = PrefixServlet(s2, "/inner")

	s := CombineServlets(s1, s2)
	r := mux.NewRouter()
	s.RegisterRouting(r)

	{
		res := httptest.NewRecorder()
		req, err := http.NewRequest("GET", "/ok", nil)
		assert.NoError(t, err)
		r.ServeHTTP(res, req)

		assert.Equal(t, "great success", res.Body.String())
		assert.Equal(t, "bar", res.Header().Get("foo"), "s1 should have its middleware")
		assert.Equal(t, "", res.Header().Get("foox"), "s1 should not inherit s2's middlewares")
		assert.Equal(t, "", res.Header().Get("fooz"), "s1 should not inherit s2's middlewares")
	}

	{
		res := httptest.NewRecorder()
		req, err := http.NewRequest("GET", "/inner/ok2", nil)
		assert.NoError(t, err)
		r.ServeHTTP(res, req)

		assert.Equal(t, "great success", res.Body.String())
		assert.Equal(t, "", res.Header().Get("foo"), "s2 should not inherit s1's middleware")
		assert.Equal(t, "barx", res.Header().Get("foox"), "s2 should have its middleware")
		assert.Equal(t, "barz", res.Header().Get("fooz"), "s2 should have its middleware")
	}
}

func TestCombineModules(t *testing.T) {
	r := mux.NewRouter()