You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
87 lines
1.6 KiB
87 lines
1.6 KiB
package netutil
|
|
|
|
import (
|
|
"net/url"
|
|
"reflect"
|
|
"testing"
|
|
)
|
|
|
|
func TestMergeQuery(t *testing.T) {
|
|
tests := []struct {
|
|
u string
|
|
q url.Values
|
|
w string
|
|
}{
|
|
// No values
|
|
{
|
|
u: "http://example.com",
|
|
q: nil,
|
|
w: "http://example.com",
|
|
},
|
|
// No additional values
|
|
{
|
|
u: "http://example.com?foo=bar",
|
|
q: nil,
|
|
w: "http://example.com?foo=bar",
|
|
},
|
|
// Simple addition
|
|
{
|
|
u: "http://example.com",
|
|
q: url.Values{
|
|
"foo": []string{"bar"},
|
|
},
|
|
w: "http://example.com?foo=bar",
|
|
},
|
|
// Addition with existing values
|
|
{
|
|
u: "http://example.com?dog=boo",
|
|
q: url.Values{
|
|
"foo": []string{"bar"},
|
|
},
|
|
w: "http://example.com?dog=boo&foo=bar",
|
|
},
|
|
// Merge
|
|
{
|
|
u: "http://example.com?dog=boo",
|
|
q: url.Values{
|
|
"dog": []string{"elroy"},
|
|
},
|
|
w: "http://example.com?dog=boo&dog=elroy",
|
|
},
|
|
// Add and merge
|
|
{
|
|
u: "http://example.com?dog=boo",
|
|
q: url.Values{
|
|
"dog": []string{"elroy"},
|
|
"foo": []string{"bar"},
|
|
},
|
|
w: "http://example.com?dog=boo&dog=elroy&foo=bar",
|
|
},
|
|
// Multivalue merge
|
|
{
|
|
u: "http://example.com?dog=boo",
|
|
q: url.Values{
|
|
"dog": []string{"elroy", "penny"},
|
|
},
|
|
w: "http://example.com?dog=boo&dog=elroy&dog=penny",
|
|
},
|
|
}
|
|
|
|
for i, tt := range tests {
|
|
ur, err := url.Parse(tt.u)
|
|
if err != nil {
|
|
t.Errorf("case %d: failed parsing test url: %v, error: %v", i, tt.u, err)
|
|
}
|
|
|
|
got := MergeQuery(*ur, tt.q)
|
|
want, err := url.Parse(tt.w)
|
|
if err != nil {
|
|
t.Errorf("case %d: failed parsing want url: %v, error: %v", i, tt.w, err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(*want, got) {
|
|
t.Errorf("case %d: want: %v, got: %v", i, *want, got)
|
|
}
|
|
}
|
|
}
|