Test end-to-end startup performance of loading several tabs.

chrome/test/startup/startup_test.cc

 // Copyright (c) 2010 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/env_var.h"
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/platform_thread.h"
 #include "base/string_util.h"
 #include "base/sys_info.h"
 #include "base/test/test_file_util.h"
 #include "base/time.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/env_vars.h"
 #include "chrome/test/ui/ui_test.h"
 #include "chrome/test/ui_test_utils.h"
+#include "net/base/net_util.h"
 
 using base::TimeDelta;
 using base::TimeTicks;
 
 namespace {
 
 class StartupTest : public UITest {
  public:
   StartupTest() {
     show_window_ = true;
   }
   void SetUp() {}
   void TearDown() {}
 
   // Load a file on startup rather than about:blank.  This tests a longer
   // startup path, including resource loading and the loading of gears.dll.
   void SetUpWithFileURL() {
     const FilePath file_url = ui_test_utils::GetTestFilePath(
         FilePath(FilePath::kCurrentDirectory),
         FilePath(FILE_PATH_LITERAL("simple.html")));
     ASSERT_TRUE(file_util::PathExists(file_url));
     launch_arguments_.AppendLooseValue(file_url.ToWStringHack());
   }
 
+  // Load a complex html file on startup represented by |which_tab|.
+  void SetUpWithComplexFileURL(unsigned int which_tab) {
+    const char* const kTestPageCyclerDomains[] = {
+        "www.google.com", "www.nytimes.com",
+        "www.yahoo.com", "espn.go.com", "www.amazon.com"
+    };
+    unsigned int which_el = which_tab % arraysize(kTestPageCyclerDomains);
+    const char* this_domain = kTestPageCyclerDomains[which_el];
+
+    FilePath page_cycler_path;
+    PathService::Get(base::DIR_SOURCE_ROOT, &page_cycler_path);
+    page_cycler_path = page_cycler_path.AppendASCII("data")
+        .AppendASCII("page_cycler").AppendASCII("moz")
+        .AppendASCII(this_domain).AppendASCII("index.html");
+    GURL file_url = net::FilePathToFileURL(page_cycler_path).Resolve("?skip");
+    launch_arguments_.AppendLooseValue(ASCIIToWide(file_url.spec()));
+  }
+
   // Use the given profile in the test data extensions/profiles dir.  This tests
   // startup with extensions installed.
   void SetUpWithExtensionsProfile(const char* profile) {
     FilePath data_dir;
     PathService::Get(chrome::DIR_TEST_DATA, &data_dir);
     data_dir = data_dir.AppendASCII("extensions").AppendASCII("profiles").
         AppendASCII(profile);
     set_template_user_data(data_dir);
 
     // For now, these tests still depend on toolstrips.
     launch_arguments_.AppendSwitch(switches::kEnableExtensionToolstrips);
   }
 
+  void RunPerfTestWithManyTabs(const char *test_name,
+      int tab_count, bool restore_session);
+
   void RunStartupTest(const char* graph, const char* trace,
       bool test_cold, bool important, UITest::ProfileType profile_type) {
     profile_type_ = profile_type;
 
     // Sets the profile data for the run.  For now, this is only used for
     // the non-default themes test.
     if (profile_type != UITest::DEFAULT_THEME) {
       set_template_user_data(UITest::ComputeTypicalUserDataSource(
           profile_type));
     }
@@ skipping 73 matching lines @@
                  true /* important */, UITest::DEFAULT_THEME);
 }
 
 // TODO(mpcomplete): Should we have reference timings for all these?
 
 TEST_F(StartupTest, PerfCold) {
   RunStartupTest("cold", "t", true /* cold */, false /* not important */,
                  UITest::DEFAULT_THEME);
 }
 
+void StartupTest::RunPerfTestWithManyTabs(const char *test_name,
+    int tab_count, bool restore_session) {
+  // Initialize session with |tab_count| tabs.
+  for (int i = 0; i < tab_count; ++i)
+    SetUpWithComplexFileURL(i);
+
+  if (restore_session) {
+    // Start the browser with these urls so we can save the session and exit.
+    UITest::SetUp();
+    // Set flags to ensure profile is saved and can be restored.
+#if defined(OS_MACOSX)
+    shutdown_type_ = UITestBase::USER_QUIT;
+#endif
+    clear_profile_ = false;
+    // Quit and set flags to restore session.
+    UITest::TearDown();
+    // Clear all arguments for session restore, or the number of open tabs
+    // will grow with each restore.
+    launch_arguments_ = CommandLine(CommandLine::ARGUMENTS_ONLY);
+    // The session will be restored once per cycle for numCycles test cycles,
+    // and each time, UITest::SetUp will wait for |tab_count| tabs to
+    // finish loading.
+    launch_arguments_.AppendSwitchWithValue(switches::kRestoreLastSession,
+                                            IntToWString(tab_count));
+  }
+  RunStartupTest("warm", test_name,
+                 false, false,
+                 UITest::DEFAULT_THEME);
+}
+
+TEST_F(StartupTest, PerfFewTabs) {
+  RunPerfTestWithManyTabs("few_tabs", 5, false);
+}
+
+TEST_F(StartupTest, PerfSeveralTabs) {
+  RunPerfTestWithManyTabs("several_tabs", 20, false);
+}
+
+TEST_F(StartupTest, PerfRestoreFewTabs) {
+  RunPerfTestWithManyTabs("restore_few_tabs", 5, true);
+}
+
+TEST_F(StartupTest, PerfRestoreSeveralTabs) {
+  RunPerfTestWithManyTabs("restore_several_tabs", 20, true);
+}
+
 TEST_F(StartupTest, PerfExtensionEmpty) {
   SetUpWithFileURL();
   SetUpWithExtensionsProfile("empty");
   RunStartupTest("warm", "extension_empty",
                  false /* cold */, false /* not important */,
                  UITest::DEFAULT_THEME);
 }
 
 TEST_F(StartupTest, PerfExtensionToolstrips1) {
   SetUpWithFileURL();
@@ skipping 58 matching lines @@
                  false /* not important */, UITest::CUSTOM_FRAME);
 }
 
 TEST_F(StartupTest, PerfColdNativeFrameGtkTheme) {
   RunStartupTest("warm", "custom-frame-gtk-theme", false /* warm */,
                  false /* not important */, UITest::CUSTOM_FRAME_NATIVE_THEME);
 }
 #endif
 
 }  // namespace