[Mac] AppControllerMac updates the window when a Profile is deleted.

chrome/browser/app_controller_mac_browsertest.mm

 // 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.
 
 #import <Carbon/Carbon.h>
 #import <Cocoa/Cocoa.h>
 #import <Foundation/Foundation.h>
 #import <Foundation/NSAppleEventDescriptor.h>
 #import <objc/message.h>
 #import <objc/runtime.h>
 
 #include "base/command_line.h"
 #include "base/mac/foundation_util.h"
 #include "base/mac/scoped_nsobject.h"
 #include "base/prefs/pref_service.h"
 #include "base/run_loop.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "components/bookmarks/browser/bookmark_model.h"
 #import "chrome/browser/app_controller_mac.h"
 #include "chrome/browser/apps/app_browsertest_util.h"
 #include "chrome/browser/bookmarks/bookmark_model_factory.h"
 #include "chrome/browser/browser_process.h"
+#include "chrome/browser/history/history_service_factory.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "chrome/browser/ui/cocoa/bookmarks/bookmark_menu_bridge.h"
+#include "chrome/browser/ui/cocoa/history_menu_bridge.h"
 #include "chrome/browser/ui/host_desktop.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/user_manager.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "components/bookmarks/test/bookmark_test_helpers.h"
+#include "components/history/core/browser/history_db_task.h"
+#include "components/history/core/browser/history_service.h"
 #include "components/signin/core/common/profile_management_switches.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/browser_test_utils.h"
 #include "content/public/test/test_navigation_observer.h"
 #include "extensions/browser/app_window/app_window_registry.h"
 #include "extensions/common/extension.h"
 #include "extensions/test/extension_test_message_listener.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 
 namespace {
@@ skipping 22 matching lines @@
   Method get_url =
       class_getInstanceMethod([controller class], @selector(getUrl:withReply:));
 
   ASSERT_TRUE(get_url);
 
   NSAppleEventDescriptor* shortcut_event = AppleEventToOpenUrl(url);
 
   method_invoke(controller, get_url, shortcut_event, NULL);
 }
 
