72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
|
package udp
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
var table = []struct {
|
||
|
|
data []byte
|
||
|
|
values map[string]string
|
||
|
|
err error
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
[]byte{0x2, 0x5, '/', '?', 'a', '=', 'b'},
|
||
|
|
map[string]string{"a": "b"},
|
||
|
|
nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
[]byte{0x2, 0x0},
|
||
|
|
map[string]string{},
|
||
|
|
nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
[]byte{0x2, 0x1},
|
||
|
|
nil,
|
||
|
|
errMalformedPacket,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
[]byte{0x2},
|
||
|
|
nil,
|
||
|
|
errMalformedPacket,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
[]byte{0x2, 0x8, '/', 'c', '/', 'd', '?', 'a', '=', 'b'},
|
||
|
|
map[string]string{"a": "b"},
|
||
|
|
nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
[]byte{0x2, 0x2, '/', '?', 0x2, 0x3, 'a', '=', 'b'},
|
||
|
|
map[string]string{"a": "b"},
|
||
|
|
nil,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
[]byte{0x2, 0x9, '/', '?', 'a', '=', 'b', '%', '2', '0', 'c'},
|
||
|
|
map[string]string{"a": "b c"},
|
||
|
|
nil,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHandleOptionalParameters(t *testing.T) {
|
||
|
|
for _, testCase := range table {
|
||
|
|
params, err := handleOptionalParameters(testCase.data)
|
||
|
|
if err != testCase.err {
|
||
|
|
if testCase.err == nil {
|
||
|
|
t.Fatalf("expected no parsing error for %x but got %s", testCase.data, err)
|
||
|
|
} else {
|
||
|
|
t.Fatalf("expected parsing error for %x", testCase.data)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if testCase.values != nil {
|
||
|
|
if params == nil {
|
||
|
|
t.Fatalf("expected values %v for %x", testCase.values, testCase.data)
|
||
|
|
} else {
|
||
|
|
for key, want := range testCase.values {
|
||
|
|
if got, ok := params.String(key); !ok {
|
||
|
|
t.Fatalf("params missing entry %s for data %x", key, testCase.data)
|
||
|
|
} else if got != want {
|
||
|
|
t.Fatalf("expected param %s=%s, but was %s for data %x", key, want, got, testCase.data)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|