Added a debug info UI for Blimp

blimp/net/stream_packet_writer_unittest.cc

 // Copyright 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 <string>
 
 #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"
 #include "net/base/io_buffer.h"
 #include "net/base/net_errors.h"
 #include "net/base/test_completion_callback.h"
 #include "net/socket/socket.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using testing::_;
 using testing::DoAll;
 using testing::InSequence;
 using testing::Mock;
 using testing::NotNull;
 using testing::Return;
 using testing::SaveArg;
 
 namespace blimp {
 namespace {
 
 class StreamPacketWriterTest : public testing::Test {
  public:
   StreamPacketWriterTest()
       : 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";
   scoped_refptr<net::DrainableIOBuffer> test_data_;
 
   base::MessageLoop message_loop_;
   MockStreamSocket socket_;
+  BlimpConnectionStatistics statistics_;
   StreamPacketWriter message_writer_;
   testing::InSequence mock_sequence_;
 
  private:
   DISALLOW_COPY_AND_ASSIGN(StreamPacketWriterTest);
 };
 
 // Successful write with 1 async header write and 1 async payload write.
 TEST_F(StreamPacketWriterTest, TestWriteAsync) {
   net::TestCompletionCallback writer_cb;
@@ skipping 33 matching lines @@
       .WillOnce(DoAll(SaveArg<2>(&payload_cb), Return(net::ERR_IO_PENDING)))
       .RetiresOnSaturation();
   EXPECT_CALL(socket_,
               Write(BufferEquals(payload.substr(1, payload.size() - 1)),
                     payload.size() - 1, _))
       .WillOnce(DoAll(SaveArg<2>(&payload_cb), Return(net::ERR_IO_PENDING)))
       .RetiresOnSaturation();
 
   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);
 
   // Payload is written - first one byte, then the remainder.
   payload_cb.Run(1);
   payload_cb.Run(payload.size() - 1);
 
   EXPECT_EQ(net::OK, writer_cb.WaitForResult());
 }
@@ skipping 84 matching lines @@
   message_writer_.WritePacket(test_data_, writer_cb.callback());
   EXPECT_EQ(net::ERR_FAILED, writer_cb.WaitForResult());
 }
 
 // Verify that asynchronous header write completions don't cause a
 // use-after-free error if the writer object is deleted.
 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())),
                              kPacketHeaderSizeBytes, _))
       .WillOnce(DoAll(SaveArg<2>(&header_cb), Return(net::ERR_IO_PENDING)));
   writer->WritePacket(test_data_, writer_cb.callback());
   Mock::VerifyAndClearExpectations(&socket_);
 
   // Header write completion callback is invoked after the writer died.
   writer.reset();
   header_cb.Run(kPacketHeaderSizeBytes);
 }
 
 // Verify that asynchronous payload write completions don't cause a
 // use-after-free error if the writer object is deleted.
 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, _))
       .WillOnce(DoAll(SaveArg<2>(&header_cb), Return(net::ERR_IO_PENDING)));
   EXPECT_CALL(socket_,
               Write(BufferEquals(test_data_str_), test_data_str_.size(), _))
       .WillOnce(DoAll(SaveArg<2>(&payload_cb), Return(net::ERR_IO_PENDING)));
 
   writer->WritePacket(test_data_, writer_cb.callback());
 
   // Header write completes successfully.
   header_cb.Run(kPacketHeaderSizeBytes);
 
   // Payload write completion callback is invoked after the writer died.
   writer.reset();
   payload_cb.Run(test_data_str_.size());
 }
 
 }  // namespace
 }  // namespace blimp