+void RunClosureWhenProfileInitialized(const base::Closure& closure,
+                                      Profile* profile,
+                                      Profile::CreateStatus status) {
+  if (status == Profile::CREATE_STATUS_INITIALIZED)
+    closure.Run();
+}
+
+class WaitForHistoryTask : public history::HistoryDBTask {
+ public:
+  WaitForHistoryTask() {}
+
+  bool RunOnDBThread(history::HistoryBackend* backend,
+                     history::HistoryDatabase* db) override {
+    return true;
+  }
+
+  void DoneRunOnMainThread() override { base::MessageLoop::current()->Quit(); }
+
+ private:
+  ~WaitForHistoryTask() override {}
+
+  DISALLOW_COPY_AND_ASSIGN(WaitForHistoryTask);
+};
+
+void WaitForHistoryBackendToRun(Profile* profile) {

[Robert Sesek, 2015/04/21 18:02:42]

Can you use

base::RunLoop run_loop;
HistoryService::FlushForTest(run_loop.QuitClosure());
run_loop.Run();

instead?

(I'm not sure what FlushForTest is for).

[Mike Lerman, 2015/04/21 19:33:16]

We actually don't need to load the Backend at all in this test - we are just
testing the Service pointers change.

+  base::CancelableTaskTracker task_tracker;
+  scoped_ptr<history::HistoryDBTask> task(new WaitForHistoryTask());
+  history::HistoryService* history = HistoryServiceFactory::GetForProfile(
+      profile, ServiceAccessType::EXPLICIT_ACCESS);
+  history->ScheduleDBTask(task.Pass(), &task_tracker);
+  content::RunMessageLoop();
+}
+
 }  // namespace
 
 @interface TestOpenShortcutOnStartup : NSObject
 - (void)applicationWillFinishLaunching:(NSNotification*)notification;
 @end
 
 @implementation TestOpenShortcutOnStartup
 
 - (void)applicationWillFinishLaunching:(NSNotification*)notification {
   if (!g_open_shortcut_url.is_valid())
@@ skipping 328 matching lines @@
                 ->GetLastCommittedURL());
 }
 
 class AppControllerMainMenuBrowserTest : public InProcessBrowserTest {
  protected:
   AppControllerMainMenuBrowserTest() {
   }
 };
 
 IN_PROC_BROWSER_TEST_F(AppControllerMainMenuBrowserTest,
+    HistoryMenuResetAfterProfileDeletion) {
+  ProfileManager* profile_manager = g_browser_process->profile_manager();
+  AppController* ac = [NSApp delegate];
+
+  // Use the existing profile as profile 1.
+  Profile* profile1 = browser()->profile();
+
+  // // Create profile 2.

[Robert Sesek, 2015/04/21 18:02:42]

nit: extra //

[Mike Lerman, 2015/04/21 19:33:16]

Done.

+  base::FilePath profile2_path =
+      profile_manager->GenerateNextProfileDirectoryPath();
+  base::RunLoop run_loop;
+  profile_manager->CreateProfileAsync(

[Robert Sesek, 2015/04/21 18:02:42]

Can you create the profile synchronously, like on line 545?

[Mike Lerman, 2015/04/21 19:33:16]

The command on line 545 doesn't do all the same things CreateProfileAsync does.
CreateProfile creates a Profile*. CreateProfileAsync calls CreateProfile, but
also adds the Profile* to the ProfileManager and the ProfileInfoCache - the
latter is key since the AppController learns the profile is deleted by Observing
the ProfileInfoCache.

[Robert Sesek, 2015/04/21 20:12:30]

Got it. Thanks for the explanation.

+      profile2_path,
+      base::Bind(&RunClosureWhenProfileInitialized,
+                 run_loop.QuitClosure()),
+                 base::string16(),
+                 base::string16(),
+                 std::string());
+  run_loop.Run();
+  Profile* profile2 = profile_manager->GetProfileByPath(profile2_path);
+  DCHECK(profile2);
+
+  // Switch the controller to profile1.
+  [ac windowChangedToProfile:profile1];
+  WaitForHistoryBackendToRun(profile1);
+  base::RunLoop().RunUntilIdle();
+
+  // Verify the controller's History Menu corresponds to profile1.
+  DCHECK_EQ([ac historyMenuBridge]->service(),
+      HistoryServiceFactory::GetForProfile(profile1,
+                                           ServiceAccessType::EXPLICIT_ACCESS));
+
+  // Switch the controller to profile2.
+  [ac windowChangedToProfile:profile2];
+  WaitForHistoryBackendToRun(profile2);
+  base::RunLoop().RunUntilIdle();
+
+  // Verify the controller's History Menu has changed.
+  DCHECK_EQ([ac historyMenuBridge]->service(),
+      HistoryServiceFactory::GetForProfile(profile2,
+                                           ServiceAccessType::EXPLICIT_ACCESS));
+  DCHECK_NE(
+      HistoryServiceFactory::GetForProfile(profile1,
+                                           ServiceAccessType::EXPLICIT_ACCESS),
+      HistoryServiceFactory::GetForProfile(profile2,
+                                           ServiceAccessType::EXPLICIT_ACCESS));
+
+  // Delete profile2.
+  profile_manager->ScheduleProfileForDeletion(
+      profile2->GetPath(), ProfileManager::CreateCallback());
+  base::RunLoop().RunUntilIdle();
+
+  // Verify the controller's history is back to profile1.
+  DCHECK_EQ([ac historyMenuBridge]->service(),
+      HistoryServiceFactory::GetForProfile(profile1,
+                                           ServiceAccessType::EXPLICIT_ACCESS));
+}
+
+IN_PROC_BROWSER_TEST_F(AppControllerMainMenuBrowserTest,
     BookmarksMenuIsRestoredAfterProfileSwitch) {
   ProfileManager* profile_manager = g_browser_process->profile_manager();
   base::scoped_nsobject<AppController> ac([[AppController alloc] init]);
   [ac awakeFromNib];
 
   // Constants for bookmarks that we will create later.
   const base::string16 title1(base::ASCIIToUTF16("Dinosaur Comics"));
   const GURL url1("http://qwantz.com//");
 
   const base::string16 title2(base::ASCIIToUTF16("XKCD"));
@@ skipping 175 matching lines @@
   ui_test_utils::NavigateToURL(browser3, test_url1);
   EXPECT_EQ(g_handoff_url, GURL());
 
   // Activate the original browser window.
   Browser* browser1 = active_browser_list->get(0);
   browser1->window()->Show();
   EXPECT_EQ(g_handoff_url, test_url2);
 }
 
 }  // namespace