github.com/cloudwego/hertz@v0.9.3/pkg/protocol/header_test.go

/*
 * Copyright 2022 CloudWeGo Authors
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * The MIT License (MIT)
 *
 * Copyright (c) 2015-present Aliaksandr Valialkin, VertaMedia, Kirill Danshin, Erik Dubbelboer, FastHTTP Authors
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 * This file may have been modified by CloudWeGo authors. All CloudWeGo
 * Modifications are Copyright 2022 CloudWeGo Authors.
 */

package protocol

import (
	"bytes"
	"fmt"
	"net/http"
	"strings"
	"testing"

	"github.com/cloudwego/hertz/internal/bytestr"
	"github.com/cloudwego/hertz/pkg/common/hlog"
	"github.com/cloudwego/hertz/pkg/common/test/assert"
	"github.com/cloudwego/hertz/pkg/protocol/consts"
)

func TestRequestHeaderSetRawHeaders(t *testing.T) {
	h := RequestHeader{}
	h.SetRawHeaders([]byte("foo"))
	assert.DeepEqual(t, h.rawHeaders, []byte("foo"))
}

func TestResponseHeaderSetHeaderLength(t *testing.T) {
	h := ResponseHeader{}
	h.SetHeaderLength(15)
	assert.DeepEqual(t, h.headerLength, 15)
	assert.DeepEqual(t, h.GetHeaderLength(), 15)
}

func TestSetNoHTTP11(t *testing.T) {
	rh := ResponseHeader{}
	rh.SetProtocol(consts.HTTP10)
	assert.DeepEqual(t, consts.HTTP10, rh.protocol)

	rh.SetProtocol(consts.HTTP11)
	assert.DeepEqual(t, consts.HTTP11, rh.protocol)
	assert.True(t, rh.IsHTTP11())

	h := RequestHeader{}
	h.SetProtocol(consts.HTTP10)
	assert.DeepEqual(t, consts.HTTP10, h.protocol)

	h.SetProtocol(consts.HTTP11)
	assert.DeepEqual(t, consts.HTTP11, h.protocol)
	assert.True(t, h.IsHTTP11())
}

func TestResponseHeaderSetContentType(t *testing.T) {
	h := ResponseHeader{}
	h.SetContentType("foo")
	assert.DeepEqual(t, h.contentType, []byte("foo"))
}

func TestSetContentLengthBytes(t *testing.T) {
	h := RequestHeader{}
	h.SetContentLengthBytes([]byte("foo"))
	assert.DeepEqual(t, h.contentLengthBytes, []byte("foo"))

	rh := ResponseHeader{}
	rh.SetContentLengthBytes([]byte("foo"))
	assert.DeepEqual(t, rh.contentLengthBytes, []byte("foo"))
}

func TestInitContentLengthWithValue(t *testing.T) {
	initLength := 100
	h := RequestHeader{}
	h.InitContentLengthWithValue(initLength)
	assert.DeepEqual(t, h.contentLength, initLength)

	rh := ResponseHeader{}
	rh.InitContentLengthWithValue(initLength)
	assert.DeepEqual(t, rh.contentLength, initLength)
}

func TestSetContentEncoding(t *testing.T) {
	rh := ResponseHeader{}
	rh.SetContentEncoding("gzip")
	assert.DeepEqual(t, rh.contentEncoding, []byte("gzip"))
}

func Test_peekRawHeader(t *testing.T) {
	s := "Expect: 100-continue\r\nUser-Agent: foo\r\nHost: 127.0.0.1\r\nConnection: Keep-Alive\r\nContent-Length: 5\r\nContent-Type: foo/bar\r\n\r\nabcdef4343"
	assert.DeepEqual(t, []byte("127.0.0.1"), peekRawHeader([]byte(s), []byte("Host")))
}

