1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115
|
package rtcp
import (
"errors"
"reflect"
"testing"
)
func TestHeaderUnmarshal(t *testing.T) {
for _, test := range []struct {
Name string
Data []byte
Want Header
WantError error
}{
{
Name: "valid",
Data: []byte{
// v=2, p=0, count=1, RR, len=7
0x81, 0xc9, 0x00, 0x07,
},
Want: Header{
Padding: false,
Count: 1,
Type: TypeReceiverReport,
Length: 7,
},
},
{
Name: "also valid",
Data: []byte{
// v=2, p=1, count=1, BYE, len=7
0xa1, 0xcc, 0x00, 0x07,
},
Want: Header{
Padding: true,
Count: 1,
Type: TypeApplicationDefined,
Length: 7,
},
},
{
Name: "bad version",
Data: []byte{
// v=0, p=0, count=0, RR, len=4
0x00, 0xc9, 0x00, 0x04,
},
WantError: errBadVersion,
},
} {
var h Header
err := h.Unmarshal(test.Data)
if got, want := err, test.WantError; !errors.Is(got, want) {
t.Fatalf("Unmarshal %q header: err = %v, want %v", test.Name, got, want)
}
if err != nil {
continue
}
if got, want := h, test.Want; !reflect.DeepEqual(got, want) {
t.Fatalf("Unmarshal %q header: got %v, want %v", test.Name, got, want)
}
}
}
func TestHeaderRoundTrip(t *testing.T) {
for _, test := range []struct {
Name string
Header Header
WantError error
}{
{
Name: "valid",
Header: Header{
Padding: true,
Count: 31,
Type: TypeSenderReport,
Length: 4,
},
},
{
Name: "also valid",
Header: Header{
Padding: false,
Count: 28,
Type: TypeReceiverReport,
Length: 65535,
},
},
{
Name: "invalid count",
Header: Header{
Count: 40,
},
WantError: errInvalidHeader,
},
} {
data, err := test.Header.Marshal()
if got, want := err, test.WantError; !errors.Is(got, want) {
t.Errorf("Marshal %q: err = %v, want %v", test.Name, got, want)
}
if err != nil {
continue
}
var decoded Header
if err := decoded.Unmarshal(data); err != nil {
t.Errorf("Unmarshal %q: %v", test.Name, err)
}
if got, want := decoded, test.Header; !reflect.DeepEqual(got, want) {
t.Errorf("%q header round trip: got %#v, want %#v", test.Name, got, want)
}
}
}
|