Migrate HostZoomMap to live in StoragePartition.

chrome/browser/profiles/host_zoom_map_browsertest.cc

 // Copyright 2014 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 "content/public/browser/host_zoom_map.h"
 
 #include <algorithm>
 #include <string>
 #include <vector>
 
 #include "base/bind.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/path_service.h"
 #include "base/prefs/pref_service.h"
 #include "base/strings/stringprintf.h"
 #include "base/values.h"
 #include "chrome/browser/chrome_page_zoom.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/profiles/profile_impl.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/ui/zoom/chrome_zoom_level_prefs.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_paths.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/testing_profile.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/test/test_utils.h"
 #include "net/dns/mock_host_resolver.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 #include "net/test/embedded_test_server/http_response.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "url/gurl.h"
 
+using content::HostZoomMap;
+
 namespace {
 
 class ZoomLevelChangeObserver {
  public:
   explicit ZoomLevelChangeObserver(Profile* profile)
       : message_loop_runner_(new content::MessageLoopRunner) {
     content::HostZoomMap* host_zoom_map = static_cast<content::HostZoomMap*>(
         content::HostZoomMap::GetDefaultForBrowserContext(profile));
     subscription_ = host_zoom_map->AddZoomLevelChangedCallback(base::Bind(
         &ZoomLevelChangeObserver::OnZoomLevelChanged, base::Unretained(this)));
@@ skipping 286 matching lines @@
 
   // Change the default zoom level and observe.
   SetDefaultZoomLevel(kNewDefaultZoomLevel);
   observer.BlockUntilZoomLevelForHostHasChanged(test_url.host());
   EXPECT_TRUE(
       content::ZoomValuesEqual(kNewDefaultZoomLevel, GetZoomLevel(test_url)));
   EXPECT_EQ(kNewDefaultZoomLevel, zoom_level_prefs->GetDefaultZoomLevelPref());
   EXPECT_EQ(0.0, profile_prefs->GetDouble(prefs::kDefaultZoomLevelDeprecated));
 }
 
+// Test four things:
+//  1. Host zoom maps of parent profile and child profile are different.
+//  2. Child host zoom map inherits zoom level at construction.
+//  3. Change of zoom level doesn't propagate from child to parent.
+//  4. Change of zoom level propagates from parent to child.
+IN_PROC_BROWSER_TEST_F(HostZoomMapBrowserTest,
+                       OffTheRecordProfileHostZoomMap) {
+  // Constants for test case.
+  const std::string host("example.com");
+  const double zoom_level_25 = 2.5;
+  const double zoom_level_30 = 3.0;
+  const double zoom_level_40 = 4.0;
+
+  Profile* parent_profile = browser()->profile();
+  Profile* child_profile =
+      static_cast<ProfileImpl*>(parent_profile)->GetOffTheRecordProfile();
+  HostZoomMap* parent_zoom_map =
+      HostZoomMap::GetDefaultForBrowserContext(parent_profile);
+  ASSERT_TRUE(parent_zoom_map);
+
+  parent_zoom_map->SetZoomLevelForHost(host, zoom_level_25);
+  ASSERT_EQ(parent_zoom_map->GetZoomLevelForHostAndScheme("http", host),
+      zoom_level_25);
+
+  // Prepare child host zoom map.
+  HostZoomMap* child_zoom_map =
+      HostZoomMap::GetDefaultForBrowserContext(child_profile);
+  ASSERT_TRUE(child_zoom_map);
+
+  // Verify.
+  EXPECT_NE(parent_zoom_map, child_zoom_map);
+
+  EXPECT_EQ(parent_zoom_map->GetZoomLevelForHostAndScheme("http", host),
+            child_zoom_map->GetZoomLevelForHostAndScheme("http", host)) <<
+                "Child must inherit from parent.";
+
+  child_zoom_map->SetZoomLevelForHost(host, zoom_level_30);
+  ASSERT_EQ(
+      child_zoom_map->GetZoomLevelForHostAndScheme("http", host),
+      zoom_level_30);
+
+  EXPECT_NE(parent_zoom_map->GetZoomLevelForHostAndScheme("http", host),
+            child_zoom_map->GetZoomLevelForHostAndScheme("http", host)) <<
+                "Child change must not propagate to parent.";
+
+  parent_zoom_map->SetZoomLevelForHost(host, zoom_level_40);
+  ASSERT_EQ(
+      parent_zoom_map->GetZoomLevelForHostAndScheme("http", host),
+      zoom_level_40);
+
+  EXPECT_EQ(parent_zoom_map->GetZoomLevelForHostAndScheme("http", host),
+            child_zoom_map->GetZoomLevelForHostAndScheme("http", host)) <<
+                "Parent change should propagate to child.";
+  base::RunLoop().RunUntilIdle();
+}