Cast channel: Add cast.channel.getLogs extension API.

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"
+#include "third_party/zlib/zlib.h"
 
 namespace extensions {
 namespace core_api {
 namespace cast_channel {
 
 using proto::AggregatedSocketEvent;
 using proto::EventType;
 using proto::Log;
 using proto::SocketEvent;
 
 class CastChannelLoggerTest : public testing::Test {
  public:
   // |logger_| will take ownership of |clock_|.
   CastChannelLoggerTest()
       : clock_(new base::SimpleTestTickClock),
         logger_(new Logger(scoped_ptr<base::TickClock>(clock_),
                            base::TimeTicks())) {}
   virtual ~CastChannelLoggerTest() {}
 
+  bool Uncompress(const char* input, int length, std::string* output) {
+    z_stream stream = {0};
+
+    stream.next_in = reinterpret_cast<uint8*>(const_cast<char*>(input));
+    stream.avail_in = length;
+    stream.next_out = reinterpret_cast<uint8*>(&(*output)[0]);
+    stream.avail_out = output->size();
+
+    bool success = false;
+    while (stream.avail_in > 0 && stream.avail_out > 0) {
+      // 16 is added to read in gzip format.
+      int result = inflateInit2(&stream, MAX_WBITS + 16);
+      DCHECK_EQ(Z_OK, result);
+
+      result = inflate(&stream, Z_FINISH);
+      success = (result == Z_STREAM_END);
+      if (!success) {
+        DVLOG(2) << "inflate() failed. Result: " << result;
+        break;
+      }
+
+      result = inflateEnd(&stream);
+      DCHECK(result == Z_OK);
+    }
+
+    if (stream.avail_in == 0) {
+      success = true;
+      output->resize(output->size() - stream.avail_out);
+    }
+    return success;
+  }
+
+  scoped_ptr<Log> GetLog() {
+    int length = 0;
+    scoped_ptr<char[]> output = logger_->GetLogs(&length);
+    if (!output.get())
+      return scoped_ptr<Log>();
+
+    // 100kb should be enough for test purposes.
+    std::string uncompressed(100000, 0);
+    if (!Uncompress(output.get(), length, &uncompressed))
+      return scoped_ptr<Log>();
+
+    scoped_ptr<Log> log(new Log);
+    if (!log->ParseFromString(uncompressed))
+      return scoped_ptr<Log>();
+
+    return log.Pass();
+  }
+
  protected:
   base::SimpleTestTickClock* clock_;
   scoped_refptr<Logger> logger_;
 };
 
 TEST_F(CastChannelLoggerTest, BasicLogging) {
   logger_->LogSocketEvent(1, EventType::CAST_SOCKET_CREATED);
   clock_->Advance(base::TimeDelta::FromMicroseconds(1));
   logger_->LogSocketEventWithDetails(
       1, EventType::TCP_SOCKET_CONNECT, "TCP socket");
   clock_->Advance(base::TimeDelta::FromMicroseconds(1));
   logger_->LogSocketEvent(2, EventType::CAST_SOCKET_CREATED);
   clock_->Advance(base::TimeDelta::FromMicroseconds(1));
   logger_->LogSocketEventWithRv(1, EventType::SSL_SOCKET_CONNECT, -1);
   clock_->Advance(base::TimeDelta::FromMicroseconds(1));
   logger_->LogSocketEventForMessage(
       2, EventType::MESSAGE_ENQUEUED, "foo_namespace", "details");
   clock_->Advance(base::TimeDelta::FromMicroseconds(1));
 
   AuthResult auth_result =
       AuthResult::Create("No response", AuthResult::ERROR_NO_RESPONSE);
 
   logger_->LogSocketChallengeReplyEvent(2, auth_result);
   clock_->Advance(base::TimeDelta::FromMicroseconds(1));
 
   auth_result = AuthResult::CreateWithNSSError(
       "Parsing failed", AuthResult::ERROR_NSS_CERT_PARSING_FAILED, 4);
   logger_->LogSocketChallengeReplyEvent(2, auth_result);
 
-  std::string output;
-  bool success = logger_->LogToString(&output);
-  ASSERT_TRUE(success);
+  scoped_ptr<Log> log = GetLog();
+  ASSERT_TRUE(log.get() != NULL);

[mark a. foltz, 2014/08/14 00:09:38]

Sorry, to be clear the ASSERTs would move into GetLog() so you wouldn't need to
add one here. GetLog would always return a non-NULL Log.

[imcheng, 2014/08/14 00:59:19]

Done.

 
-  Log log;
-  success = log.ParseFromString(output);
-  ASSERT_TRUE(success);
-
-  ASSERT_EQ(2, log.aggregated_socket_event_size());
+  ASSERT_EQ(2, log->aggregated_socket_event_size());
   {
     const AggregatedSocketEvent& aggregated_socket_event =
-        log.aggregated_socket_event(0);
+        log->aggregated_socket_event(0);
     EXPECT_EQ(1, aggregated_socket_event.id());
     EXPECT_EQ(3, aggregated_socket_event.socket_event_size());
     {
       const SocketEvent& event = aggregated_socket_event.socket_event(0);
       EXPECT_EQ(EventType::CAST_SOCKET_CREATED, event.type());
       EXPECT_EQ(0, event.timestamp_micros());
     }
     {
       const SocketEvent& event = aggregated_socket_event.socket_event(1);
       EXPECT_EQ(EventType::TCP_SOCKET_CONNECT, event.type());
       EXPECT_EQ(1, event.timestamp_micros());
       EXPECT_EQ("TCP socket", event.details());
     }
     {
       const SocketEvent& event = aggregated_socket_event.socket_event(2);
       EXPECT_EQ(EventType::SSL_SOCKET_CONNECT, event.type());
       EXPECT_EQ(3, event.timestamp_micros());
       EXPECT_EQ(-1, event.return_value());
     }
   }
   {
     const AggregatedSocketEvent& aggregated_socket_event =
-        log.aggregated_socket_event(1);
+        log->aggregated_socket_event(1);
     EXPECT_EQ(2, aggregated_socket_event.id());
     EXPECT_EQ(4, aggregated_socket_event.socket_event_size());
     {
       const SocketEvent& event = aggregated_socket_event.socket_event(0);
       EXPECT_EQ(EventType::CAST_SOCKET_CREATED, event.type());
       EXPECT_EQ(2, event.timestamp_micros());
     }
     {
       const SocketEvent& event = aggregated_socket_event.socket_event(1);
       EXPECT_EQ(EventType::MESSAGE_ENQUEUED, event.type());
@@ skipping 18 matching lines @@
       EXPECT_EQ(4, event.return_value());
     }
   }
 }
 
 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);
+  scoped_ptr<Log> log = GetLog();
+  ASSERT_TRUE(log.get() != NULL);
 
-  Log log;
-  success = log.ParseFromString(output);
-  ASSERT_TRUE(success);
-
-  ASSERT_EQ(kMaxSocketsToLog, log.aggregated_socket_event_size());
-  EXPECT_EQ(5, log.num_evicted_aggregated_socket_events());
-  EXPECT_EQ(5, log.num_evicted_socket_events());
+  ASSERT_EQ(kMaxSocketsToLog, log->aggregated_socket_event_size());
+  EXPECT_EQ(5, log->num_evicted_aggregated_socket_events());
+  EXPECT_EQ(5, log->num_evicted_socket_events());
 
   const AggregatedSocketEvent& aggregated_socket_event =
-      log.aggregated_socket_event(0);
+      log->aggregated_socket_event(0);
   EXPECT_EQ(5, aggregated_socket_event.id());
 }
 
 TEST_F(CastChannelLoggerTest, TooManyEvents) {
   for (int i = 0; i < kMaxEventsPerSocket + 5; i++) {
     logger_->LogSocketEvent(1, EventType::CAST_SOCKET_CREATED);
     clock_->Advance(base::TimeDelta::FromMicroseconds(1));
   }
 
-  std::string output;
-  bool success = logger_->LogToString(&output);
-  ASSERT_TRUE(success);
+  scoped_ptr<Log> log = GetLog();
+  ASSERT_TRUE(log.get() != NULL);
 
-  Log log;
-  success = log.ParseFromString(output);
-  ASSERT_TRUE(success);
-
-  ASSERT_EQ(1, log.aggregated_socket_event_size());
-  EXPECT_EQ(0, log.num_evicted_aggregated_socket_events());
-  EXPECT_EQ(5, log.num_evicted_socket_events());
+  ASSERT_EQ(1, log->aggregated_socket_event_size());
+  EXPECT_EQ(0, log->num_evicted_aggregated_socket_events());
+  EXPECT_EQ(5, log->num_evicted_socket_events());
 
   const AggregatedSocketEvent& aggregated_socket_event =
-      log.aggregated_socket_event(0);
+      log->aggregated_socket_event(0);
   ASSERT_EQ(kMaxEventsPerSocket, aggregated_socket_event.socket_event_size());
   EXPECT_EQ(5, aggregated_socket_event.socket_event(0).timestamp_micros());
 }
 
 TEST_F(CastChannelLoggerTest, Reset) {
   logger_->LogSocketEvent(1, EventType::CAST_SOCKET_CREATED);
 
-  std::string output;
-  bool success = logger_->LogToString(&output);
-  ASSERT_TRUE(success);
+  scoped_ptr<Log> log = GetLog();
+  ASSERT_TRUE(log.get() != NULL);
 
-  Log log;
-  success = log.ParseFromString(output);
-  ASSERT_TRUE(success);
-
-  EXPECT_EQ(1, log.aggregated_socket_event_size());
+  EXPECT_EQ(1, log->aggregated_socket_event_size());
 
   logger_->Reset();
 
-  success = logger_->LogToString(&output);
-  ASSERT_TRUE(success);
-  success = log.ParseFromString(output);
-  ASSERT_TRUE(success);
+  log = GetLog();
+  ASSERT_TRUE(log.get() != NULL);
 
-  EXPECT_EQ(0, log.aggregated_socket_event_size());
+  EXPECT_EQ(0, log->aggregated_socket_event_size());
 }
 
 }  // namespace cast_channel
 }  // namespace api
 }  // namespace extensions