Cast channel logging: log additional statistics.

extensions/browser/api/cast_channel/logger_unittest.cc

 // Copyright 2014 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 "base/test/simple_test_tick_clock.h"
 #include "extensions/browser/api/cast_channel/cast_auth_util.h"
 #include "extensions/browser/api/cast_channel/logger.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace extensions {
@@ skipping 104 matching lines @@
       const SocketEvent& event = aggregated_socket_event.socket_event(3);
       EXPECT_EQ(EventType::AUTH_CHALLENGE_REPLY, event.type());
       EXPECT_EQ(6, event.timestamp_micros());
       EXPECT_EQ(proto::CHALLENGE_REPLY_ERROR_NSS_CERT_PARSING_FAILED,
                 event.challenge_reply_error_type());
       EXPECT_EQ(4, event.return_value());
     }
   }
 }
 
+TEST_F(CastChannelLoggerTest, LogSocketReadWrite) {
+  logger_->LogSocketEventWithRv(1, EventType::SOCKET_READ, 50);
+  clock_->Advance(base::TimeDelta::FromMicroseconds(1));
+  logger_->LogSocketEventWithRv(1, EventType::SOCKET_READ, 30);
+  clock_->Advance(base::TimeDelta::FromMicroseconds(1));
+  logger_->LogSocketEventWithRv(1, EventType::SOCKET_READ, -1);
+  clock_->Advance(base::TimeDelta::FromMicroseconds(1));
+  logger_->LogSocketEventWithRv(1, EventType::SOCKET_WRITE, 20);
+  clock_->Advance(base::TimeDelta::FromMicroseconds(1));
+
+  logger_->LogSocketEventWithRv(2, EventType::SOCKET_READ, 100);
+  clock_->Advance(base::TimeDelta::FromMicroseconds(1));
+  logger_->LogSocketEventWithRv(2, EventType::SOCKET_WRITE, 100);
+  clock_->Advance(base::TimeDelta::FromMicroseconds(1));
+  logger_->LogSocketEventWithRv(2, EventType::SOCKET_WRITE, -5);
+  clock_->Advance(base::TimeDelta::FromMicroseconds(1));
+
+  std::string output;
+  bool success = logger_->LogToString(&output);
+  ASSERT_TRUE(success);
+
+  Log log;
+  success = log.ParseFromString(output);
+  ASSERT_TRUE(success);
+
+  EXPECT_EQ(180, log.total_bytes_read());
+  EXPECT_EQ(120, log.total_bytes_written());
+  ASSERT_EQ(2, log.aggregated_socket_event_size());
+  {
+    const AggregatedSocketEvent& aggregated_socket_event =
+        log.aggregated_socket_event(0);
+    EXPECT_EQ(1, aggregated_socket_event.id());
+    EXPECT_EQ(4, aggregated_socket_event.socket_event_size());
+    EXPECT_EQ(80, aggregated_socket_event.bytes_read());
+    EXPECT_EQ(20, aggregated_socket_event.bytes_written());
+  }
+  {
+    const AggregatedSocketEvent& aggregated_socket_event =
+        log.aggregated_socket_event(1);
+    EXPECT_EQ(2, aggregated_socket_event.id());
+    EXPECT_EQ(3, aggregated_socket_event.socket_event_size());
+    EXPECT_EQ(100, aggregated_socket_event.bytes_read());
+    EXPECT_EQ(100, aggregated_socket_event.bytes_written());
+  }
+}
+
 TEST_F(CastChannelLoggerTest, TooManySockets) {
   for (int i = 0; i < kMaxSocketsToLog + 5; i++) {
     logger_->LogSocketEvent(i, EventType::CAST_SOCKET_CREATED);
   }
 
   std::string output;
   bool success = logger_->LogToString(&output);
   ASSERT_TRUE(success);
 
   Log log;
@@ skipping 52 matching lines @@
   ASSERT_TRUE(success);
   success = log.ParseFromString(output);
   ASSERT_TRUE(success);
 
   EXPECT_EQ(0, log.aggregated_socket_event_size());
 }
 
 }  // namespace cast_channel
 }  // namespace api
 }  // namespace extensions