Modifying the BackgroundModeManager to handle multiple profiles.

chrome/browser/background_mode_manager_unittest.cc

 // Copyright (c) 2011 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/memory/scoped_ptr.h"
 #include "chrome/browser/background_mode_manager.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/testing_browser_process.h"
@@ skipping 11 matching lines @@
   BackgroundModeManagerTest() {}
   ~BackgroundModeManagerTest() {}
   void SetUp() {
     command_line_.reset(new CommandLine(CommandLine::NO_PROGRAM));
   }
   scoped_ptr<CommandLine> command_line_;
 };
 
 class TestBackgroundModeManager : public BackgroundModeManager {
  public:
-  TestBackgroundModeManager(Profile* profile, CommandLine* cl)
-      : BackgroundModeManager(profile, cl),
+  explicit TestBackgroundModeManager(CommandLine* command_line)
+      : BackgroundModeManager(command_line),
         enabled_(true) {}
   MOCK_METHOD1(EnableLaunchOnStartup, void(bool));
-  MOCK_METHOD0(CreateStatusTrayIcon, void());
-  MOCK_METHOD0(RemoveStatusTrayIcon, void());
+  MOCK_METHOD1(CreateStatusTrayIcon, void(Profile*));  // NOLINT
+  MOCK_METHOD1(RemoveStatusTrayIcon, void(Profile*));  // NOLINT
   virtual bool IsBackgroundModePrefEnabled() { return enabled_; }
   void SetEnabled(bool enabled) { enabled_ = enabled; }
  private:
   bool enabled_;
 };
 
 TEST_F(BackgroundModeManagerTest, BackgroundAppLoadUnload) {
   InSequence s;
   TestingProfile profile;
-  TestBackgroundModeManager manager(&profile, command_line_.get());
-  EXPECT_CALL(manager, CreateStatusTrayIcon());
-  EXPECT_CALL(manager, RemoveStatusTrayIcon());
+  TestBackgroundModeManager manager(command_line_.get());
+  manager.RegisterProfile(&profile);
+  EXPECT_CALL(manager, CreateStatusTrayIcon(&profile));
+  EXPECT_CALL(manager, RemoveStatusTrayIcon(&profile));
   EXPECT_FALSE(BrowserList::WillKeepAlive());
   // Call to AppLoaded() will cause the status tray to be created, then call to
   // unloaded will result in call to remove the icon.
   manager.OnBackgroundAppLoaded();
   EXPECT_TRUE(BrowserList::WillKeepAlive());
   manager.OnBackgroundAppUnloaded();
   EXPECT_FALSE(BrowserList::WillKeepAlive());
 }
 
 TEST_F(BackgroundModeManagerTest, BackgroundAppInstallUninstall) {
   InSequence s;
   TestingProfile profile;
-  TestBackgroundModeManager manager(&profile, command_line_.get());
+  TestBackgroundModeManager manager(command_line_.get());
+  manager.RegisterProfile(&profile);
   // Call to AppInstalled() will cause chrome to be set to launch on startup,
   // and call to AppUninstalled() set chrome to not launch on startup.
   EXPECT_CALL(manager, EnableLaunchOnStartup(true));
-  EXPECT_CALL(manager, CreateStatusTrayIcon());
-  EXPECT_CALL(manager, RemoveStatusTrayIcon());
+  EXPECT_CALL(manager, CreateStatusTrayIcon(&profile));
+  EXPECT_CALL(manager, RemoveStatusTrayIcon(&profile));
   EXPECT_CALL(manager, EnableLaunchOnStartup(false));
-  manager.OnBackgroundAppInstalled(NULL);
+  manager.OnBackgroundAppInstalled(NULL, &profile);
   manager.OnBackgroundAppLoaded();
   manager.OnBackgroundAppUnloaded();
   manager.OnBackgroundAppUninstalled();
 }
 
 // App installs while disabled should do nothing.
 TEST_F(BackgroundModeManagerTest, BackgroundAppInstallUninstallWhileDisabled) {
   InSequence s;
   TestingProfile profile;
-  TestBackgroundModeManager manager(&profile, command_line_.get());
+  TestBackgroundModeManager manager(command_line_.get());
+  manager.RegisterProfile(&profile);
   // Turn off background mode.
   manager.SetEnabled(false);
   manager.DisableBackgroundMode();
 
   // Status tray icons will not be created, launch on startup status will be set
   // to "do not launch on startup".
   EXPECT_CALL(manager, EnableLaunchOnStartup(false));
-  manager.OnBackgroundAppInstalled(NULL);
+  manager.OnBackgroundAppInstalled(NULL, &profile);
   manager.OnBackgroundAppLoaded();
   manager.OnBackgroundAppUnloaded();
   manager.OnBackgroundAppUninstalled();
 
   // Re-enable background mode.
   manager.SetEnabled(true);
   manager.EnableBackgroundMode();
 }
 
 
 // App installs while disabled should do nothing.
 TEST_F(BackgroundModeManagerTest, EnableAfterBackgroundAppInstall) {
   InSequence s;
   TestingProfile profile;
-  TestBackgroundModeManager manager(&profile, command_line_.get());
+  TestBackgroundModeManager manager(command_line_.get());
+  manager.RegisterProfile(&profile);
   EXPECT_CALL(manager, EnableLaunchOnStartup(true));
-  EXPECT_CALL(manager, CreateStatusTrayIcon());
-  EXPECT_CALL(manager, RemoveStatusTrayIcon());
+  EXPECT_CALL(manager, CreateStatusTrayIcon(&profile));
+  EXPECT_CALL(manager, RemoveStatusTrayIcon(&profile));
   EXPECT_CALL(manager, EnableLaunchOnStartup(false));
-  EXPECT_CALL(manager, CreateStatusTrayIcon());
+  EXPECT_CALL(manager, CreateStatusTrayIcon(&profile));
   EXPECT_CALL(manager, EnableLaunchOnStartup(true));
-  EXPECT_CALL(manager, RemoveStatusTrayIcon());
+  EXPECT_CALL(manager, RemoveStatusTrayIcon(&profile));
   EXPECT_CALL(manager, EnableLaunchOnStartup(false));
 
   // Install app, should show status tray icon.
-  manager.OnBackgroundAppInstalled(NULL);
+  manager.OnBackgroundAppInstalled(NULL, &profile);
   manager.OnBackgroundAppLoaded();
 
   // Turn off background mode - should hide status tray icon.
   manager.SetEnabled(false);
   manager.DisableBackgroundMode();
 
   // Turn back on background mode - should show status tray icon.
   manager.SetEnabled(true);
   manager.EnableBackgroundMode();
 
   // Uninstall app, should hide status tray icon again.
   manager.OnBackgroundAppUnloaded();
   manager.OnBackgroundAppUninstalled();
 }
