| Index: net/quic/quic_packet_creator_test.cc
|
| diff --git a/net/quic/quic_packet_creator_test.cc b/net/quic/quic_packet_creator_test.cc
|
| index c03913d383c2f1df8b0dd739ef11784ab0ff77a3..d4835b1b36e0c5fe01c1b6c769b35aba3cc8d04a 100644
|
| --- a/net/quic/quic_packet_creator_test.cc
|
| +++ b/net/quic/quic_packet_creator_test.cc
|
| @@ -32,6 +32,9 @@ class QuicPacketCreatorTest : public ::testing::Test {
|
| }
|
| ~QuicPacketCreatorTest() {
|
| STLDeleteValues(&packets_);
|
| + for (QuicFrames::iterator it = frames_.begin(); it != frames_.end(); ++it) {
|
| + QuicConnection::DeleteEnclosedFrame(&(*it));
|
| + }
|
| }
|
|
|
| void ProcessPackets() {
|
| @@ -43,6 +46,7 @@ class QuicPacketCreatorTest : public ::testing::Test {
|
| }
|
|
|
| vector<QuicPacketCreator::PacketPair> packets_;
|
| + QuicFrames frames_;
|
| QuicFramer framer_;
|
| testing::StrictMock<MockFramerVisitor> framer_visitor_;
|
| QuicStreamId id_;
|
| @@ -53,7 +57,8 @@ class QuicPacketCreatorTest : public ::testing::Test {
|
| };
|
|
|
| TEST_F(QuicPacketCreatorTest, DataToStreamBasic) {
|
| - size_t bytes_consumed = utils_.DataToStream(id_, data_, 0, true, &packets_);
|
| + size_t bytes_consumed = utils_.DataToStream(
|
| + id_, data_, 0, true, &packets_, &frames_);
|
|
|
| ASSERT_EQ(1u, packets_.size());
|
| ASSERT_EQ(1u, utils_.sequence_number());
|
| @@ -70,7 +75,8 @@ TEST_F(QuicPacketCreatorTest, DataToStreamBasic) {
|
|
|
| TEST_F(QuicPacketCreatorTest, DataToStreamFec) {
|
| utils_.options()->use_fec = true;
|
| - size_t bytes_consumed = utils_.DataToStream(id_, data_, 0, true, &packets_);
|
| + size_t bytes_consumed = utils_.DataToStream(
|
| + id_, data_, 0, true, &packets_, &frames_);
|
|
|
| ASSERT_EQ(2u, packets_.size());
|
| ASSERT_EQ(2u, utils_.sequence_number());
|
| @@ -93,7 +99,8 @@ TEST_F(QuicPacketCreatorTest, DataToStreamFec) {
|
|
|
| TEST_F(QuicPacketCreatorTest, DataToStreamFecHandled) {
|
| utils_.options()->use_fec = true;
|
| - size_t bytes_consumed = utils_.DataToStream(id_, data_, 0, true, &packets_);
|
| + size_t bytes_consumed = utils_.DataToStream(
|
| + id_, data_, 0, true, &packets_, &frames_);
|
| ASSERT_EQ(data_.size(), bytes_consumed);
|
|
|
| ASSERT_EQ(2u, packets_.size());
|
| @@ -131,7 +138,8 @@ TEST_F(QuicPacketCreatorTest, DataToStreamFecHandled) {
|
| }
|
|
|
| TEST_F(QuicPacketCreatorTest, DataToStreamSkipFin) {
|
| - size_t bytes_consumed = utils_.DataToStream(id_, data_, 0, false, &packets_);
|
| + size_t bytes_consumed = utils_.DataToStream(
|
| + id_, data_, 0, false, &packets_, &frames_);
|
| ASSERT_EQ(data_.size(), bytes_consumed);
|
|
|
| ASSERT_EQ(1u, packets_.size());
|
| @@ -149,7 +157,8 @@ TEST_F(QuicPacketCreatorTest, DataToStreamSkipFin) {
|
| TEST_F(QuicPacketCreatorTest, NoData) {
|
| data_ = "";
|
|
|
| - size_t bytes_consumed = utils_.DataToStream(id_, data_, 0, true, &packets_);
|
| + size_t bytes_consumed = utils_.DataToStream(
|
| + id_, data_, 0, true, &packets_, &frames_);
|
| ASSERT_EQ(data_.size(), bytes_consumed);
|
|
|
| ASSERT_EQ(1u, packets_.size());
|
| @@ -169,7 +178,8 @@ TEST_F(QuicPacketCreatorTest, MultiplePackets) {
|
| utils_.options()->max_packet_length =
|
| ciphertext_size + QuicUtils::StreamFramePacketOverhead(1);
|
|
|
| - size_t bytes_consumed = utils_.DataToStream(id_, data_, 0, true, &packets_);
|
| + size_t bytes_consumed = utils_.DataToStream(
|
| + id_, data_, 0, true, &packets_, &frames_);
|
| ASSERT_EQ(data_.size(), bytes_consumed);
|
|
|
| ASSERT_EQ(2u, packets_.size());
|
| @@ -198,7 +208,8 @@ TEST_F(QuicPacketCreatorTest, MultiplePacketsWithLimits) {
|
| ciphertext_size + QuicUtils::StreamFramePacketOverhead(1);
|
| utils_.options()->max_num_packets = 1;
|
|
|
| - size_t bytes_consumed = utils_.DataToStream(id_, data_, 0, true, &packets_);
|
| + size_t bytes_consumed = utils_.DataToStream(
|
| + id_, data_, 0, true, &packets_, &frames_);
|
| ASSERT_EQ(kPayloadBytesPerPacket, bytes_consumed);
|
|
|
| ASSERT_EQ(1u, packets_.size());
|
|
|