Upstream version 10.39.225.0

[platform/framework/web/crosswalk.git] / src / content / public / test / browser_test_base.cc

diff --git a/src/content/public/test/browser_test_base.cc b/src/content/public/test/browser_test_base.cc
index e31d937..c5083ee 100644 (file)
--- a/src/content/public/test/browser_test_base.cc
+++ b/src/content/public/test/browser_test_base.cc
@@ -14,6 +14,7 @@
 #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/browser/tracing/tracing_controller_impl.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/common/content_switches.h"
 #include "content/public/common/main_function_params.h"
@@ -121,6 +122,12 @@ class LocalHostResolverProc : public net::HostResolverProc {
   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&);
@@ -269,7 +276,38 @@ void BrowserTestBase::ProxyRunTestOnMainThreadLoop() {
     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(
+        TracingControllerImpl::CreateFileSink(
+            trace_file,
+            base::Bind(&TraceDisableRecordingComplete,
+                       run_loop.QuitClosure(),
+                       trace_file)));
+    run_loop.Run();
+  }
 }
 
 void BrowserTestBase::CreateTestServer(const base::FilePath& test_server_base) {