func TestResponseHeader_SetContentLength(t *testing.T) {
	rh := new(ResponseHeader)
	rh.SetContentLength(-1)
	assert.True(t, strings.Contains(string(rh.Header()), "Transfer-Encoding: chunked"))
	rh.SetContentLength(-2)
	assert.True(t, strings.Contains(string(rh.Header()), "Transfer-Encoding: identity"))
}

func TestResponseHeader_SetContentRange(t *testing.T) {
	rh := new(ResponseHeader)
	rh.SetContentRange(1, 5, 10)
	assert.DeepEqual(t, rh.bufKV.value, []byte("bytes 1-5/10"))
}

func TestSetCanonical(t *testing.T) {
	h := ResponseHeader{}
	h.SetCanonical([]byte(consts.HeaderContentType), []byte("foo"))
	h.SetCanonical([]byte(consts.HeaderServer), []byte("foo1"))
	h.SetCanonical([]byte(consts.HeaderSetCookie), []byte("foo2"))
	h.SetCanonical([]byte(consts.HeaderContentLength), []byte("3"))
	h.SetCanonical([]byte(consts.HeaderConnection), []byte("foo4"))
	h.SetCanonical([]byte(consts.HeaderTransferEncoding), []byte("foo5"))
	h.SetCanonical([]byte(consts.HeaderTrailer), []byte("foo7"))
	h.SetCanonical([]byte("bar"), []byte("foo6"))

	assert.DeepEqual(t, []byte("foo"), h.ContentType())
	assert.DeepEqual(t, []byte("foo1"), h.Server())
	assert.DeepEqual(t, true, strings.Contains(string(h.Header()), "foo2"))
	assert.DeepEqual(t, 3, h.ContentLength())
	assert.DeepEqual(t, false, h.ConnectionClose())
	assert.DeepEqual(t, false, strings.Contains(string(h.ContentType()), "foo5"))
	assert.DeepEqual(t, true, strings.Contains(string(h.Header()), "Trailer: Foo7"))
	assert.DeepEqual(t, true, strings.Contains(string(h.Header()), "bar: foo6"))
}

func TestHasAcceptEncodingBytes(t *testing.T) {
	h := RequestHeader{}
	h.Set(consts.HeaderAcceptEncoding, "gzip")
	assert.True(t, h.HasAcceptEncodingBytes([]byte("gzip")))
}

func TestRequestHeaderGet(t *testing.T) {
	h := RequestHeader{}
	rightVal := "yyy"
	h.Set("xxx", rightVal)
	val := h.Get("xxx")
	if val != rightVal {
		t.Fatalf("Unexpected %v. Expected %v", val, rightVal)
	}
}

func TestResponseHeaderGet(t *testing.T) {
	h := ResponseHeader{}
	rightVal := "yyy"
	h.Set("xxx", rightVal)
	val := h.Get("xxx")
	assert.DeepEqual(t, val, rightVal)
}

func TestRequestHeaderGetAll(t *testing.T) {
	h := RequestHeader{}
	h.Set("Foo-Bar", "foo")
	h.Add("Foo-Bar", "bar")
	h.Add("Foo-Bar", "foo-bar")
	values := h.GetAll("Foo-Bar")
	assert.DeepEqual(t, values, []string{"foo", "bar", "foo-bar"})
}

func TestResponseHeaderGetAll(t *testing.T) {
	h := ResponseHeader{}
	h.Set("Foo-Bar", "foo")
	h.Add("Foo-Bar", "bar")
	h.Add("Foo-Bar", "foo-bar")
	values := h.GetAll("Foo-Bar")
	assert.DeepEqual(t, values, []string{"foo", "bar", "foo-bar"})
}

