Out-of-process import on Windows.

chrome/browser/first_run/first_run_browsertest.cc

 // Copyright (c) 2012 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/command_line.h"
+#include "base/files/file_path.h"
 #include "base/prefs/pref_service.h"
+#include "base/string_util.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/extensions/component_loader.h"
 #include "chrome/browser/first_run/first_run.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
@@ skipping 40 matching lines @@
 
 IN_PROC_BROWSER_TEST_F(FirstRunBrowserTest, SetShouldShowWelcomePage) {
   EXPECT_FALSE(first_run::ShouldShowWelcomePage());
   first_run::SetShouldShowWelcomePage();
   EXPECT_TRUE(first_run::ShouldShowWelcomePage());
   EXPECT_FALSE(first_run::ShouldShowWelcomePage());
 }
 
 #if !defined(OS_CHROMEOS)
 namespace {
-class FirstRunIntegrationBrowserTest : public InProcessBrowserTest {
+
+// A generic test class to be subclassed by test classes testing specific
+// master_preferences. All subclasses must call SetMasterPreferencesForTest()
+// from their SetUp() method before deferring the remainder of Setup() to this
+// class.
+class FirstRunMasterPrefsBrowserTestBase : public InProcessBrowserTest {
  public:
-  FirstRunIntegrationBrowserTest() {}
- protected:
-  virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
-    InProcessBrowserTest::SetUpCommandLine(command_line);
-    command_line->AppendSwitch(switches::kForceFirstRun);
-    EXPECT_FALSE(first_run::DidPerformProfileImport(NULL));
-
-    extensions::ComponentLoader::EnableBackgroundExtensionsForTesting();
-
-    // The forked import process should run BrowserMain.
-    CommandLine import_arguments((CommandLine::NoProgram()));
-    import_arguments.AppendSwitch(content::kLaunchAsBrowser);
-    first_run::SetExtraArgumentsForImportProcess(import_arguments);
-  }
- private:
-  DISALLOW_COPY_AND_ASSIGN(FirstRunIntegrationBrowserTest);
-};
-
-class FirstRunMasterPrefsBrowserTest : public FirstRunIntegrationBrowserTest {
- public:
-  FirstRunMasterPrefsBrowserTest() {}
+  FirstRunMasterPrefsBrowserTestBase() {}
 
  protected:
   virtual void SetUp() OVERRIDE {
+    // All users of this test class need to call SetMasterPreferencesForTest()
+    // before this class' SetUp() is invoked.
+    ASSERT_TRUE(text_.get());
+
     ASSERT_TRUE(file_util::CreateTemporaryFile(&prefs_file_));
-    // TODO(tapted): Make this reusable.
-    const char text[] =
-        "{\n"
-        "  \"distribution\": {\n"
-        "    \"import_bookmarks\": false,\n"
-        "    \"import_history\": false,\n"
-        "    \"import_home_page\": false,\n"
-        "    \"import_search_engine\": false\n"
-        "  }\n"
-        "}\n";
-    EXPECT_TRUE(file_util::WriteFile(prefs_file_, text, strlen(text)));
+    EXPECT_TRUE(file_util::WriteFile(prefs_file_, text_->c_str(),
+                                     text_->size()));
     first_run::SetMasterPrefsPathForTesting(prefs_file_);
 
     // This invokes BrowserMain, and does the import, so must be done last.
-    FirstRunIntegrationBrowserTest::SetUp();
+    InProcessBrowserTest::SetUp();
   }
 
   virtual void TearDown() OVERRIDE {
     EXPECT_TRUE(file_util::Delete(prefs_file_, false));
-    FirstRunIntegrationBrowserTest::TearDown();
+    InProcessBrowserTest::TearDown();
+  }
+
+  virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
+    InProcessBrowserTest::SetUpCommandLine(command_line);
+    command_line->AppendSwitch(switches::kForceFirstRun);
+    EXPECT_EQ(first_run::AUTO_IMPORT_NONE, first_run::auto_import_state());
+
+    extensions::ComponentLoader::EnableBackgroundExtensionsForTesting();
+  }
+
+  void SetMasterPreferencesForTest(const char text[]) {
+    text_.reset(new std::string(text));
   }
 
  private:
   base::FilePath prefs_file_;
+  scoped_ptr<std::string> text_;
 
-  DISALLOW_COPY_AND_ASSIGN(FirstRunMasterPrefsBrowserTest);
+  DISALLOW_COPY_AND_ASSIGN(FirstRunMasterPrefsBrowserTestBase);
 };
