Function TestRequestHeaderFirstLineSettersSanitizeNewlines

header_test.go:2804–2904

(t *testing.T)

}

func TestRequestHeaderFirstLineSettersSanitizeNewlines(t *testing.T) {
	t.Parallel()

	testCases := []struct {
		name          string
		set           func(*RequestHeader)
		value         func(*RequestHeader) []byte
		wantValue     string
		wantFirstLine string
		wantNoHTTP11  bool
	}{
		{
			name: "SetMethod",
			set: func(h *RequestHeader) {
				h.SetMethod("GET\r\nInjected-Method: true")
			},
			value:         func(h *RequestHeader) []byte { return h.Method() },
			wantValue:     "GET  Injected-Method: true",
			wantFirstLine: "GET  Injected-Method: true / HTTP/1.1",
		},
		{
			name: "SetMethodBytes",
			set: func(h *RequestHeader) {
				h.SetMethodBytes([]byte("GET\r\nInjected-Method-Bytes: true"))
			},
			value:         func(h *RequestHeader) []byte { return h.Method() },
			wantValue:     "GET  Injected-Method-Bytes: true",
			wantFirstLine: "GET  Injected-Method-Bytes: true / HTTP/1.1",
		},
		{
			name: "SetRequestURI",
			set: func(h *RequestHeader) {
				h.SetRequestURI("/\r\nInjected-URI: true")
			},
			value:         func(h *RequestHeader) []byte { return h.RequestURI() },
			wantValue:     "/  Injected-URI: true",
			wantFirstLine: "GET /  Injected-URI: true HTTP/1.1",
		},
		{
			name: "SetRequestURIBytes",
			set: func(h *RequestHeader) {
				h.SetRequestURIBytes([]byte("/\r\nInjected-URI-Bytes: true"))
			},
			value:         func(h *RequestHeader) []byte { return h.RequestURI() },
			wantValue:     "/  Injected-URI-Bytes: true",
			wantFirstLine: "GET /  Injected-URI-Bytes: true HTTP/1.1",
		},
		{
			name: "SetProtocol",
			set: func(h *RequestHeader) {
				h.SetProtocol("HTTP/1.1\r\nInjected-Protocol: true")
			},
			value:         func(h *RequestHeader) []byte { return h.Protocol() },
			wantValue:     "HTTP/1.1  Injected-Protocol: true",
			wantFirstLine: "GET / HTTP/1.1  Injected-Protocol: true",
			wantNoHTTP11:  true,
		},
		{