File: rtpdump_test.go

package info (click to toggle)
golang-github-pion-webrtc.v3 3.1.56-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 2,392 kB
  • sloc: javascript: 595; sh: 28; makefile: 5
file content (119 lines) | stat: -rw-r--r-- 2,167 bytes parent folder | download | duplicates (2)
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
116
117
118
119
package rtpdump

import (
	"errors"
	"net"
	"reflect"
	"testing"
	"time"
)

func TestHeaderRoundTrip(t *testing.T) {
	for _, test := range []struct {
		Header Header
	}{
		{
			Header: Header{
				Start:  time.Unix(0, 0).UTC(),
				Source: net.IPv4(0, 0, 0, 0),
				Port:   0,
			},
		},
		{
			Header: Header{
				Start:  time.Date(2019, 3, 25, 1, 1, 1, 0, time.UTC),
				Source: net.IPv4(1, 2, 3, 4),
				Port:   8080,
			},
		},
	} {
		d, err := test.Header.Marshal()
		if err != nil {
			t.Fatal(err)
		}

		var hdr Header
		if err := hdr.Unmarshal(d); err != nil {
			t.Fatal(err)
		}

		if got, want := hdr, test.Header; !reflect.DeepEqual(got, want) {
			t.Fatalf("Unmarshal(%v.Marshal()) = %v, want identical", got, want)
		}
	}
}

func TestMarshalHeader(t *testing.T) {
	for _, test := range []struct {
		Name    string
		Header  Header
		Want    []byte
		WantErr error
	}{
		{
			Name: "nil source",
			Header: Header{
				Start:  time.Unix(0, 0).UTC(),
				Source: nil,
				Port:   0,
			},
			Want: []byte{
				0x00, 0x00, 0x00, 0x00,
				0x00, 0x00, 0x00, 0x00,
				0x00, 0x00, 0x00, 0x00,
				0x00, 0x00, 0x00, 0x00,
			},
		},
	} {
		data, err := test.Header.Marshal()
		if got, want := err, test.WantErr; !errors.Is(got, want) {
			t.Fatalf("Marshal(%q) err=%v, want %v", test.Name, got, want)
		}

		if got, want := data, test.Want; !reflect.DeepEqual(got, want) {
			t.Fatalf("Marshal(%q) = %v, want %v", test.Name, got, want)
		}
	}
}

func TestPacketRoundTrip(t *testing.T) {
	for _, test := range []struct {
		Packet Packet
	}{
		{
			Packet: Packet{
				Offset:  0,
				IsRTCP:  false,
				Payload: []byte{0},
			},
		},
		{
			Packet: Packet{
				Offset:  0,
				IsRTCP:  true,
				Payload: []byte{0},
			},
		},
		{
			Packet: Packet{
				Offset:  123 * time.Millisecond,
				IsRTCP:  false,
				Payload: []byte{1, 2, 3, 4},
			},
		},
	} {
		d, err := test.Packet.Marshal()
		if err != nil {
			t.Fatal(err)
		}

		var pkt Packet
		if err := pkt.Unmarshal(d); err != nil {
			t.Fatal(err)
		}

		if got, want := pkt, test.Packet; !reflect.DeepEqual(got, want) {
			t.Fatalf("Unmarshal(%v.Marshal()) = %v, want identical", got, want)
		}
	}
}