func TestRequestHeaderVisitAll(t *testing.T) {
	h := RequestHeader{}
	h.Set("xxx", "yyy")
	h.Set("xxx2", "yyy2")
	h.SetHost("host")
	h.SetContentLengthBytes([]byte("content-length"))
	h.Set(consts.HeaderContentType, "content-type")
	h.Set(consts.HeaderUserAgent, "user-agent")
	err := h.Trailer().SetTrailers([]byte("foo, bar"))
	if err != nil {
		t.Fatalf("Set trailer err %v", err)
	}
	h.SetCookie("foo", "bar")
	h.Set(consts.HeaderConnection, "close")
	h.VisitAll(func(k, v []byte) {
		key := string(k)
		value := string(v)
		switch key {
		case consts.HeaderHost:
			assert.DeepEqual(t, value, "host")
		case consts.HeaderContentLength:
			assert.DeepEqual(t, value, "content-length")
		case consts.HeaderContentType:
			assert.DeepEqual(t, value, "content-type")
		case consts.HeaderUserAgent:
			assert.DeepEqual(t, value, "user-agent")
		case consts.HeaderTrailer:
			assert.DeepEqual(t, value, "Foo, Bar")
		case consts.HeaderCookie:
			assert.DeepEqual(t, value, "foo=bar")
		case consts.HeaderConnection:
			assert.DeepEqual(t, value, "close")
		case "Xxx":
			assert.DeepEqual(t, value, "yyy")
		case "Xxx2":
			assert.DeepEqual(t, value, "yyy2")
		default:
			t.Fatalf("Unexpected key %v", key)
		}
	})
}

func TestRequestHeaderCookie(t *testing.T) {
	var h RequestHeader
	h.SetCookie("foo", "bar")
	cookie := h.Cookie("foo")
	assert.DeepEqual(t, []byte("bar"), cookie)
}

