chrome/test/base/tracing_browsertest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/test/base/tracing.h"
   6
   7 #include "base/debug/trace_event.h"
   8 #include "base/message_loop/message_loop.h"
   9 #include "base/run_loop.h"
  10 #include "chrome/browser/ui/browser.h"
  11 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  12 #include "chrome/test/base/in_process_browser_test.h"
  13 #include "chrome/test/base/ui_test_utils.h"
  14 #include "content/public/browser/render_view_host.h"
  15 #include "content/public/browser/web_contents.h"
  16 #include "content/public/test/browser_test_utils.h"
  17 #include "testing/gtest/include/gtest/gtest.h"
  18
  19 namespace {
  20
  21 using tracing::BeginTracingWithWatch;
  22 using tracing::WaitForWatchEvent;
  23 using tracing::EndTracing;
  24
  25 const char* g_category = "test_tracing";
  26 const char* g_event = "TheEvent";
  27
  28 class TracingBrowserTest : public InProcessBrowserTest {
  29  protected:
  30   // Execute some no-op javascript on the current tab - this triggers a trace
  31   // event in RenderViewImpl::OnScriptEvalRequest (from the renderer process).
  32   void ExecuteJavascriptOnCurrentTab() {
  33     content::RenderViewHost* rvh = browser()->tab_strip_model()->
  34         GetActiveWebContents()->GetRenderViewHost();
  35     ASSERT_TRUE(rvh);
  36     ASSERT_TRUE(content::ExecuteScript(rvh, ";"));
  37   }
  38 };
  39
  40 void AddEvents(int num) {
  41   for (int i = 0; i < num; ++i)
  42     TRACE_EVENT_INSTANT0(g_category, g_event, TRACE_EVENT_SCOPE_THREAD);
  43 }
  44
  45 IN_PROC_BROWSER_TEST_F(TracingBrowserTest, BeginTracingWithWatch) {
  46   base::TimeDelta no_timeout;
  47   base::TimeDelta short_timeout = base::TimeDelta::FromMilliseconds(5);
  48   std::string json_events;
  49
  50   // One event before wait.
  51   ASSERT_TRUE(BeginTracingWithWatch(g_category, g_category, g_event, 1));
  52   AddEvents(1);
  53   EXPECT_TRUE(WaitForWatchEvent(no_timeout));
  54   ASSERT_TRUE(EndTracing(&json_events));
  55
  56   // One event after wait.
  57   ASSERT_TRUE(BeginTracingWithWatch(g_category, g_category, g_event, 1));
  58   base::MessageLoop::current()->PostTask(FROM_HERE, base::Bind(&AddEvents, 1));
  59   EXPECT_TRUE(WaitForWatchEvent(no_timeout));
  60   ASSERT_TRUE(EndTracing(&json_events));
  61
  62   // Not enough events timeout.
  63   ASSERT_TRUE(BeginTracingWithWatch(g_category, g_category, g_event, 2));
  64   AddEvents(1);
  65   EXPECT_FALSE(WaitForWatchEvent(short_timeout));
  66   ASSERT_TRUE(EndTracing(&json_events));
  67
  68   // Multi event before wait.
  69   ASSERT_TRUE(BeginTracingWithWatch(g_category, g_category, g_event, 5));
  70   AddEvents(5);
  71   EXPECT_TRUE(WaitForWatchEvent(no_timeout));
  72   ASSERT_TRUE(EndTracing(&json_events));
  73
  74   // Multi event after wait.
  75   ASSERT_TRUE(BeginTracingWithWatch(g_category, g_category, g_event, 5));
  76   base::MessageLoop::current()->PostTask(FROM_HERE, base::Bind(&AddEvents, 5));
  77   EXPECT_TRUE(WaitForWatchEvent(no_timeout));
  78   ASSERT_TRUE(EndTracing(&json_events));
  79
  80   // Child process events from same process.
  81   ASSERT_TRUE(BeginTracingWithWatch(g_category, g_category,
  82                                     "OnScriptEvalRequest", 2));
  83   ASSERT_NO_FATAL_FAILURE(ExecuteJavascriptOnCurrentTab());
  84   ASSERT_NO_FATAL_FAILURE(ExecuteJavascriptOnCurrentTab());
  85   EXPECT_TRUE(WaitForWatchEvent(no_timeout));
  86   ASSERT_TRUE(EndTracing(&json_events));
  87
  88   // Child process events from different processes.
  89   GURL url1("chrome://tracing/");
  90   GURL url2("chrome://credits/");
  91   ASSERT_TRUE(BeginTracingWithWatch(g_category, g_category,
  92                                     "OnScriptEvalRequest", 2));
  93   // Open two tabs to different URLs to encourage two separate renderer
  94   // processes. Each will fire an event that will be counted towards the total.
  95   ui_test_utils::NavigateToURLWithDisposition(browser(), url1,
  96       NEW_FOREGROUND_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION);
  97   ASSERT_NO_FATAL_FAILURE(ExecuteJavascriptOnCurrentTab());
  98   ui_test_utils::NavigateToURLWithDisposition(browser(), url2,
  99       NEW_FOREGROUND_TAB, ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION);
 100   ASSERT_NO_FATAL_FAILURE(ExecuteJavascriptOnCurrentTab());
 101   EXPECT_TRUE(WaitForWatchEvent(no_timeout));
 102   ASSERT_TRUE(EndTracing(&json_events));
 103 }
 104
 105 }  // namespace