Function TestCustomHeaderFlagRejectsInvalidHeaders

action/command/api_test.go:48–83

(t *testing.T)

}

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

	tests := []struct {
		name  string
		value string
	}{
		{
			name:  "missing separator",
			value: "Cookie",
		},
		{
			name:  "empty name",
			value: ": value",
		},
		{
			name:  "newline in value",
			value: "Cookie: value\nX-Other: injected",
		},
		{
			name:  "authorization is managed by bytebase action",
			value: "Authorization: Bearer token",
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			t.Parallel()

			headers := http.Header{}
			var headerErr error
			require.NoError(t, newCustomHeaderFlag(&headers, &headerErr).Set(tt.value))
			require.Error(t, headerErr)
		})
	}
}

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