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
|
package gcc
import (
"testing"
"time"
"github.com/pion/interceptor/internal/cc"
"github.com/stretchr/testify/assert"
)
func TestArrivalGroup(t *testing.T) {
cases := []struct {
name string
acks []cc.Acknowledgment
expected arrivalGroup
}{
{
name: "createsEmptyArrivalGroup",
acks: []cc.Acknowledgment{},
expected: arrivalGroup{
packets: nil,
arrival: time.Time{},
departure: time.Time{},
},
},
{
name: "createsArrivalGroupContainingSingleACK",
acks: []cc.Acknowledgment{{
SequenceNumber: 0,
Size: 0,
Departure: time.Time{},
Arrival: time.Time{},
}},
expected: arrivalGroup{
packets: []cc.Acknowledgment{{
SequenceNumber: 0,
Size: 0,
Departure: time.Time{},
Arrival: time.Time{},
}},
arrival: time.Time{},
departure: time.Time{},
},
},
{
name: "setsTimesToLastACK",
acks: []cc.Acknowledgment{{
SequenceNumber: 0,
Size: 0,
Departure: time.Time{},
Arrival: time.Time{},
}, {
SequenceNumber: 0,
Size: 0,
Departure: time.Time{}.Add(time.Second),
Arrival: time.Time{}.Add(time.Second),
}},
expected: arrivalGroup{
packets: []cc.Acknowledgment{{
SequenceNumber: 0,
Size: 0,
Departure: time.Time{},
Arrival: time.Time{},
}, {
SequenceNumber: 0,
Size: 0,
Departure: time.Time{}.Add(time.Second),
Arrival: time.Time{}.Add(time.Second),
}},
arrival: time.Time{}.Add(time.Second),
departure: time.Time{}.Add(time.Second),
},
},
}
for _, tc := range cases {
tc := tc
t.Run(tc.name, func(t *testing.T) {
ag := arrivalGroup{}
for _, ack := range tc.acks {
ag.add(ack)
}
assert.Equal(t, tc.expected, ag)
})
}
}
|