func TestRequestHeaderCookies(t *testing.T) {
	var h RequestHeader
	h.SetCookie("foo", "bar")
	h.SetCookie("привет", "мир")
	cookies := h.Cookies()
	assert.DeepEqual(t, 2, len(cookies))
	assert.DeepEqual(t, []byte("foo"), cookies[0].Key())
	assert.DeepEqual(t, []byte("bar"), cookies[0].Value())
	assert.DeepEqual(t, []byte("привет"), cookies[1].Key())
	assert.DeepEqual(t, []byte("мир"), cookies[1].Value())
}

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

	var h RequestHeader
	h.Set("Foo-Bar", "baz")
	h.Set("aaa", "bbb")
	h.Set("ccc", "ddd")
	h.Set(consts.HeaderConnection, "keep-alive")
	h.Set(consts.HeaderContentType, "aaa")
	h.Set(consts.HeaderServer, "aaabbb")
	h.Set(consts.HeaderContentLength, "1123")
	h.Set(consts.HeaderTrailer, "foo, bar")
	h.Set(consts.HeaderUserAgent, "foo-bar")
	h.SetHost("foobar")
	h.SetCookie("foo", "bar")

	h.del([]byte("Foo-Bar"))
	h.del([]byte("Connection"))
	h.DelBytes([]byte("Content-Type"))
	h.del([]byte(consts.HeaderServer))
	h.del([]byte("Content-Length"))
	h.del([]byte("Set-Cookie"))
	h.del([]byte("Host"))
	h.del([]byte(consts.HeaderTrailer))
	h.del([]byte(consts.HeaderUserAgent))
	h.DelCookie("foo")
	h.Del("ccc")

	hv := h.Peek("aaa")
	if string(hv) != "bbb" {
		t.Fatalf("unexpected header value: %q. Expecting %q", hv, "bbb")
	}
	hv = h.Peek("ccc")
	if string(hv) != "" {
		t.Fatalf("unexpected header value: %q. Expecting %q", hv, "")
	}
	hv = h.Peek("Foo-Bar")
	if len(hv) > 0 {
		t.Fatalf("non-zero header value: %q", hv)
	}
	hv = h.Peek(consts.HeaderConnection)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	hv = h.Peek(consts.HeaderContentType)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	hv = h.Peek(consts.HeaderServer)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	hv = h.Peek(consts.HeaderContentLength)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	hv = h.FullCookie()
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	hv = h.Peek(consts.HeaderCookie)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	hv = h.Peek(consts.HeaderTrailer)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	hv = h.Peek(consts.HeaderUserAgent)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	if h.ContentLength() != 0 {
		t.Fatalf("unexpected content-length: %d. Expecting 0", h.ContentLength())
	}
}

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

	var h ResponseHeader
	h.Set("Foo-Bar", "baz")
	h.Set("aaa", "bbb")
	h.Set(consts.HeaderConnection, "keep-alive")
	h.Set(consts.HeaderContentType, "aaa")
	h.Set(consts.HeaderContentEncoding, "gzip")
	h.Set(consts.HeaderServer, "aaabbb")
	h.Set(consts.HeaderContentLength, "1123")
	h.Set(consts.HeaderTrailer, "foo, bar")

	var c Cookie
	c.SetKey("foo")
	c.SetValue("bar")
	h.SetCookie(&c)

	h.Del("foo-bar")
	h.Del("connection")
	h.DelBytes([]byte("content-type"))
	h.Del(consts.HeaderServer)
	h.Del("content-length")
	h.Del("set-cookie")
	h.Del("content-encoding")
	h.Del(consts.HeaderTrailer)

	hv := h.Peek("aaa")
	if string(hv) != "bbb" {
		t.Fatalf("unexpected header value: %q. Expecting %q", hv, "bbb")
	}
	hv = h.Peek("Foo-Bar")
	if len(hv) > 0 {
		t.Fatalf("non-zero header value: %q", hv)
	}
	hv = h.Peek(consts.HeaderConnection)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	hv = h.Peek(consts.HeaderContentType)
	if string(hv) != string(bytestr.DefaultContentType) {
		t.Fatalf("unexpected content-type: %q. Expecting %q", hv, bytestr.DefaultContentType)
	}
	hv = h.Peek(consts.HeaderContentEncoding)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	hv = h.Peek(consts.HeaderServer)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
	hv = h.Peek(consts.HeaderContentLength)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}

	hv = h.Peek(consts.HeaderTrailer)
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}

	if h.Cookie(&c) {
		t.Fatalf("unexpected cookie obtained: %v", &c)
	}

	if h.ContentLength() != 0 {
		t.Fatalf("unexpected content-length: %d. Expecting 0", h.ContentLength())
	}
}

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

	cookieName := "foobar"

	var h ResponseHeader
	c := AcquireCookie()
	c.SetKey(cookieName)
	c.SetValue("aasdfsdaf")
	h.SetCookie(c)

	h.DelClientCookieBytes([]byte(cookieName))
	if !h.Cookie(c) {
		t.Fatalf("expecting cookie %q", c.Key())
	}
	if !c.Expire().Equal(CookieExpireDelete) {
		t.Fatalf("unexpected cookie expiration time: %s. Expecting %s", c.Expire(), CookieExpireDelete)
	}
	if len(c.Value()) > 0 {
		t.Fatalf("unexpected cookie value: %q. Expecting empty value", c.Value())
	}
	ReleaseCookie(c)
}

func TestResponseHeaderResetConnectionClose(t *testing.T) {
	h := ResponseHeader{}
	h.Set(consts.HeaderConnection, "close")
	hv := h.Peek(consts.HeaderConnection)
	assert.DeepEqual(t, hv, []byte("close"))
	h.SetConnectionClose(true)
	h.ResetConnectionClose()
	assert.False(t, h.connectionClose)
	hv = h.Peek(consts.HeaderConnection)
	if len(hv) > 0 {
		t.Fatalf("ResetConnectionClose do not work,Connection: %q", hv)
	}
}

func TestRequestHeaderResetConnectionClose(t *testing.T) {
	h := RequestHeader{}
	h.Set(consts.HeaderConnection, "close")
	hv := h.Peek(consts.HeaderConnection)
	assert.DeepEqual(t, hv, []byte("close"))
	h.connectionClose = true
	h.ResetConnectionClose()
	assert.False(t, h.connectionClose)
	hv = h.Peek(consts.HeaderConnection)
	if len(hv) > 0 {
		t.Fatalf("ResetConnectionClose do not work,Connection: %q", hv)
	}
}

