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
|
package ackhandler
import (
"maps"
"testing"
"time"
"github.com/quic-go/quic-go/internal/monotime"
"github.com/quic-go/quic-go/internal/protocol"
"github.com/stretchr/testify/require"
)
func TestLostPacketTracker(t *testing.T) {
lt := newLostPacketTracker(4)
start := monotime.Now()
lt.Add(1, start)
lt.Add(5, start.Add(time.Second))
lt.Add(8, start.Add(2*time.Second))
require.Equal(t, map[protocol.PacketNumber]monotime.Time{
1: start,
5: start.Add(time.Second),
8: start.Add(2 * time.Second),
}, maps.Collect(lt.All()))
// Lose 2 more packets. The first one should be removed.
lt.Add(10, start.Add(3*time.Second))
lt.Add(11, start.Add(4*time.Second))
require.Equal(t, map[protocol.PacketNumber]monotime.Time{
5: start.Add(time.Second),
8: start.Add(2 * time.Second),
10: start.Add(3 * time.Second),
11: start.Add(4 * time.Second),
}, maps.Collect(lt.All()))
lt.Delete(5)
lt.Delete(10)
require.Equal(t, map[protocol.PacketNumber]monotime.Time{
8: start.Add(2 * time.Second),
11: start.Add(4 * time.Second),
}, maps.Collect(lt.All()))
}
func TestLostPacketTrackerDeleteBefore(t *testing.T) {
lt := newLostPacketTracker(4)
trackedPackets := func(lt *lostPacketTracker) []protocol.PacketNumber {
var pns []protocol.PacketNumber
for pn := range lt.All() {
pns = append(pns, pn)
}
return pns
}
start := monotime.Now()
lt.Add(1, start)
lt.Add(5, start.Add(time.Second))
lt.Add(8, start.Add(2*time.Second))
lt.Add(10, start.Add(3*time.Second))
require.Equal(t, []protocol.PacketNumber{1, 5, 8, 10}, trackedPackets(lt))
lt.DeleteBefore(start) // this should be a no-op
require.Equal(t, []protocol.PacketNumber{1, 5, 8, 10}, trackedPackets(lt))
lt.DeleteBefore(start.Add(2 * time.Second))
require.Equal(t, []protocol.PacketNumber{8, 10}, trackedPackets(lt))
lt.DeleteBefore(start.Add(time.Second * 5 / 2))
require.Equal(t, []protocol.PacketNumber{10}, trackedPackets(lt))
lt.DeleteBefore(start.Add(time.Hour))
require.Empty(t, trackedPackets(lt))
}
|
