Split incognito extension content settings in session-only and persistent.

chrome/browser/content_settings/content_settings_pref_provider_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/content_settings/content_settings_pref_provider.h"
 
 #include "base/auto_reset.h"
 #include "base/command_line.h"
-#include "chrome/browser/content_settings/stub_settings_observer.h"
+#include "chrome/browser/content_settings/mock_settings_observer.h"
 #include "chrome/browser/prefs/browser_prefs.h"
 #include "chrome/browser/prefs/default_pref_store.h"
 #include "chrome/browser/prefs/overlay_persistent_pref_store.h"
 #include "chrome/browser/prefs/pref_service.h"
+#include "chrome/browser/prefs/pref_service_mock_builder.h"
 #include "chrome/browser/prefs/testing_pref_store.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/test/testing_browser_process_test.h"
 #include "chrome/test/testing_pref_service.h"
 #include "chrome/test/testing_profile.h"
 #include "content/browser/browser_thread.h"
 #include "googleurl/src/gurl.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
-namespace {
-class ContentSettingsPrefService : public PrefService {
- public:
-  ContentSettingsPrefService(PrefStore* managed_platform_prefs,
-                             PrefStore* managed_cloud_prefs,
-                             PrefStore* extension_prefs,
-                             PrefStore* command_line_prefs,
-                             PersistentPrefStore* user_prefs,
-                             PrefStore* recommended_platform_prefs,
-                             PrefStore* recommended_cloud_prefs,
-                             DefaultPrefStore* default_store)
-      : PrefService(
-          managed_platform_prefs, managed_cloud_prefs, extension_prefs,
-          command_line_prefs, user_prefs, recommended_platform_prefs,
-          recommended_cloud_prefs, default_store, false) {}
-  virtual ~ContentSettingsPrefService() {}
-};
-}
+using ::testing::_;
 
 namespace content_settings {
 
 class PrefDefaultProviderTest : public TestingBrowserProcessTest {
  public:
   PrefDefaultProviderTest()
       : ui_thread_(BrowserThread::UI, &message_loop_) {
   }
 
  protected:
@@ skipping 16 matching lines @@
   EXPECT_EQ(CONTENT_SETTING_BLOCK,
             provider.ProvideDefaultSetting(CONTENT_SETTINGS_TYPE_COOKIES));
   provider.ResetToDefaults();
   EXPECT_EQ(CONTENT_SETTING_ALLOW,
             provider.ProvideDefaultSetting(CONTENT_SETTINGS_TYPE_COOKIES));
 }
 
 TEST_F(PrefDefaultProviderTest, Observer) {
   TestingProfile profile;
   PrefDefaultProvider provider(&profile);
-  StubSettingsObserver observer;
+  MockSettingsObserver observer;
 
+  EXPECT_CALL(observer,
+              OnContentSettingsChanged(profile.GetHostContentSettingsMap(),
+                                       CONTENT_SETTINGS_TYPE_IMAGES, false,
+                                       _, true));
+  // Expect a second call because the PrefDefaultProvider in the TestingProfile
+  // also observes the default content settings preference.
+  EXPECT_CALL(observer,
+              OnContentSettingsChanged(profile.GetHostContentSettingsMap(),
+                                       CONTENT_SETTINGS_TYPE_DEFAULT, true,
+                                       _, true));
   provider.UpdateDefaultSetting(
       CONTENT_SETTINGS_TYPE_IMAGES, CONTENT_SETTING_BLOCK);
-  EXPECT_EQ(profile.GetHostContentSettingsMap(), observer.last_notifier);
-  EXPECT_TRUE(observer.last_update_all);
-  EXPECT_FALSE(observer.last_update_all_types);
-  EXPECT_EQ(1, observer.counter);
 }
 
 TEST_F(PrefDefaultProviderTest, ObserveDefaultPref) {
   TestingProfile profile;
   PrefDefaultProvider provider(&profile);
 
   PrefService* prefs = profile.GetPrefs();
 
   // Make a copy of the default pref value so we can reset it later.
   scoped_ptr<Value> default_value(prefs->FindPreference(
@@ skipping 72 matching lines @@
  protected:
   MessageLoop message_loop_;
   BrowserThread ui_thread_;
 };
 
 TEST_F(PrefProviderTest, Observer) {
   TestingProfile profile;
   profile.GetHostContentSettingsMap();
   Profile* p = &profile;
   PrefProvider pref_content_settings_provider(p);
-  StubSettingsObserver observer;
+  MockSettingsObserver observer;
   ContentSettingsPattern pattern =
       ContentSettingsPattern::FromString("[*.]example.com");
 
+  // Expect 2 calls: One from the update and one from canonicalization.
+  EXPECT_CALL(observer,
+              OnContentSettingsChanged(profile.GetHostContentSettingsMap(),
+                                       CONTENT_SETTINGS_TYPE_IMAGES, false,
+                                       pattern, false));
+  EXPECT_CALL(observer,
+              OnContentSettingsChanged(profile.GetHostContentSettingsMap(),
+                                       CONTENT_SETTINGS_TYPE_DEFAULT, true,
+                                       _, true));
   pref_content_settings_provider.SetContentSetting(
       pattern,
       pattern,
       CONTENT_SETTINGS_TYPE_IMAGES,
       "",
       CONTENT_SETTING_ALLOW);
-  EXPECT_EQ(profile.GetHostContentSettingsMap(), observer.last_notifier);
-  EXPECT_EQ(pattern, observer.last_pattern);
-  EXPECT_FALSE(observer.last_update_all);
-  EXPECT_FALSE(observer.last_update_all_types);
-  // Expect 2 calls: One from the update and one from canonicalization.
-  EXPECT_EQ(2, observer.counter);
 }
 
 // Test for regression in which the PrefProvider modified the user pref store
 // of the OTR unintentionally: http://crbug.com/74466.
 TEST_F(PrefProviderTest, Incognito) {
-  DefaultPrefStore* default_prefs = new DefaultPrefStore();
   PersistentPrefStore* user_prefs = new TestingPrefStore();
   OverlayPersistentPrefStore* otr_user_prefs =
       new OverlayPersistentPrefStore(user_prefs);
 
-  PrefService* regular_prefs = new ContentSettingsPrefService(
-      NULL,  // managed_platform_prefs
-      NULL,  // managed_cloud_prefs
-      NULL,  // extension_prefs
-      NULL,  // command_line_prefs
-      user_prefs,
-      NULL,  // recommended_platform_prefs,
-      NULL,  // recommended_cloud_prefs,
-      default_prefs);
+  PrefServiceMockBuilder builder;
+  PrefService* regular_prefs = builder.WithUserPrefs(user_prefs).Create();
 
   Profile::RegisterUserPrefs(regular_prefs);
   browser::RegisterUserPrefs(regular_prefs);
 
-  PrefService* otr_prefs = new ContentSettingsPrefService(
-        NULL,  // managed_platform_prefs
-        NULL,  // managed_cloud_prefs
-        NULL,  // extension_prefs
-        NULL,  // command_line_prefs
-        otr_user_prefs,
-        NULL,  // recommended_platform_prefs,
-        NULL,  // recommended_cloud_prefs,
-        default_prefs);
+  PrefService* otr_prefs = builder.WithUserPrefs(otr_user_prefs).Create();
+
+  Profile::RegisterUserPrefs(otr_prefs);
+  browser::RegisterUserPrefs(otr_prefs);
 
   TestingProfile profile;
   TestingProfile* otr_profile = new TestingProfile;
   profile.SetOffTheRecordProfile(otr_profile);
   profile.SetPrefService(regular_prefs);
   otr_profile->set_incognito(true);
   otr_profile->SetPrefService(otr_prefs);
   profile.GetHostContentSettingsMap();
 
   PrefProvider pref_content_settings_provider(&profile);
@@ skipping 108 matching lines @@
       CONTENT_SETTING_BLOCK);
   EXPECT_EQ(CONTENT_SETTING_BLOCK,
             pref_content_settings_provider.GetContentSetting(
                 host, host, CONTENT_SETTINGS_TYPE_PLUGINS, resource1));
   EXPECT_EQ(CONTENT_SETTING_DEFAULT,
             pref_content_settings_provider.GetContentSetting(
                 host, host, CONTENT_SETTINGS_TYPE_PLUGINS, resource2));
 }
 
 }  // namespace content_settings