Index: chrome/browser/cookies_tree_model_unittest.cc |
diff --git a/chrome/browser/cookies_tree_model_unittest.cc b/chrome/browser/cookies_tree_model_unittest.cc |
index 9d0cdd7ff0aea4fc0dcaced7e6c69b226131d0a0..9e047c74cf8b780e90f4e6f0ffbce165e4e98786 100644 |
--- a/chrome/browser/cookies_tree_model_unittest.cc |
+++ b/chrome/browser/cookies_tree_model_unittest.cc |
@@ -10,6 +10,7 @@ |
#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" |
@@ -43,6 +44,7 @@ class CookiesTreeModelTest : public testing::Test { |
virtual ~CookiesTreeModelTest() { |
// Avoid memory leaks. |
+ special_storage_policy_ = NULL; |
profile_.reset(); |
message_loop_.RunAllPending(); |
} |
@@ -74,6 +76,12 @@ class CookiesTreeModelTest : public testing::Test { |
// 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 { |
@@ -88,7 +96,8 @@ class CookiesTreeModelTest : public testing::Test { |
message_loop_.RunAllPending(); |
} |
- CookiesTreeModel* CreateCookiesTreeModelWithInitialSample(bool add_app) { |
+ scoped_ptr<CookiesTreeModel> CreateCookiesTreeModelWithInitialSample( |
+ bool add_app) { |
ContainerMap containers_map; |
containers_map[std::string()] = new LocalDataContainer( |
@@ -112,8 +121,10 @@ class CookiesTreeModelTest : public testing::Test { |
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_-> |
@@ -197,7 +208,7 @@ class CookiesTreeModelTest : public testing::Test { |
EXPECT_EQ("sbc1,sbc2", |
GetDisplayedServerBoundCerts(cookies_model)); |
} |
- return cookies_model; |
+ return make_scoped_ptr(cookies_model); |
} |
std::string GetNodesOfChildren( |
@@ -209,39 +220,38 @@ class CookiesTreeModelTest : public testing::Test { |
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(); |
} |
} |
@@ -374,6 +384,8 @@ class CookiesTreeModelTest : public testing::Test { |
// App helpers. |
scoped_refptr<MockBrowsingDataCookieHelper> |
mock_browsing_data_cookie_helper_app_; |
+ |
+ scoped_refptr<ExtensionSpecialStoragePolicy> special_storage_policy_; |
}; |
TEST_F(CookiesTreeModelTest, RemoveAll) { |
@@ -815,7 +827,7 @@ TEST_F(CookiesTreeModelTest, RemoveSingleCookieNode) { |
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"); |
@@ -905,7 +917,7 @@ TEST_F(CookiesTreeModelTest, RemoveSingleCookieNodeOf3) { |
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"); |
@@ -998,7 +1010,7 @@ TEST_F(CookiesTreeModelTest, RemoveSecondOrigin) { |
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"); |
@@ -1042,7 +1054,7 @@ TEST_F(CookiesTreeModelTest, OriginOrdering) { |
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"); |
@@ -1093,7 +1105,7 @@ TEST_F(CookiesTreeModelTest, ContentSettings) { |
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(); |
@@ -1107,9 +1119,8 @@ TEST_F(CookiesTreeModelTest, ContentSettings) { |
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()); |
@@ -1148,15 +1159,15 @@ 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())); |
@@ -1179,7 +1190,7 @@ TEST_F(CookiesTreeModelTest, CookiesFilter) { |
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"); |