Function TestQueryLogHandler

net/query_test.go:57–103

(t *testing.T)

}

func TestQueryLogHandler(t *testing.T) {
	for _, ti := range []struct {
		name     string
		routes   string
		query    string
		expected int
	}{
		{
			name:     "request without query",
			routes:   `r: * -> inlineContent("OK") -> <shunt>;`,
			query:    "",
			expected: http.StatusOK,
		},
		{
			name:     "request with query",
			routes:   `r1: Query("foo") -> status(400) -> inlineContent("FAIL") -> <shunt> ;r2: * -> inlineContent("OK") -> <shunt>;`,
			query:    "foo=bar",
			expected: http.StatusOK,
		},
		{
			name:     "request with bad query does not block",
			routes:   `r1: Query("foo") -> status(400) -> inlineContent("FAIL") -> <shunt> ;r2: * -> inlineContent("OK") -> <shunt>;`,
			query:    "foo=bar;",
			expected: http.StatusOK,
		},
	} {
		t.Run(ti.name, func(t *testing.T) {

			req, err := http.NewRequest("GET", "/path", nil)
			if err != nil {
				t.Fatal(err)
			}
			req.URL.RawQuery = ti.query

			noop := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {})
			recorder := httptest.NewRecorder()
			h := &ValidateQueryLogHandler{
				Handler: noop,
			}
			h.ServeHTTP(recorder, req)

			if recorder.Code != ti.expected {
				t.Fatalf("Failed to get expected code %d, got %d", ti.expected, recorder.Code)
			}
		})
	}
}