+
+TEST_F(BackgroundModeManagerTest, MultiProfile) {
+  InSequence s;
+  TestingProfile profile1;
+  TestingProfile profile2;
+  TestBackgroundModeManager manager(command_line_.get());
+  manager.RegisterProfile(&profile1);
+  manager.RegisterProfile(&profile2);
+  EXPECT_CALL(manager, EnableLaunchOnStartup(true));
+  EXPECT_CALL(manager, CreateStatusTrayIcon(&profile1));
+  EXPECT_CALL(manager, CreateStatusTrayIcon(&profile2));
+  EXPECT_CALL(manager, RemoveStatusTrayIcon(&profile1));
+  EXPECT_CALL(manager, RemoveStatusTrayIcon(&profile2));
+  EXPECT_CALL(manager, EnableLaunchOnStartup(false));
+  EXPECT_CALL(manager, CreateStatusTrayIcon(&profile1));
+  EXPECT_CALL(manager, CreateStatusTrayIcon(&profile2));
+  EXPECT_CALL(manager, EnableLaunchOnStartup(true));
+  EXPECT_CALL(manager, RemoveStatusTrayIcon(&profile1));
+  EXPECT_CALL(manager, RemoveStatusTrayIcon(&profile2));
+  EXPECT_CALL(manager, EnableLaunchOnStartup(false));
+  EXPECT_FALSE(BrowserList::WillKeepAlive());
+
+  // Install app, should show status tray icon.
+  manager.OnBackgroundAppInstalled(NULL, &profile1);

[rpetterson, 2011/05/20 05:53:17]

What needs to be done to be able to call these private functions as done in the
tests above? For this test I get compile errors, but I don't for any of the
others.

[Miranda Callahan, 2011/05/20 15:16:53]

Have you tried adding this line to background_mode_manager.h, around line 64?  

FRIEND_TEST_ALL_PREFIXES(BackgroundModeManagerTest,
                         MultiProfile);

[Andrew T Wilson (Slow), 2011/05/20 16:52:46]

Yeah, Miranda is correct. I really hate this pattern, btw, but it seems to be
the most common one used in Chrome.

[rpetterson, 2011/05/23 03:23:19]

Thanks!

+  manager.OnBackgroundAppLoaded();
+
+  // Install app for other profile, hsould show other status tray icon.
+  manager.OnBackgroundAppInstalled(NULL, &profile2);
+  manager.OnBackgroundAppLoaded();
+
+  // Should hide both status tray icons.
+  manager.SetEnabled(false);
+  manager.DisableBackgroundMode();
+
+  // Turn back on background mode - should show both status tray icons.
+  manager.SetEnabled(true);
+  manager.EnableBackgroundMode();
+
+  manager.OnBackgroundAppUnloaded();
+  manager.OnBackgroundAppUninstalled();
+  // There is still one background app alive
+  EXPECT_TRUE(BrowserList::WillKeepAlive());
+  manager.OnBackgroundAppUnloaded();
+  manager.OnBackgroundAppUninstalled();
+  EXPECT_FALSE(BrowserList::WillKeepAlive());
+}