func TestCheckWriteHeaderCode(t *testing.T) {
	buffer := bytes.NewBuffer(make([]byte, 0, 1024))
	hlog.SetOutput(buffer)
	checkWriteHeaderCode(99)
	assert.True(t, strings.Contains(buffer.String(), "[Warn] HERTZ: Invalid StatusCode code"))
	buffer.Reset()
	checkWriteHeaderCode(600)
	assert.True(t, strings.Contains(buffer.String(), "[Warn] HERTZ: Invalid StatusCode code"))
	buffer.Reset()
	checkWriteHeaderCode(100)
	assert.False(t, strings.Contains(buffer.String(), "[Warn] HERTZ: Invalid StatusCode code"))
	buffer.Reset()
	checkWriteHeaderCode(599)
	assert.False(t, strings.Contains(buffer.String(), "[Warn] HERTZ: Invalid StatusCode code"))
}

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

	m := make(map[string]struct{})
	var h ResponseHeader
	h.Add("aaa", "bbb")
	h.Add("content-type", "xxx")
	h.SetContentEncoding("gzip")
	m["bbb"] = struct{}{}
	m["xxx"] = struct{}{}
	m["gzip"] = struct{}{}
	for i := 0; i < 10; i++ {
		v := fmt.Sprintf("%d", i)
		h.Add("Foo-Bar", v)
		m[v] = struct{}{}
	}
	if h.Len() != 13 {
		t.Fatalf("unexpected header len %d. Expecting 13", h.Len())
	}

	h.VisitAll(func(k, v []byte) {
		switch string(k) {
		case "Aaa", "Foo-Bar", "Content-Type", "Content-Encoding":
			if _, ok := m[string(v)]; !ok {
				t.Fatalf("unexpected value found %q. key %q", v, k)
			}
			delete(m, string(v))
		default:
			t.Fatalf("unexpected key found: %q", k)
		}
	})
	if len(m) > 0 {
		t.Fatalf("%d headers are missed", len(m))
	}
}

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

	m := make(map[string]struct{})
	var h RequestHeader
	h.Add("aaa", "bbb")
	h.Add("user-agent", "xxx")
	m["bbb"] = struct{}{}
	m["xxx"] = struct{}{}
	for i := 0; i < 10; i++ {
		v := fmt.Sprintf("%d", i)
		h.Add("Foo-Bar", v)
		m[v] = struct{}{}
	}
	if h.Len() != 12 {
		t.Fatalf("unexpected header len %d. Expecting 12", h.Len())
	}

	h.VisitAll(func(k, v []byte) {
		switch string(k) {
		case "Aaa", "Foo-Bar", "User-Agent":
			if _, ok := m[string(v)]; !ok {
				t.Fatalf("unexpected value found %q. key %q", v, k)
			}
			delete(m, string(v))
		default:
			t.Fatalf("unexpected key found: %q", k)
		}
	})
	if len(m) > 0 {
		t.Fatalf("%d headers are missed", len(m))
	}
}

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

	var h ResponseHeader
	h.Add("Content-Type", "test")

	got := string(h.Peek("Content-Type"))
	expected := "test"
	if got != expected {
		t.Errorf("expected %q got %q", expected, got)
	}

	if n := strings.Count(string(h.Header()), "Content-Type: "); n != 1 {
		t.Errorf("Content-Type occurred %d times", n)
	}
}

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

	var h ResponseHeader
	h.Add("Content-Encoding", "test")

	got := string(h.ContentEncoding())
	expected := "test"
	if got != expected {
		t.Errorf("expected %q got %q", expected, got)
	}

	if n := strings.Count(string(h.Header()), "Content-Encoding: "); n != 1 {
		t.Errorf("Content-Encoding occurred %d times", n)
	}
}

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

	var h RequestHeader
	h.Add("Content-Type", "test")

	got := string(h.Peek("Content-Type"))
	expected := "test"
	if got != expected {
		t.Errorf("expected %q got %q", expected, got)
	}

	if n := strings.Count(h.String(), "Content-Type: "); n != 1 {
		t.Errorf("Content-Type occurred %d times", n)
	}
}

