Show apps protecting local data in chrome://settings/cookies.

chrome/browser/cookies_tree_model_unittest.cc

 // 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.
 
 #include "chrome/browser/cookies_tree_model.h"
 
 #include <string>
 
 #include "base/message_loop.h"
 #include "chrome/browser/content_settings/cookie_settings.h"
 #include "chrome/browser/content_settings/host_content_settings_map.h"
 #include "chrome/browser/content_settings/mock_settings_observer.h"
+#include "chrome/browser/extensions/extension_special_storage_policy.h"
 #include "chrome/browser/mock_browsing_data_appcache_helper.h"
 #include "chrome/browser/mock_browsing_data_cookie_helper.h"
 #include "chrome/browser/mock_browsing_data_database_helper.h"
 #include "chrome/browser/mock_browsing_data_file_system_helper.h"
 #include "chrome/browser/mock_browsing_data_indexed_db_helper.h"
 #include "chrome/browser/mock_browsing_data_local_storage_helper.h"
 #include "chrome/browser/mock_browsing_data_quota_helper.h"
 #include "chrome/browser/mock_browsing_data_server_bound_cert_helper.h"
 #include "chrome/test/base/testing_profile.h"
 #include "content/public/browser/notification_details.h"
@@ skipping 13 matching lines @@
 class CookiesTreeModelTest : public testing::Test {
  public:
   CookiesTreeModelTest()
       : ui_thread_(BrowserThread::UI, &message_loop_),
         file_user_blocking_(BrowserThread::FILE_USER_BLOCKING, &message_loop_),
         io_thread_(BrowserThread::IO, &message_loop_) {
   }
 
   virtual ~CookiesTreeModelTest() {
     // Avoid memory leaks.
+    special_storage_policy_ = NULL;
     profile_.reset();
     message_loop_.RunAllPending();
   }
 
   virtual void SetUp() OVERRIDE {
     profile_.reset(new TestingProfile());
     profile_->CreateRequestContext();
     mock_browsing_data_cookie_helper_ =
         new MockBrowsingDataCookieHelper(profile_->GetRequestContext());
     mock_browsing_data_database_helper_ =
@@ skipping 11 matching lines @@
     mock_browsing_data_quota_helper_ =
         new MockBrowsingDataQuotaHelper(profile_.get());
     mock_browsing_data_server_bound_cert_helper_ =
         new MockBrowsingDataServerBoundCertHelper();
 
     // It is fine to reuse the profile request context for the app, since
     // the mock cookie helper maintains its own list internally and doesn't
     // really use the request context. Same is true for the rest.
     mock_browsing_data_cookie_helper_app_ =
         new MockBrowsingDataCookieHelper(profile_->GetRequestContext());
+
+    scoped_refptr<CookieSettings> cookie_settings =
+        new CookieSettings(profile_->GetHostContentSettingsMap(),
+                           profile_->GetPrefs());
+    special_storage_policy_ =
+        new ExtensionSpecialStoragePolicy(cookie_settings);
   }
 
   virtual void TearDown() OVERRIDE {
     mock_browsing_data_server_bound_cert_helper_ = NULL;
     mock_browsing_data_quota_helper_ = NULL;
     mock_browsing_data_file_system_helper_ = NULL;
     mock_browsing_data_indexed_db_helper_ = NULL;
     mock_browsing_data_appcache_helper_ = NULL;
     mock_browsing_data_session_storage_helper_ = NULL;
     mock_browsing_data_local_storage_helper_ = NULL;
     mock_browsing_data_database_helper_ = NULL;
     message_loop_.RunAllPending();
   }
 
-  CookiesTreeModel* CreateCookiesTreeModelWithInitialSample(bool add_app) {
+  scoped_ptr<CookiesTreeModel> CreateCookiesTreeModelWithInitialSample(
+      bool add_app) {
     ContainerMap containers_map;
 
     containers_map[std::string()] = new LocalDataContainer(
         "Drive-By-Web", std::string(),
         mock_browsing_data_cookie_helper_,
         mock_browsing_data_database_helper_,
         mock_browsing_data_local_storage_helper_,
         mock_browsing_data_session_storage_helper_,
         mock_browsing_data_appcache_helper_,
         mock_browsing_data_indexed_db_helper_,
         mock_browsing_data_file_system_helper_,
         mock_browsing_data_quota_helper_,
         mock_browsing_data_server_bound_cert_helper_);
 
     if (add_app) {
       std::string app_id = "some-random-id";
       // The cookie helper is mandatory, the rest can be NULL.
       containers_map[app_id] = new LocalDataContainer(
           "Isolated App", app_id,
           mock_browsing_data_cookie_helper_app_,
           NULL, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
     }
 
-    CookiesTreeModel* cookies_model = new CookiesTreeModel(containers_map,
-                                                           false);
+    CookiesTreeModel* cookies_model =
+        new CookiesTreeModel(containers_map,
+                             special_storage_policy_,
+                             false);
     mock_browsing_data_cookie_helper_->
         AddCookieSamples(GURL("http://foo1"), "A=1");
     mock_browsing_data_cookie_helper_->
         AddCookieSamples(GURL("http://foo2"), "B=1");
     mock_browsing_data_cookie_helper_->
         AddCookieSamples(GURL("http://foo3"), "C=1");
     mock_browsing_data_cookie_helper_->Notify();
     mock_browsing_data_database_helper_->AddDatabaseSamples();
     mock_browsing_data_database_helper_->Notify();
     mock_browsing_data_local_storage_helper_->AddLocalStorageSamples();
@@ skipping 63 matching lines @@
                 GetDisplayedSessionStorages(cookies_model));
       EXPECT_EQ("http://idbhost1:1/,http://idbhost2:2/",
                 GetDisplayedIndexedDBs(cookies_model));
       EXPECT_EQ("http://fshost1:1/,http://fshost2:2/,http://fshost3:3/",
                 GetDisplayedFileSystems(cookies_model));
       EXPECT_EQ("quotahost1,quotahost2",
                 GetDisplayedQuotas(cookies_model));
       EXPECT_EQ("sbc1,sbc2",
                 GetDisplayedServerBoundCerts(cookies_model));
     }
-    return cookies_model;
+    return make_scoped_ptr(cookies_model);
   }
 
   std::string GetNodesOfChildren(
       const CookieTreeNode* node,
       CookieTreeNode::DetailedInfo::NodeType node_type) {
     if (!node->empty()) {
       std::string retval;
       for (int i = 0; i < node->child_count(); ++i) {
         retval += GetNodesOfChildren(node->GetChild(i), node_type);
       }
       return retval;
-    } else {
-      if (node->GetDetailedInfo().node_type == node_type) {
-        switch (node_type) {
-          case CookieTreeNode::DetailedInfo::TYPE_SESSION_STORAGE:
-            return node->GetDetailedInfo().
-                session_storage_info->origin_url.spec() + ",";
-          case CookieTreeNode::DetailedInfo::TYPE_LOCAL_STORAGE:
-            return node->GetDetailedInfo().
-                local_storage_info->origin_url.spec() + ",";
-          case CookieTreeNode::DetailedInfo::TYPE_DATABASE:
-            return node->GetDetailedInfo().database_info->database_name + ",";
-          case CookieTreeNode::DetailedInfo::TYPE_COOKIE:
-            return node->GetDetailedInfo().cookie->Name() + ",";
-          case CookieTreeNode::DetailedInfo::TYPE_APPCACHE:
-            return node->GetDetailedInfo().appcache_info->manifest_url.spec() +
-                   ",";
-          case CookieTreeNode::DetailedInfo::TYPE_INDEXED_DB:
-            return node->GetDetailedInfo().indexed_db_info->origin.spec() +
-                   ",";
-          case CookieTreeNode::DetailedInfo::TYPE_FILE_SYSTEM:
-            return node->GetDetailedInfo().file_system_info->origin.spec() +
-                   ",";
-          case CookieTreeNode::DetailedInfo::TYPE_QUOTA:
-            return node->GetDetailedInfo().quota_info->host + ",";
-          case CookieTreeNode::DetailedInfo::TYPE_SERVER_BOUND_CERT:
-            return node->GetDetailedInfo(
-                ).server_bound_cert->server_identifier() + ",";
-          default:
-            return "";
-        }
-      } else {
-        return "";
-      }
+    }
+
+    if (node->GetDetailedInfo().node_type != node_type)
+      return std::string();
+
+    switch (node_type) {
+      case CookieTreeNode::DetailedInfo::TYPE_SESSION_STORAGE:
+        return node->GetDetailedInfo().
+            session_storage_info->origin_url.spec() + ",";
+      case CookieTreeNode::DetailedInfo::TYPE_LOCAL_STORAGE:
+        return node->GetDetailedInfo().
+            local_storage_info->origin_url.spec() + ",";
+      case CookieTreeNode::DetailedInfo::TYPE_DATABASE:
+        return node->GetDetailedInfo().database_info->database_name + ",";
+      case CookieTreeNode::DetailedInfo::TYPE_COOKIE:
+        return node->GetDetailedInfo().cookie->Name() + ",";
+      case CookieTreeNode::DetailedInfo::TYPE_APPCACHE:
+        return node->GetDetailedInfo().appcache_info->manifest_url.spec() +
+               ",";
+      case CookieTreeNode::DetailedInfo::TYPE_INDEXED_DB:
+        return node->GetDetailedInfo().indexed_db_info->origin.spec() +
+               ",";
+      case CookieTreeNode::DetailedInfo::TYPE_FILE_SYSTEM:
+        return node->GetDetailedInfo().file_system_info->origin.spec() +
+               ",";
+      case CookieTreeNode::DetailedInfo::TYPE_QUOTA:
+        return node->GetDetailedInfo().quota_info->host + ",";
+      case CookieTreeNode::DetailedInfo::TYPE_SERVER_BOUND_CERT:
+        return node->GetDetailedInfo(
+            ).server_bound_cert->server_identifier() + ",";
+      default:
+        return std::string();
     }
   }
 
   std::string GetCookiesOfChildren(const CookieTreeNode* node) {
     return GetNodesOfChildren(node, CookieTreeNode::DetailedInfo::TYPE_COOKIE);
   }
 
   std::string GetDatabasesOfChildren(const CookieTreeNode* node) {
     return GetNodesOfChildren(node,
                               CookieTreeNode::DetailedInfo::TYPE_DATABASE);
@@ skipping 112 matching lines @@
   scoped_refptr<MockBrowsingDataFileSystemHelper>
       mock_browsing_data_file_system_helper_;
   scoped_refptr<MockBrowsingDataQuotaHelper>
       mock_browsing_data_quota_helper_;
   scoped_refptr<MockBrowsingDataServerBoundCertHelper>
       mock_browsing_data_server_bound_cert_helper_;
 
   // App helpers.
   scoped_refptr<MockBrowsingDataCookieHelper>
       mock_browsing_data_cookie_helper_app_;
+
+  scoped_refptr<ExtensionSpecialStoragePolicy> special_storage_policy_;
 };
 
 TEST_F(CookiesTreeModelTest, RemoveAll) {
   scoped_ptr<CookiesTreeModel> cookies_model(
       CreateCookiesTreeModelWithInitialSample(false));
 
   // Reset the selection of the first row.
   {
     SCOPED_TRACE("Before removing");
     EXPECT_EQ("A,B,C",
@@ skipping 421 matching lines @@
       "Drive-By-Web", std::string(),
       mock_browsing_data_cookie_helper_,
       mock_browsing_data_database_helper_,
       mock_browsing_data_local_storage_helper_,
       mock_browsing_data_session_storage_helper_,
       mock_browsing_data_appcache_helper_,
       mock_browsing_data_indexed_db_helper_,
       mock_browsing_data_file_system_helper_,
       mock_browsing_data_quota_helper_,
       mock_browsing_data_server_bound_cert_helper_);
-  CookiesTreeModel cookies_model(container_map, false);
+  CookiesTreeModel cookies_model(container_map, special_storage_policy_, false);
 
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo1"), "A=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo2"), "B=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo3"), "C=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo3"), "D=1");
   mock_browsing_data_cookie_helper_->Notify();
@@ skipping 69 matching lines @@
       "Drive-By-Web", std::string(),
       mock_browsing_data_cookie_helper_,
       mock_browsing_data_database_helper_,
       mock_browsing_data_local_storage_helper_,
       mock_browsing_data_session_storage_helper_,
       mock_browsing_data_appcache_helper_,
       mock_browsing_data_indexed_db_helper_,
       mock_browsing_data_file_system_helper_,
       mock_browsing_data_quota_helper_,
       mock_browsing_data_server_bound_cert_helper_);
-  CookiesTreeModel cookies_model(container_map, false);
+  CookiesTreeModel cookies_model(container_map, special_storage_policy_,false);
 
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo1"), "A=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo2"), "B=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo3"), "C=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo3"), "D=1");
   mock_browsing_data_cookie_helper_->
@@ skipping 72 matching lines @@
       "Drive-By-Web", std::string(),
       mock_browsing_data_cookie_helper_,
       mock_browsing_data_database_helper_,
       mock_browsing_data_local_storage_helper_,
       mock_browsing_data_session_storage_helper_,
       mock_browsing_data_appcache_helper_,
       mock_browsing_data_indexed_db_helper_,
       mock_browsing_data_file_system_helper_,
       mock_browsing_data_quota_helper_,
       mock_browsing_data_server_bound_cert_helper_);
-  CookiesTreeModel cookies_model(container_map, false);
+  CookiesTreeModel cookies_model(container_map, special_storage_policy_,false);
 
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo1"), "A=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo2"), "B=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo3"), "C=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo3"), "D=1");
   mock_browsing_data_cookie_helper_->
@@ skipping 23 matching lines @@
       "Drive-By-Web", std::string(),
       mock_browsing_data_cookie_helper_,
       mock_browsing_data_database_helper_,
       mock_browsing_data_local_storage_helper_,
       mock_browsing_data_session_storage_helper_,
       mock_browsing_data_appcache_helper_,
       mock_browsing_data_indexed_db_helper_,
       mock_browsing_data_file_system_helper_,
       mock_browsing_data_quota_helper_,
       mock_browsing_data_server_bound_cert_helper_);
