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/browser/ui/user_data_dir_dialog.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"
@@ skipping 104 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 {
+ public:
+  ChromeMainUserDataDirTest(): callback_invoked_(false) {}
+
+  bool is_callback_invoked() { return callback_invoked_; }
+  void ShowUserDataDirDialog() {
+    // 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_;
+  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.
+    chrome::InstallCustomShowUserDataDirDialogCallbackForTest(
+        base::Bind(&ChromeMainUserDataDirTest::ShowUserDataDirDialog, this));
+  }
+};
+
+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