func TestSetMultipartFormBoundary(t *testing.T) {
	h := RequestHeader{}
	h.SetMultipartFormBoundary("foo")
	assert.DeepEqual(t, h.contentType, []byte("multipart/form-data; boundary=foo"))
}

func TestRequestHeaderSetByteRange(t *testing.T) {
	var h RequestHeader
	h.SetByteRange(1, 5)
	hv := h.Peek(consts.HeaderRange)
	assert.DeepEqual(t, hv, []byte("bytes=1-5"))
}

func TestRequestHeaderSetMethodBytes(t *testing.T) {
	var h RequestHeader
	h.SetMethodBytes([]byte("foo"))
	assert.DeepEqual(t, h.Method(), []byte("foo"))
}

func TestRequestHeaderSetBytesKV(t *testing.T) {
	var h RequestHeader
	h.SetBytesKV([]byte("foo"), []byte("foo1"))
	hv := h.Peek("foo")
	assert.DeepEqual(t, hv, []byte("foo1"))
}

func TestResponseHeaderSetBytesV(t *testing.T) {
	var h ResponseHeader
	h.SetBytesV("foo", []byte("foo1"))
	hv := h.Peek("foo")
	assert.DeepEqual(t, hv, []byte("foo1"))
}

func TestRequestHeaderInitBufValue(t *testing.T) {
	var h RequestHeader
	slice := make([]byte, 0, 10)
	h.InitBufValue(10)
	assert.DeepEqual(t, cap(h.bufKV.value), cap(slice))
	assert.DeepEqual(t, h.GetBufValue(), slice)
}

func TestRequestHeaderDelAllCookies(t *testing.T) {
	var h RequestHeader
	h.SetCanonical([]byte(consts.HeaderSetCookie), []byte("foo2"))
	h.DelAllCookies()
	hv := h.FullCookie()
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
}

func TestResponseHeaderDelAllCookies(t *testing.T) {
	var h ResponseHeader
	h.SetCanonical([]byte(consts.HeaderSetCookie), []byte("foo"))
	h.DelAllCookies()
	hv := h.FullCookie()
	if len(hv) > 0 {
		t.Fatalf("non-zero value: %q", hv)
	}
}

func TestRequestHeaderSetNoDefaultContentType(t *testing.T) {
	var h RequestHeader
	h.SetMethod(http.MethodPost)
	b := h.AppendBytes(nil)
	assert.DeepEqual(t, b, []byte("POST / HTTP/1.1\r\nContent-Type: application/x-www-form-urlencoded\r\n\r\n"))
	h.SetNoDefaultContentType(true)
	b = h.AppendBytes(nil)
	assert.DeepEqual(t, b, []byte("POST / HTTP/1.1\r\n\r\n"))
}

