Add support for tracing in browser tests.

content/public/test/browser_test_base.cc

 // Copyright (c) 2011 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 "content/public/test/browser_test_base.h"
 
 #include "base/bind.h"
 #include "base/command_line.h"
 #include "base/debug/stack_trace.h"
 #include "base/i18n/icu_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/sys_info.h"
 #include "base/test/test_timeouts.h"
 #include "content/public/app/content_main.h"
 #include "content/browser/renderer_host/render_process_host_impl.h"
 #include "content/public/browser/browser_thread.h"
+#include "content/public/browser/tracing_controller.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/main_function_params.h"
 #include "content/public/test/test_launcher.h"
 #include "content/public/test/test_utils.h"
 #include "net/base/net_errors.h"
 #include "net/dns/mock_host_resolver.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 #include "ui/compositor/compositor_switches.h"
 #include "ui/gl/gl_implementation.h"
 #include "ui/gl/gl_switches.h"
@@ skipping 86 matching lines @@
     }
 
     return ResolveUsingPrevious(host, address_family, host_resolver_flags,
                                 addrlist, os_error);
   }
 
  private:
   virtual ~LocalHostResolverProc() {}
 };
 
+void TraceDisableRecordingComplete(const base::Closure& quit,
+                                   const base::FilePath& file_path) {
+  LOG(ERROR) << "Tracing written to: " << file_path.value();
+  quit.Run();
+}
+
 }  // namespace
 
 extern int BrowserMain(const MainFunctionParams&);
 
 BrowserTestBase::BrowserTestBase()
     : expected_exit_code_(0),
       enable_pixel_output_(false),
       use_software_compositing_(false) {
 #if defined(OS_MACOSX)
   base::mac::SetOverrideAmIBundled(true);
@@ skipping 128 matching lines @@
 void BrowserTestBase::TearDown() {
 }
 
 void BrowserTestBase::ProxyRunTestOnMainThreadLoop() {
 #if defined(OS_POSIX)
   if (handle_sigterm_) {
     g_browser_process_pid = base::GetCurrentProcId();
     signal(SIGTERM, DumpStackTraceSignalHandler);
   }
 #endif  // defined(OS_POSIX)
+
+  if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kEnableTracing)) {
+    base::debug::CategoryFilter category_filter(
+        CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
+            switches::kEnableTracing));
+    TracingController::GetInstance()->EnableRecording(
+        category_filter,
+        base::debug::TraceOptions(base::debug::RECORD_CONTINUOUSLY),
+        TracingController::EnableRecordingDoneCallback());
+  }
+
   RunTestOnMainThreadLoop();
+
+  if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kEnableTracing)) {
+    base::FilePath trace_file =
+        CommandLine::ForCurrentProcess()->GetSwitchValuePath(
+            switches::kEnableTracingOutput);
+    // If there was no file specified, put a hardcoded one in the current
+    // working directory.
+    if (trace_file.empty())
+      trace_file = base::FilePath().AppendASCII("trace.json");
+
+    // Wait for tracing to collect results from the renderers.
+    base::RunLoop run_loop;
+    TracingController::GetInstance()->DisableRecording(
+        trace_file,
+        base::Bind(&TraceDisableRecordingComplete, run_loop.QuitClosure()));
+    run_loop.Run();
+  }
 }
 
 void BrowserTestBase::CreateTestServer(const base::FilePath& test_server_base) {
   CHECK(!test_server_.get());
   test_server_.reset(new net::SpawnedTestServer(
       net::SpawnedTestServer::TYPE_HTTP,
       net::SpawnedTestServer::kLocalhost,
       test_server_base));
 }
 
@@ skipping 19 matching lines @@
   use_software_compositing_ = true;
 }
 
 bool BrowserTestBase::UsingOSMesa() const {
   CommandLine* cmd = CommandLine::ForCurrentProcess();
   return cmd->GetSwitchValueASCII(switches::kUseGL) ==
          gfx::kGLImplementationOSMesaName;
 }
 
 }  // namespace content