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_ =
       new MockBrowsingDataDatabaseHelper(profile_.get());
     mock_browsing_data_local_storage_helper_ =
       new MockBrowsingDataLocalStorageHelper(profile_.get());
     mock_browsing_data_session_storage_helper_ =
       new MockBrowsingDataLocalStorageHelper(profile_.get());
     mock_browsing_data_appcache_helper_ =
       new MockBrowsingDataAppCacheHelper(profile_.get());
     mock_browsing_data_indexed_db_helper_ =
       new MockBrowsingDataIndexedDBHelper();
     mock_browsing_data_file_system_helper_ =
       new MockBrowsingDataFileSystemHelper(profile_.get());
     mock_browsing_data_quota_helper_ =
       new MockBrowsingDataQuotaHelper(profile_.get());
     mock_browsing_data_server_bound_cert_helper_ =
       new MockBrowsingDataServerBoundCertHelper();
+    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() {
+  scoped_ptr<CookiesTreeModel> CreateCookiesTreeModelWithInitialSample() {
     CookiesTreeModel* cookies_model = new CookiesTreeModel(
         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_,
+        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();
@@ skipping 41 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 108 matching lines @@
   scoped_refptr<MockBrowsingDataAppCacheHelper>
       mock_browsing_data_appcache_helper_;
   scoped_refptr<MockBrowsingDataIndexedDBHelper>
       mock_browsing_data_indexed_db_helper_;
   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_;
+  scoped_refptr<ExtensionSpecialStoragePolicy> special_storage_policy_;
 };
 
 TEST_F(CookiesTreeModelTest, RemoveAll) {
   scoped_ptr<CookiesTreeModel> cookies_model(
       CreateCookiesTreeModelWithInitialSample());
 
   // Reset the selection of the first row.
   {
     SCOPED_TRACE("Before removing");
     EXPECT_EQ("A,B,C",
@@ skipping 18 matching lines @@
   mock_browsing_data_database_helper_->Reset();
   mock_browsing_data_local_storage_helper_->Reset();
   mock_browsing_data_session_storage_helper_->Reset();
   mock_browsing_data_indexed_db_helper_->Reset();
   mock_browsing_data_file_system_helper_->Reset();
 
   cookies_model->DeleteAllStoredObjects();
 
   // Make sure the nodes are also deleted from the model's cache.
   // http://crbug.com/43249
-  cookies_model->UpdateSearchResults(std::wstring());
+  cookies_model->UpdateSearchResults(string16());
 
   {
     SCOPED_TRACE("After removing");
     EXPECT_EQ(1, cookies_model->GetRoot()->GetTotalNodeCount());
     EXPECT_EQ(0, cookies_model->GetRoot()->child_count());
     EXPECT_EQ(std::string(""), GetDisplayedCookies(cookies_model.get()));
     EXPECT_TRUE(mock_browsing_data_cookie_helper_->AllDeleted());
     EXPECT_TRUE(mock_browsing_data_database_helper_->AllDeleted());
     EXPECT_TRUE(mock_browsing_data_local_storage_helper_->AllDeleted());
     EXPECT_FALSE(mock_browsing_data_session_storage_helper_->AllDeleted());
@@ skipping 397 matching lines @@
 TEST_F(CookiesTreeModelTest, RemoveSingleCookieNode) {
   CookiesTreeModel cookies_model(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_,
+                                 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 61 matching lines @@
 TEST_F(CookiesTreeModelTest, RemoveSingleCookieNodeOf3) {
   CookiesTreeModel cookies_model(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_,
+                                 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 65 matching lines @@
 TEST_F(CookiesTreeModelTest, RemoveSecondOrigin) {
   CookiesTreeModel cookies_model(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_,
+                                 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 19 matching lines @@
 TEST_F(CookiesTreeModelTest, OriginOrdering) {
   CookiesTreeModel cookies_model(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_,
+                                 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 23 matching lines @@
   GURL host("http://example.com/");
   CookiesTreeModel cookies_model(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_,
+                                 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);
+  CookieTreeHostNode* origin = root->GetOrCreateHostNode(host);
 
   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(),
                   false));
   EXPECT_CALL(observer,
               OnContentSettingsChanged(content_settings,
                   CONTENT_SETTINGS_TYPE_COOKIES,
                   false,
                   ContentSettingsPattern::FromURL(host),
                   ContentSettingsPattern::Wildcard(),
                   false));
   origin->CreateContentException(
       cookie_settings, CONTENT_SETTING_SESSION_ONLY);
   EXPECT_TRUE(cookie_settings->IsReadingCookieAllowed(host, host));
   EXPECT_TRUE(cookie_settings->IsCookieSessionOnly(host));
 }
 
 TEST_F(CookiesTreeModelTest, FileSystemFilter) {
   scoped_ptr<CookiesTreeModel> cookies_model(
       CreateCookiesTreeModelWithInitialSample());
 
-  cookies_model->UpdateSearchResults(std::wstring(L"fshost1"));
+  cookies_model->UpdateSearchResults(ASCIIToUTF16("fshost1"));
   EXPECT_EQ("http://fshost1:1/",
             GetDisplayedFileSystems(cookies_model.get()));
 
-  cookies_model->UpdateSearchResults(std::wstring(L"fshost2"));
+  cookies_model->UpdateSearchResults(ASCIIToUTF16("fshost2"));
   EXPECT_EQ("http://fshost2:2/",
             GetDisplayedFileSystems(cookies_model.get()));
 
-  cookies_model->UpdateSearchResults(std::wstring(L"fshost3"));
+  cookies_model->UpdateSearchResults(ASCIIToUTF16("fshost3"));
   EXPECT_EQ("http://fshost3:3/",
             GetDisplayedFileSystems(cookies_model.get()));
 
-  cookies_model->UpdateSearchResults(std::wstring());
+  cookies_model->UpdateSearchResults(string16());
   EXPECT_EQ("http://fshost1:1/,http://fshost2:2/,http://fshost3:3/",
             GetDisplayedFileSystems(cookies_model.get()));
 }
 
 }  // namespace