func TestRequestHeader_PeekAll(t *testing.T) {
	t.Parallel()
	h := &RequestHeader{}
	h.Add(consts.HeaderConnection, "keep-alive")
	h.Add("Content-Type", "aaa")
	h.Add(consts.HeaderHost, "aaabbb")
	h.Add("User-Agent", "asdfas")
	h.Add("Content-Length", "1123")
	h.Add("Cookie", "foobar=baz")
	h.Add("aaa", "aaa")
	h.Add("aaa", "bbb")

	expectRequestHeaderAll(t, h, consts.HeaderConnection, [][]byte{[]byte("keep-alive")})
	expectRequestHeaderAll(t, h, "Content-Type", [][]byte{[]byte("aaa")})
	expectRequestHeaderAll(t, h, consts.HeaderHost, [][]byte{[]byte("aaabbb")})
	expectRequestHeaderAll(t, h, "User-Agent", [][]byte{[]byte("asdfas")})
	expectRequestHeaderAll(t, h, "Content-Length", [][]byte{[]byte("1123")})
	expectRequestHeaderAll(t, h, "Cookie", [][]byte{[]byte("foobar=baz")})
	expectRequestHeaderAll(t, h, "aaa", [][]byte{[]byte("aaa"), []byte("bbb")})

	h.DelBytes([]byte("Content-Type"))
	h.DelBytes([]byte((consts.HeaderHost)))
	h.DelBytes([]byte("aaa"))
	expectRequestHeaderAll(t, h, "Content-Type", [][]byte{})
	expectRequestHeaderAll(t, h, consts.HeaderHost, [][]byte{})
	expectRequestHeaderAll(t, h, "aaa", [][]byte{})
}

func expectRequestHeaderAll(t *testing.T, h *RequestHeader, key string, expectedValue [][]byte) {
	if len(h.PeekAll(key)) != len(expectedValue) {
		t.Fatalf("Unexpected size for key %q: %d. Expected %d", key, len(h.PeekAll(key)), len(expectedValue))
	}
	assert.DeepEqual(t, h.PeekAll(key), expectedValue)
}

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

	h := &ResponseHeader{}
	h.Add(consts.HeaderContentType, "aaa/bbb")
	h.Add(consts.HeaderContentEncoding, "gzip")
	h.Add(consts.HeaderConnection, "close")
	h.Add(consts.HeaderContentLength, "1234")
	h.Add(consts.HeaderServer, "aaaa")
	h.Add(consts.HeaderSetCookie, "cccc")
	h.Add("aaa", "aaa")
	h.Add("aaa", "bbb")

	expectResponseHeaderAll(t, h, consts.HeaderContentType, [][]byte{[]byte("aaa/bbb")})
	expectResponseHeaderAll(t, h, consts.HeaderContentEncoding, [][]byte{[]byte("gzip")})
	expectResponseHeaderAll(t, h, consts.HeaderConnection, [][]byte{[]byte("close")})
	expectResponseHeaderAll(t, h, consts.HeaderContentLength, [][]byte{[]byte("1234")})
	expectResponseHeaderAll(t, h, consts.HeaderServer, [][]byte{[]byte("aaaa")})
	expectResponseHeaderAll(t, h, consts.HeaderSetCookie, [][]byte{[]byte("cccc")})
	expectResponseHeaderAll(t, h, "aaa", [][]byte{[]byte("aaa"), []byte("bbb")})

	h.Del(consts.HeaderContentType)
	h.Del(consts.HeaderContentEncoding)
	expectResponseHeaderAll(t, h, consts.HeaderContentType, [][]byte{bytestr.DefaultContentType})
	expectResponseHeaderAll(t, h, consts.HeaderContentEncoding, [][]byte{})
}

