Reland: Convert ExtensionProcessManager to BrowserContext, part 1

chrome/browser/extensions/extension_process_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 "chrome/browser/extensions/extension_process_manager.h"
+
+#include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/extensions/extension_error_reporter.h"
 #include "chrome/test/base/testing_profile.h"
+#include "content/public/browser/notification_service.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/site_instance.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/platform_test.h"
 
 using content::SiteInstance;
 
-namespace {
-
 // make the test a PlatformTest to setup autorelease pools properly on mac
 class ExtensionProcessManagerTest : public testing::Test {
  public:
   static void SetUpTestCase() {
     ExtensionErrorReporter::Init(false);  // no noisy errors
   }
 
   virtual void SetUp() {
     ExtensionErrorReporter::GetInstance()->ClearErrors();
   }
+
+  // Returns true if the notification |type| is registered for |manager| with
+  // source |profile|. Pass NULL for |profile| for all sources.
+  static bool IsRegistered(ExtensionProcessManager* manager,
+                    int type,
+                    TestingProfile* profile) {
+    return manager->registrar_.IsRegistered(
+        manager, type, content::Source<Profile>(profile));
+  }
+
+  // Allows access to a protected method.
+  static Profile* GetProfile(ExtensionProcessManager* manager) {
+    return manager->GetProfile();
+  }
 };
 
-}  // namespace
+// Test that notification registration works properly.
+TEST_F(ExtensionProcessManagerTest, ExtensionNotificationRegistration) {
+  // Test for a normal profile.
+  scoped_ptr<TestingProfile> original_profile(new TestingProfile);
+  scoped_ptr<ExtensionProcessManager> manager1(
+      ExtensionProcessManager::Create(original_profile.get()));
+
+  EXPECT_EQ(original_profile.get(), GetProfile(manager1.get()));
+  EXPECT_EQ(0u, manager1->background_hosts().size());
+
+  // It observes some notifications from all sources.
+  EXPECT_TRUE(IsRegistered(
+      manager1.get(), chrome::NOTIFICATION_BROWSER_WINDOW_READY, NULL));
+
+  // It observes other notifications from this profile.
+  EXPECT_TRUE(IsRegistered(manager1.get(),
+                           chrome::NOTIFICATION_EXTENSIONS_READY,
+                           original_profile.get()));
+  EXPECT_TRUE(IsRegistered(manager1.get(),
+                           chrome::NOTIFICATION_EXTENSION_LOADED,
+                           original_profile.get()));
+  EXPECT_TRUE(IsRegistered(manager1.get(),
+                           chrome::NOTIFICATION_EXTENSION_UNLOADED,
+                           original_profile.get()));
+  EXPECT_TRUE(IsRegistered(manager1.get(),
+                           chrome::NOTIFICATION_EXTENSION_HOST_DESTROYED,
+                           original_profile.get()));
+
+  // Now add an incognito profile associated with the master above.
+  TestingProfile::Builder builder;
+  builder.SetIncognito();
+  scoped_ptr<TestingProfile> incognito_profile = builder.Build();
+  incognito_profile->SetOriginalProfile(original_profile.get());
+  scoped_ptr<ExtensionProcessManager> manager2(
+      ExtensionProcessManager::Create(incognito_profile.get()));
+
+  EXPECT_EQ(incognito_profile.get(), GetProfile(manager2.get()));
+  EXPECT_EQ(0u, manager2->background_hosts().size());
+
+  // Some notifications are observed for the original profile.
+  EXPECT_TRUE(IsRegistered(manager2.get(),
+                           chrome::NOTIFICATION_EXTENSION_LOADED,
+                           original_profile.get()));
+
+  // Some notifications are observed for the incognito profile.
+  EXPECT_TRUE(IsRegistered(manager2.get(),
+                           chrome::NOTIFICATION_EXTENSION_HOST_DESTROYED,
+                           incognito_profile.get()));
+
+  // Some notifications are observed for both incognito and original.
+  EXPECT_TRUE(IsRegistered(manager2.get(),
+                           chrome::NOTIFICATION_PROFILE_DESTROYED,
+                           original_profile.get()));
+  EXPECT_TRUE(IsRegistered(manager2.get(),
+                           chrome::NOTIFICATION_PROFILE_DESTROYED,
+                           incognito_profile.get()));
+
+  // Some are not observed at all.
+  EXPECT_FALSE(IsRegistered(manager2.get(),
+                            chrome::NOTIFICATION_EXTENSIONS_READY,
+                            original_profile.get()));
+
+  // This notification is observed for incognito profiles only.
+  EXPECT_TRUE(IsRegistered(manager2.get(),
+                           chrome::NOTIFICATION_PROFILE_DESTROYED,
+                           incognito_profile.get()));
+}
 
 // Test that extensions get grouped in the right SiteInstance (and therefore
 // process) based on their URLs.
 TEST_F(ExtensionProcessManagerTest, ProcessGrouping) {
   // Extensions in different profiles should always be different SiteInstances.
   // Note: we don't initialize these, since we're not testing that
   // functionality.  This means we can get away with a NULL UserScriptMaster.
   TestingProfile profile1;
   scoped_ptr<ExtensionProcessManager> manager1(
       ExtensionProcessManager::Create(&profile1));
@@ skipping 15 matching lines @@
   EXPECT_EQ(site11, site12);
 
   scoped_refptr<SiteInstance> site21 =
       manager1->GetSiteInstanceForURL(ext2_url1);
   EXPECT_NE(site11, site21);
 
   scoped_refptr<SiteInstance> other_profile_site =
       manager2->GetSiteInstanceForURL(ext1_url1);
   EXPECT_NE(site11, other_profile_site);
 }