summaryrefslogtreecommitdiff
path: root/chromium/net/quic/quic_multipath_transmissions_map_test.cc
blob: ec31910ba75a2e0428f9d539ebf6cddecf631313 (plain)
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
// Copyright (c) 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "net/quic/quic_multipath_transmissions_map.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace net {
namespace test {
namespace {

TEST(QuicAcrossPathsTransmissionMapTest, OnPacketRetransmittedOnDifferentPath) {
  QuicMultipathTransmissionsMap transmission_map;
  // Packet0's original transmission sent on path 1 with packet number 1.
  QuicPathIdPacketNumber packet0_0(1, 1);
  // Packet0's retransmission sent on path 2 with packet number 1.
  QuicPathIdPacketNumber packet0_1(2, 1);
  // packet0's 2nd retransmission sent on path 3 with packet number 1.
  QuicPathIdPacketNumber packet0_2(3, 1);

  transmission_map.OnPacketRetransmittedOnDifferentPath(packet0_0, packet0_1);
  const QuicMultipathTransmissionsMap::MultipathTransmissionsList*
      transmission_list1 =
          transmission_map.MaybeGetTransmissionsOnOtherPaths(packet0_0);
  EXPECT_EQ(packet0_0, (*transmission_list1)[0]);
  EXPECT_EQ(packet0_1, (*transmission_list1)[1]);

  transmission_map.OnPacketRetransmittedOnDifferentPath(packet0_1, packet0_2);
  const QuicMultipathTransmissionsMap::MultipathTransmissionsList*
      transmission_list2 =
          transmission_map.MaybeGetTransmissionsOnOtherPaths(packet0_0);
  EXPECT_EQ(packet0_0, (*transmission_list2)[0]);
  EXPECT_EQ(packet0_1, (*transmission_list2)[1]);
  EXPECT_EQ(packet0_2, (*transmission_list2)[2]);
  // Make sure there is no memory leakage.
}

TEST(QuicAcrossPathsTransmissionMapTest, MaybeGetTransmissionsOnOtherPaths) {
  QuicMultipathTransmissionsMap transmission_map;
  // Packet0's original transmission sent on path 1 with packet number 1.
  QuicPathIdPacketNumber packet0_0(1, 1);
  // Packet0's retransmission sent on path 2 with packet number 1.
  QuicPathIdPacketNumber packet0_1(2, 1);
  // packet0's 2nd retransmission sent on path 3 with packet number 1.
  QuicPathIdPacketNumber packet0_2(3, 1);

  transmission_map.OnPacketRetransmittedOnDifferentPath(packet0_0, packet0_1);
  transmission_map.OnPacketRetransmittedOnDifferentPath(packet0_1, packet0_2);

  const QuicMultipathTransmissionsMap::MultipathTransmissionsList*
      transmission_list1 =
          transmission_map.MaybeGetTransmissionsOnOtherPaths(packet0_0);
  const QuicMultipathTransmissionsMap::MultipathTransmissionsList*
      transmission_list2 =
          transmission_map.MaybeGetTransmissionsOnOtherPaths(packet0_1);
  const QuicMultipathTransmissionsMap::MultipathTransmissionsList*
      transmission_list3 =
          transmission_map.MaybeGetTransmissionsOnOtherPaths(packet0_2);
  // Make sure all three pointers point to the same list.
  EXPECT_EQ(transmission_list1, transmission_list2);
  EXPECT_EQ(transmission_list2, transmission_list3);
  EXPECT_EQ(packet0_0, (*transmission_list1)[0]);
  EXPECT_EQ(packet0_1, (*transmission_list1)[1]);
  EXPECT_EQ(packet0_2, (*transmission_list1)[2]);

  // Packet1 which is not transmitted across path.
  QuicPathIdPacketNumber packet1_0(1, 2);
  EXPECT_EQ(nullptr,
            transmission_map.MaybeGetTransmissionsOnOtherPaths(packet1_0));
  // Make sure there is no memory leakage.
}

TEST(QuicAcrossPathsTransmissionMapTest, OnPacketHandled) {
  QuicMultipathTransmissionsMap transmission_map;

  // Packet's original transmission sent on path 1 with packet number 1.
  QuicPathIdPacketNumber packet0_0(1, 1);
  // Packet's retransmission sent on path 2 with packet number 1.
  QuicPathIdPacketNumber packet0_1(2, 1);
  // packet's 2nd retransmission sent on path 3 with packet number 1.
  QuicPathIdPacketNumber packet0_2(3, 1);
  transmission_map.OnPacketRetransmittedOnDifferentPath(packet0_0, packet0_1);
  transmission_map.OnPacketRetransmittedOnDifferentPath(packet0_1, packet0_2);

  // Packet1's original transmission sent on path 1 with packet number 2.
  QuicPathIdPacketNumber packet1_0(1, 2);
  // Packet1's retransmission sent on path 2 with packet number 2.
  QuicPathIdPacketNumber packet1_1(2, 2);
  transmission_map.OnPacketRetransmittedOnDifferentPath(packet1_0, packet1_1);

  transmission_map.OnPacketHandled(packet0_0);
  EXPECT_EQ(nullptr,
            transmission_map.MaybeGetTransmissionsOnOtherPaths(packet0_0));
  EXPECT_EQ(nullptr,
            transmission_map.MaybeGetTransmissionsOnOtherPaths(packet0_1));
  EXPECT_EQ(nullptr,
            transmission_map.MaybeGetTransmissionsOnOtherPaths(packet0_2));
  const QuicMultipathTransmissionsMap::MultipathTransmissionsList*
      transmission_list =
          transmission_map.MaybeGetTransmissionsOnOtherPaths(packet1_0);
  EXPECT_EQ(packet1_0, (*transmission_list)[0]);
  EXPECT_EQ(packet1_1, (*transmission_list)[1]);
  // Packet 1 is received on path 2.
  transmission_map.OnPacketHandled(packet1_1);
  EXPECT_EQ(nullptr,
            transmission_map.MaybeGetTransmissionsOnOtherPaths(packet1_0));
  EXPECT_EQ(nullptr,
            transmission_map.MaybeGetTransmissionsOnOtherPaths(packet1_1));
  // Make sure there is no memory leakage.
}

}  // namespace
}  // namespace test
}  // namespace net