func expectResponseHeaderAll(t *testing.T, h *ResponseHeader, key string, expectedValue [][]byte) {
	if len(h.PeekAll(key)) != len(expectedValue) {
		t.Fatalf("Unexpected size for key %q: %d. Expected %d", key, len(h.PeekAll(key)), len(expectedValue))
	}
	assert.DeepEqual(t, h.PeekAll(key), expectedValue)
}

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

	h, hCopy := &RequestHeader{}, &RequestHeader{}
	h.SetProtocol(consts.HTTP10)
	h.SetMethod(consts.MethodPatch)
	h.SetNoDefaultContentType(true)
	h.Add(consts.HeaderConnection, "keep-alive")
	h.Add("Content-Type", "aaa")
	h.Add(consts.HeaderHost, "aaabbb")
	h.Add("User-Agent", "asdfas")
	h.Add("Content-Length", "1123")
	h.Add("Cookie", "foobar=baz")
	h.Add("aaa", "aaa")
	h.Add("aaa", "bbb")

	h.CopyTo(hCopy)
	expectRequestHeaderAll(t, hCopy, consts.HeaderConnection, [][]byte{[]byte("keep-alive")})
	expectRequestHeaderAll(t, hCopy, "Content-Type", [][]byte{[]byte("aaa")})
	expectRequestHeaderAll(t, hCopy, consts.HeaderHost, [][]byte{[]byte("aaabbb")})
	expectRequestHeaderAll(t, hCopy, "User-Agent", [][]byte{[]byte("asdfas")})
	expectRequestHeaderAll(t, hCopy, "Content-Length", [][]byte{[]byte("1123")})
	expectRequestHeaderAll(t, hCopy, "Cookie", [][]byte{[]byte("foobar=baz")})
	expectRequestHeaderAll(t, hCopy, "aaa", [][]byte{[]byte("aaa"), []byte("bbb")})
	assert.DeepEqual(t, hCopy.GetProtocol(), consts.HTTP10)
	assert.DeepEqual(t, hCopy.noDefaultContentType, true)
	assert.DeepEqual(t, string(hCopy.Method()), consts.MethodPatch)
}

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

	h, hCopy := &ResponseHeader{}, &ResponseHeader{}
	h.SetProtocol(consts.HTTP10)
	h.SetHeaderLength(100)
	h.SetNoDefaultContentType(true)
	h.Add(consts.HeaderContentType, "aaa/bbb")
	h.Add(consts.HeaderContentEncoding, "gzip")
	h.Add(consts.HeaderConnection, "close")
	h.Add(consts.HeaderContentLength, "1234")
	h.Add(consts.HeaderServer, "aaaa")
	h.Add(consts.HeaderSetCookie, "cccc")
	h.Add("aaa", "aaa")
	h.Add("aaa", "bbb")

	h.CopyTo(hCopy)
	expectResponseHeaderAll(t, hCopy, consts.HeaderContentType, [][]byte{[]byte("aaa/bbb")})
	expectResponseHeaderAll(t, hCopy, consts.HeaderContentEncoding, [][]byte{[]byte("gzip")})
	expectResponseHeaderAll(t, hCopy, consts.HeaderConnection, [][]byte{[]byte("close")})
	expectResponseHeaderAll(t, hCopy, consts.HeaderContentLength, [][]byte{[]byte("1234")})
	expectResponseHeaderAll(t, hCopy, consts.HeaderServer, [][]byte{[]byte("aaaa")})
	expectResponseHeaderAll(t, hCopy, consts.HeaderSetCookie, [][]byte{[]byte("cccc")})
	expectResponseHeaderAll(t, hCopy, "aaa", [][]byte{[]byte("aaa"), []byte("bbb")})
	assert.DeepEqual(t, hCopy.GetProtocol(), consts.HTTP10)
	assert.DeepEqual(t, hCopy.noDefaultContentType, true)
	assert.DeepEqual(t, hCopy.GetHeaderLength(), 100)
}

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

	var h ResponseHeader
	h.noDefaultDate = true
	headers := string(h.Header())

	if strings.Contains(headers, "\r\nDate: ") {
		t.Fatalf("ResponseDateNoDefaultNotEmpty fail, response: \n%+v\noutcome: \n%q\n", h, headers) //nolint:govet
	}
}

func TestSetTrailerWithROString(t *testing.T) {
	h := &RequestHeader{}
	h.Add(consts.HeaderTrailer, "foo,bar,hertz")
	assert.DeepEqual(t, "Foo, Bar, Hertz", h.Get(consts.HeaderTrailer))

	h1 := &ResponseHeader{}
	h1.Add(consts.HeaderTrailer, "foo,bar,hertz")
	assert.DeepEqual(t, "Foo, Bar, Hertz", h1.Get(consts.HeaderTrailer))
}