1
0
mirror of https://github.com/golang/go synced 2024-11-11 17:51:49 -07:00

net/http: add Request.CookiesNamed

Implements a new method http.Request.CookiesName, that allows
retrieving all cookies that match the given name.

Fixes #61472

Change-Id: I405d8771b4195af9ff6b4dfde3cfcd316c23b70c
GitHub-Last-Rev: 6ad0094995
GitHub-Pull-Request: golang/go#61473
Reviewed-on: https://go-review.googlesource.com/c/go/+/511516
Reviewed-by: Emmanuel Odeke <emmanuel@orijtech.com>
LUCI-TryBot-Result: Go LUCI <golang-scoped@luci-project-accounts.iam.gserviceaccount.com>
Reviewed-by: Carlos Amedee <carlos@golang.org>
Reviewed-by: Damien Neil <dneil@google.com>
This commit is contained in:
Timo Furrer 2024-02-28 06:07:16 +00:00 committed by Emmanuel Odeke
parent db423dde85
commit 6dca707656
3 changed files with 81 additions and 0 deletions

1
api/next/61472.txt Normal file
View File

@ -0,0 +1 @@
pkg net/http, method (*Request) CookiesNamed(string) []*Cookie #61472

View File

@ -431,6 +431,15 @@ func (r *Request) Cookies() []*Cookie {
return readCookies(r.Header, "")
}
// CookiesNamed parses and returns the named HTTP cookies sent with the request
// or an empty slice if none matched.
func (r *Request) CookiesNamed(name string) []*Cookie {
if name == "" {
return []*Cookie{}
}
return readCookies(r.Header, name)
}
// ErrNoCookie is returned by Request's Cookie method when a cookie is not found.
var ErrNoCookie = errors.New("http: named cookie not present")

View File

@ -10,6 +10,7 @@ import (
"context"
"crypto/rand"
"encoding/base64"
"encoding/json"
"errors"
"fmt"
"io"
@ -1256,6 +1257,76 @@ func TestRequestCookie(t *testing.T) {
}
}
func TestRequestCookiesByName(t *testing.T) {
tests := []struct {
in []*Cookie
filter string
want []*Cookie
}{
{
in: []*Cookie{
{Name: "foo", Value: "foo-1"},
{Name: "bar", Value: "bar"},
},
filter: "foo",
want: []*Cookie{{Name: "foo", Value: "foo-1"}},
},
{
in: []*Cookie{
{Name: "foo", Value: "foo-1"},
{Name: "foo", Value: "foo-2"},
{Name: "bar", Value: "bar"},
},
filter: "foo",
want: []*Cookie{
{Name: "foo", Value: "foo-1"},
{Name: "foo", Value: "foo-2"},
},
},
{
in: []*Cookie{
{Name: "bar", Value: "bar"},
},
filter: "foo",
want: []*Cookie{},
},
{
in: []*Cookie{
{Name: "bar", Value: "bar"},
},
filter: "",
want: []*Cookie{},
},
{
in: []*Cookie{},
filter: "foo",
want: []*Cookie{},
},
}
for _, tt := range tests {
t.Run(tt.filter, func(t *testing.T) {
req, err := NewRequest("GET", "http://example.com/", nil)
if err != nil {
t.Fatal(err)
}
for _, c := range tt.in {
req.AddCookie(c)
}
got := req.CookiesNamed(tt.filter)
if !reflect.DeepEqual(got, tt.want) {
asStr := func(v any) string {
blob, _ := json.MarshalIndent(v, "", " ")
return string(blob)
}
t.Fatalf("Result mismatch\n\tGot: %s\n\tWant: %s", asStr(got), asStr(tt.want))
}
})
}
}
const (
fileaContents = "This is a test file."
filebContents = "Another test file."