Show user data dialog earlier on Windows.

chrome/browser/chrome_main_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 "chrome/test/ui/ui_test.h"
 
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/process_util.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_commands.h"
 #include "chrome/browser/ui/browser_finder.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_notification_types.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/web_contents.h"
 #include "net/base/net_util.h"
 
+#if defined(OS_WIN)
+#include "chrome/browser/user_data_dir_extractor_win.h"
+#endif
+
 // These tests don't apply to the Mac version; see GetCommandLineForRelaunch
 // for details.
 #if !defined(OS_MACOSX)
 
 class ChromeMainTest : public InProcessBrowserTest {
  public:
   ChromeMainTest() {}
 
   void Relaunch(const CommandLine& new_command_line) {
     base::LaunchProcess(new_command_line, base::LaunchOptions(), NULL);
@@ skipping 93 matching lines @@
         content::NotificationService::AllSources());
   Relaunch(new_command_line);
   tab_observer.Wait();
 
   // There should be one normal and one incognito window now.
   ASSERT_EQ(2u, chrome::GetTotalBrowserCount());
   ASSERT_EQ(1u, chrome::GetTabbedBrowserCount(profile, host_desktop_type));
 }
 
 #endif  // !OS_MACOSX
+
+// These tests only apply to the Windows version; see chrome_browser_main.cc:
+// function GetUserDataDir() for details.
+#if defined(OS_WIN)
+class ChromeMainUserDataDirTest : public InProcessBrowserTest {

[sky, 2013/03/25 15:43:14]

Move this to user_data_dir_extractor_win_browsertest, or
chrome_main_browsertest_win.

[hshi1, 2013/03/25 17:18:55]

Done.

+ public:
+  ChromeMainUserDataDirTest(): callback_invoked_(false) {}

[sky, 2013/03/25 15:43:14]

nit: space between () and :

+
+  bool is_callback_invoked() { return callback_invoked_; }

[sky, 2013/03/25 15:43:14]

field should either be named is_callback_invoked_ to match.

[hshi1, 2013/03/25 17:18:55]

Done.

+  void ShowUserDataDirDialog() {

[sky, 2013/03/25 15:43:14]

Move to private.

[hshi1, 2013/03/25 17:18:55]

Done.

+    // Set callback invoked flag and restore the correct user data dir.
+    callback_invoked_ = true;
+    PathService::Override(chrome::DIR_USER_DATA, user_data_dir_);
+  }
+
+ protected:
+  base::FilePath user_data_dir_;

[sky, 2013/03/25 15:43:14]

fields at end of section.

[hshi1, 2013/03/25 17:18:55]

Done.

+  base::Closure callback_;
+  bool callback_invoked_;
+
+  virtual void SetUp() OVERRIDE {
+    InProcessBrowserTest::SetUp();
+
+    // Save the actual user data dir.
+    PathService::Get(chrome::DIR_USER_DATA, &user_data_dir_);
+
+    // Override user data dir to a bogus value. Make sure to set |create| flag
+    // to false so that PathService::OverrideAndCreateIfNeeded does not create
+    // a folder at this path.
+    base::FilePath bogus_user_data_dir("/bogus/should/not/exist");
+    PathService::OverrideAndCreateIfNeeded(chrome::DIR_USER_DATA,
+                                           bogus_user_data_dir,
+                                           false);
+
+    // Install custom ShowUserDataDirDialog callback.
+    callback_ = base::Bind(&ChromeMainUserDataDirTest::ShowUserDataDirDialog,
+                           this);
+    chrome::InstallCustomShowUserDataDirDialogCallbackForTest(&callback_);
+  }
+};

[sky, 2013/03/25 15:43:14]

private:: DISALLOW...

[hshi1, 2013/03/25 17:18:55]

Done.

+
+IN_PROC_BROWSER_TEST_F(ChromeMainUserDataDirTest, BogusUserDataDir) {
+  // The ShowUserDataDirDialog should have been invoked and we did not crash.
+  ASSERT_TRUE(is_callback_invoked());
+}
+#endif  // OS_WIN