+
+template<const char Text[]>
+class FirstRunMasterPrefsBrowserTestT
+    : public FirstRunMasterPrefsBrowserTestBase {
+ public:
+  FirstRunMasterPrefsBrowserTestT() {}
+
+ protected:
+  virtual void SetUp() OVERRIDE {
+    SetMasterPreferencesForTest(Text);
+    FirstRunMasterPrefsBrowserTestBase::SetUp();
+  }
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(FirstRunMasterPrefsBrowserTestT);
+};
+
+}  // namespace
+
+// TODO(tapted): Investigate why this fails on Linux bots but does not
+// reproduce locally. See http://crbug.com/178062 .
+// TODO(tapted): Investigate why this fails on mac_asan flakily
+// http://crbug.com/181499 .
+#if defined(OS_LINUX) || (defined(OS_MACOSX) && defined(ADDRESS_SANITIZER))
+#define MAYBE_ImportDefault DISABLED_ImportDefault
+#else
+#define MAYBE_ImportDefault ImportDefault
+#endif
+
+extern const char kImportDefault[] =
+    "{\n"
+    "}\n";
+typedef FirstRunMasterPrefsBrowserTestT<kImportDefault>
+    FirstRunMasterPrefsImportDefault;
+IN_PROC_BROWSER_TEST_F(FirstRunMasterPrefsImportDefault, MAYBE_ImportDefault) {
+  int auto_import_state = first_run::auto_import_state();
+  // Aura builds skip over the import process.
+#if defined(USE_AURA)
+  EXPECT_EQ(first_run::AUTO_IMPORT_CALLED, auto_import_state);
+#else
+  EXPECT_EQ(first_run::AUTO_IMPORT_CALLED |
+                first_run::AUTO_IMPORT_PROFILE_IMPORTED,
+            auto_import_state);
+#endif
 }
 
 // TODO(tapted): Investigate why this fails on Linux bots but does not
 // reproduce locally. See http://crbug.com/178062 .
 // TODO(tapted): Investigate why this fails on mac_asan flakily
 // http://crbug.com/181499 .
 #if defined(OS_LINUX) || (defined(OS_MACOSX) && defined(ADDRESS_SANITIZER))
-#define MAYBE_WaitForImport DISABLED_WaitForImport
+#define MAYBE_ImportBookmarksFile DISABLED_ImportBookmarksFile
 #else
-#define MAYBE_WaitForImport WaitForImport
+#define MAYBE_ImportBookmarksFile ImportBookmarksFile
 #endif
 
-IN_PROC_BROWSER_TEST_F(FirstRunIntegrationBrowserTest, MAYBE_WaitForImport) {
-  bool success = false;
-  EXPECT_TRUE(first_run::DidPerformProfileImport(&success));
+// The bookmarks file doesn't actually need to exist for this integration test
+// to trigger the interaction being tested.
+extern const char kImportBookmarksFile[] =
+    "{\n"
+    "  \"distribution\": {\n"
+    "     \"import_bookmarks_from_file\": \"/foo/doesntexists.wtv\"\n"
+    "  }\n"
+    "}\n";
+typedef FirstRunMasterPrefsBrowserTestT<kImportBookmarksFile>
+    FirstRunMasterPrefsImportBookmarksFile;
+IN_PROC_BROWSER_TEST_F(FirstRunMasterPrefsImportBookmarksFile,
+                       MAYBE_ImportBookmarksFile) {
+  int auto_import_state = first_run::auto_import_state();
   // Aura builds skip over the import process.
 #if defined(USE_AURA)
-  EXPECT_FALSE(success);
+  EXPECT_EQ(first_run::AUTO_IMPORT_CALLED, auto_import_state);
 #else
-  EXPECT_TRUE(success);
+  EXPECT_EQ(first_run::AUTO_IMPORT_CALLED |
+                first_run::AUTO_IMPORT_PROFILE_IMPORTED |
+                first_run::AUTO_IMPORT_BOOKMARKS_FILE_IMPORTED,
+            auto_import_state);
 #endif
 }
 
 // Test an import with all import options disabled. This is a regression test
 // for http://crbug.com/169984 where this would cause the import process to
 // stay running, and the NTP to be loaded with no apps.
-IN_PROC_BROWSER_TEST_F(FirstRunMasterPrefsBrowserTest,
+extern const char kImportNothing[] =
+    "{\n"
+    "  \"distribution\": {\n"
+    "    \"import_bookmarks\": false,\n"
+    "    \"import_history\": false,\n"
+    "    \"import_home_page\": false,\n"
+    "    \"import_search_engine\": false\n"
+    "  }\n"
+    "}\n";
+typedef FirstRunMasterPrefsBrowserTestT<kImportNothing>
+    FirstRunMasterPrefsImportNothing;
+IN_PROC_BROWSER_TEST_F(FirstRunMasterPrefsImportNothing,
                        ImportNothingAndShowNewTabPage) {
-  EXPECT_TRUE(first_run::DidPerformProfileImport(NULL));
+  EXPECT_EQ(first_run::AUTO_IMPORT_CALLED, first_run::auto_import_state());
   ui_test_utils::NavigateToURLWithDisposition(
       browser(), GURL(chrome::kChromeUINewTabURL), CURRENT_TAB,
       ui_test_utils::BROWSER_TEST_WAIT_FOR_NAVIGATION);
   content::WebContents* tab = browser()->tab_strip_model()->GetWebContentsAt(0);
   EXPECT_EQ(1, tab->GetMaxPageID());
 }
 
 #endif  // !defined(OS_CHROMEOS)