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
|
package gcc
import (
"testing"
"time"
"github.com/pion/interceptor/internal/cc"
"github.com/stretchr/testify/assert"
)
func TestRateCalculator(t *testing.T) {
t0 := time.Now()
cases := []struct {
name string
acks []cc.Acknowledgment
expected []int
}{
{
name: "emptyCreatesNoRate",
acks: []cc.Acknowledgment{},
expected: []int{},
},
{
name: "ignoresZeroArrivalTimes",
acks: []cc.Acknowledgment{{
SequenceNumber: 0,
Size: 0,
Departure: time.Time{},
Arrival: time.Time{},
}},
expected: []int{},
},
{
name: "singleAckCreatesRate",
acks: []cc.Acknowledgment{{
SequenceNumber: 0,
Size: 1000,
Departure: time.Time{},
Arrival: t0,
}},
expected: []int{8000},
},
{
name: "twoAcksCalculateCorrectRates",
acks: []cc.Acknowledgment{{
SequenceNumber: 0,
Size: 125,
Departure: time.Time{},
Arrival: t0,
}, {
SequenceNumber: 0,
Size: 125,
Departure: time.Time{},
Arrival: t0.Add(100 * time.Millisecond),
}},
expected: []int{1000, 20_000},
},
{
name: "steadyACKsCalculateCorrectRates",
acks: getACKStream(10, 1200, 100*time.Millisecond),
expected: []int{
9_600,
192_000,
144_000,
128_000,
120_000,
115_200,
115_200,
115_200,
115_200,
115_200,
},
},
}
for _, tc := range cases {
tc := tc
t.Run(tc.name, func(t *testing.T) {
rc := newRateCalculator(500 * time.Millisecond)
in := make(chan []cc.Acknowledgment)
out := make(chan int)
onRateUpdate := func(rate int) {
out <- rate
}
go func() {
defer close(out)
rc.run(in, onRateUpdate)
}()
go func() {
in <- tc.acks
close(in)
}()
received := []int{}
for r := range out {
received = append(received, r)
}
assert.Equal(t, tc.expected, received)
})
}
}
func getACKStream(length int, size int, interval time.Duration) []cc.Acknowledgment {
res := []cc.Acknowledgment{}
t0 := time.Now()
for i := 0; i < length; i++ {
res = append(res, cc.Acknowledgment{
Size: size,
Arrival: t0,
})
t0 = t0.Add(interval)
}
return res
}
|
