| Index: blimp/net/stream_packet_writer_unittest.cc
|
| diff --git a/blimp/net/stream_packet_writer_unittest.cc b/blimp/net/stream_packet_writer_unittest.cc
|
| index a5519bfaba741cb0bbfa1de71c9681e4df813014..127a19a2c5de7a19847a760c8c3ac67dfa02fe52 100644
|
| --- a/blimp/net/stream_packet_writer_unittest.cc
|
| +++ b/blimp/net/stream_packet_writer_unittest.cc
|
| @@ -6,6 +6,7 @@
|
|
|
| #include "base/message_loop/message_loop.h"
|
| #include "base/run_loop.h"
|
| +#include "blimp/net/blimp_connection_statistics.h"
|
| #include "blimp/net/common.h"
|
| #include "blimp/net/stream_packet_writer.h"
|
| #include "blimp/net/test_common.h"
|
| @@ -33,7 +34,7 @@ class StreamPacketWriterTest : public testing::Test {
|
| : test_data_(
|
| new net::DrainableIOBuffer(new net::StringIOBuffer(test_data_str_),
|
| test_data_str_.size())),
|
| - message_writer_(&socket_) {}
|
| + message_writer_(&socket_, &statistics_) {}
|
|
|
| protected:
|
| const std::string test_data_str_ = "U WOT M8";
|
| @@ -41,6 +42,7 @@ class StreamPacketWriterTest : public testing::Test {
|
|
|
| base::MessageLoop message_loop_;
|
| MockStreamSocket socket_;
|
| + BlimpConnectionStatistics statistics_;
|
| StreamPacketWriter message_writer_;
|
| testing::InSequence mock_sequence_;
|
|
|
| @@ -94,6 +96,9 @@ TEST_F(StreamPacketWriterTest, TestPartialWriteAsync) {
|
|
|
| message_writer_.WritePacket(test_data_, writer_cb.callback());
|
|
|
| + EXPECT_EQ(static_cast<int>(payload.size()),
|
| + statistics_.Get(BlimpConnectionStatistics::BYTES_SENT));
|
| +
|
| // Header is written - first one byte, then the remainder.
|
| header_cb.Run(1);
|
| header_cb.Run(header.size() - 1);
|
| @@ -198,7 +203,8 @@ TEST_F(StreamPacketWriterTest, DeletedDuringHeaderWrite) {
|
| net::TestCompletionCallback writer_cb;
|
| net::CompletionCallback header_cb;
|
| net::CompletionCallback payload_cb;
|
| - std::unique_ptr<StreamPacketWriter> writer(new StreamPacketWriter(&socket_));
|
| + std::unique_ptr<StreamPacketWriter> writer(
|
| + new StreamPacketWriter(&socket_, &statistics_));
|
|
|
| // Write header.
|
| EXPECT_CALL(socket_, Write(BufferEquals(EncodeHeader(test_data_str_.size())),
|
| @@ -218,7 +224,8 @@ TEST_F(StreamPacketWriterTest, DeletedDuringPayloadWrite) {
|
| net::TestCompletionCallback writer_cb;
|
| net::CompletionCallback header_cb;
|
| net::CompletionCallback payload_cb;
|
| - std::unique_ptr<StreamPacketWriter> writer(new StreamPacketWriter(&socket_));
|
| + std::unique_ptr<StreamPacketWriter> writer(
|
| + new StreamPacketWriter(&socket_, &statistics_));
|
|
|
| EXPECT_CALL(socket_, Write(BufferEquals(EncodeHeader(test_data_str_.size())),
|
| kPacketHeaderSizeBytes, _))
|
|
|