fasthttp/cookie_test.go

package fasthttp

import (
	"strings"
	"testing"
)

func TestParseCookies(t *testing.T) {
	testParseCookies(t, "", "")
	testParseCookies(t, "=", "")
	testParseCookies(t, "foo", "foo")
	testParseCookies(t, "=foo", "foo")
	testParseCookies(t, "bar=", "bar=")
	testParseCookies(t, "xxx=aa;bb=c; =d; ;;e=g", "xxx=aa; bb=c; d; e=g")
	testParseCookies(t, "a;b;c; d=1;d=2", "c; d=2")
	testParseCookies(t, "   %D0%B8%D0%B2%D0%B5%D1%82=a%20b%3Bc   ;s%20s=aaa  ", "%D0%B8%D0%B2%D0%B5%D1%82=a%20b%3Bc; s%20s=aaa")
}

func testParseCookies(t *testing.T, s, expectedS string) {
	var kv argsKV
	cookies := parseCookies(nil, []byte(s), &kv)
	ss := string(appendCookieBytes(nil, cookies))
	if ss != expectedS {
		t.Fatalf("Unexpected cookies after parsing: %q. Expected %q. String to parse %q", ss, expectedS, s)
	}
}

func TestAppendCookieBytes(t *testing.T) {
	testAppendCookieBytes(t, "=", "")
	testAppendCookieBytes(t, "foo=", "foo=")
	testAppendCookieBytes(t, "=bar", "bar")
	testAppendCookieBytes(t, "привет=a b;c&s s=aaa", "%D0%BF%D1%80%D0%B8%D0%B2%D0%B5%D1%82=a%20b%3Bc; s%20s=aaa")
}

func testAppendCookieBytes(t *testing.T, s, expectedS string) {
	var cookies []argsKV
	for _, ss := range strings.Split(s, "&") {
		tmp := strings.SplitN(ss, "=", 2)
		if len(tmp) != 2 {
			t.Fatalf("Cannot find '=' in %q, part of %q", ss, s)
		}
		cookies = append(cookies, argsKV{
			key:   []byte(tmp[0]),
			value: []byte(tmp[1]),
		})
	}

	prefix := "foobar"
	result := string(appendCookieBytes([]byte(prefix), cookies))
	if result[:len(prefix)] != prefix {
		t.Fatalf("unexpected prefix %q. Expected %q for cookie %q", result[:len(prefix)], prefix, s)
	}
	result = result[len(prefix):]
	if result != expectedS {
		t.Fatalf("Unexpected result %q. Expected %q for cookie %q", result, expectedS, s)
	}
}