-  CookiesTreeModel cookies_model(container_map, false);
+  CookiesTreeModel cookies_model(container_map, special_storage_policy_, false);
 
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://a.foo2.com"), "A=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo2.com"), "B=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://b.foo1.com"), "C=1");
   // Leading dot on the foo4
   mock_browsing_data_cookie_helper_->AddCookieSamples(
       GURL("http://foo4.com"), "D=1; domain=.foo4.com; path=/;");
@@ skipping 30 matching lines @@
       name, browser_id,
       mock_browsing_data_cookie_helper_,
       mock_browsing_data_database_helper_,
       mock_browsing_data_local_storage_helper_,
       mock_browsing_data_session_storage_helper_,
       mock_browsing_data_appcache_helper_,
       mock_browsing_data_indexed_db_helper_,
       mock_browsing_data_file_system_helper_,
       mock_browsing_data_quota_helper_,
       mock_browsing_data_server_bound_cert_helper_);
-  CookiesTreeModel cookies_model(container_map, false);
+  CookiesTreeModel cookies_model(container_map, special_storage_policy_, false);
 
   mock_browsing_data_cookie_helper_->AddCookieSamples(host, "A=1");
   mock_browsing_data_cookie_helper_->Notify();
 
   TestingProfile profile;
   HostContentSettingsMap* content_settings =
       profile.GetHostContentSettingsMap();
   CookieSettings* cookie_settings =
       CookieSettings::Factory::GetForProfile(&profile);
   MockSettingsObserver observer;
 
   CookieTreeRootNode* root =
       static_cast<CookieTreeRootNode*>(cookies_model.GetRoot());
-  CookieTreeOriginNode* origin = root->GetOrCreateOriginNode(host,
-                                                             browser_id,
-                                                             name);
+  CookieTreeHostNode* origin =
+      root->GetOrCreateHostNode(host, browser_id, name);
 
   EXPECT_EQ(1, origin->child_count());
   EXPECT_TRUE(origin->CanCreateContentException());
   EXPECT_CALL(observer,
               OnContentSettingsChanged(
                   content_settings,
                   CONTENT_SETTINGS_TYPE_COOKIES,
                   false,
                   ContentSettingsPattern::FromURLNoWildcard(host),
                   ContentSettingsPattern::Wildcard(),
@@ skipping 18 matching lines @@
   EXPECT_EQ(ASCIIToUTF16("Isolated App, app-origin1"),
       cookies_model->GetRoot()->GetChild(16)->GetTitle());
   EXPECT_EQ(ASCIIToUTF16("Isolated App, app-origin2"),
       cookies_model->GetRoot()->GetChild(17)->GetTitle());
 }
 
 TEST_F(CookiesTreeModelTest, FileSystemFilter) {
   scoped_ptr<CookiesTreeModel> cookies_model(
       CreateCookiesTreeModelWithInitialSample(false));
 
-  cookies_model->UpdateSearchResults(string16(ASCIIToUTF16("fshost1")));
+  cookies_model->UpdateSearchResults(ASCIIToUTF16("fshost1"));
   EXPECT_EQ("http://fshost1:1/",
             GetDisplayedFileSystems(cookies_model.get()));
 
-  cookies_model->UpdateSearchResults(string16(ASCIIToUTF16("fshost2")));
+  cookies_model->UpdateSearchResults(ASCIIToUTF16("fshost2"));
   EXPECT_EQ("http://fshost2:2/",
             GetDisplayedFileSystems(cookies_model.get()));
 
-  cookies_model->UpdateSearchResults(string16(ASCIIToUTF16("fshost3")));
+  cookies_model->UpdateSearchResults(ASCIIToUTF16("fshost3"));
   EXPECT_EQ("http://fshost3:3/",
             GetDisplayedFileSystems(cookies_model.get()));
 
   cookies_model->UpdateSearchResults(string16());
   EXPECT_EQ("http://fshost1:1/,http://fshost2:2/,http://fshost3:3/",
             GetDisplayedFileSystems(cookies_model.get()));
 }
 
 TEST_F(CookiesTreeModelTest, CookiesFilter) {
   ContainerMap container_map;
 
   container_map[std::string()] = new LocalDataContainer(
       "Drive-By-Web", std::string(),
       mock_browsing_data_cookie_helper_,
       mock_browsing_data_database_helper_,
       mock_browsing_data_local_storage_helper_,
       mock_browsing_data_session_storage_helper_,
       mock_browsing_data_appcache_helper_,
       mock_browsing_data_indexed_db_helper_,
       mock_browsing_data_file_system_helper_,
       mock_browsing_data_quota_helper_,
       mock_browsing_data_server_bound_cert_helper_);
-  CookiesTreeModel cookies_model(container_map, false);
+  CookiesTreeModel cookies_model(container_map, special_storage_policy_, false);
 
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://123.com"), "A=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo1.com"), "B=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo2.com"), "C=1");
   mock_browsing_data_cookie_helper_->
       AddCookieSamples(GURL("http://foo3.com"), "D=1");
   mock_browsing_data_cookie_helper_->Notify();
   EXPECT_EQ("A,B,C,D", GetDisplayedCookies(&cookies_model));
 
   cookies_model.UpdateSearchResults(string16(ASCIIToUTF16("foo")));
   EXPECT_EQ("B,C,D", GetDisplayedCookies(&cookies_model));
 
   cookies_model.UpdateSearchResults(string16(ASCIIToUTF16("2")));
   EXPECT_EQ("A,C", GetDisplayedCookies(&cookies_model));
 
   cookies_model.UpdateSearchResults(string16(ASCIIToUTF16("foo3")));
   EXPECT_EQ("D", GetDisplayedCookies(&cookies_model));
 
   cookies_model.UpdateSearchResults(string16());
   EXPECT_EQ("A,B,C,D", GetDisplayedCookies(&cookies_model));
 }
 
 }  // namespace