Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(278)

Side by Side Diff: chrome/browser/extensions/extension_special_storage_policy_unittest.cc

Issue 7619010: Session-only local storage cleared on exit. (Closed) Base URL: http://git.chromium.org/git/chromium.git@trunk
Patch Set: Code review comments (tiny). Created 9 years, 4 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "base/values.h" 5 #include "base/values.h"
6 #include "chrome/browser/extensions/extension_special_storage_policy.h" 6 #include "chrome/browser/extensions/extension_special_storage_policy.h"
7 #include "chrome/common/extensions/extension.h" 7 #include "chrome/common/extensions/extension.h"
8 #include "chrome/common/extensions/extension_constants.h" 8 #include "chrome/common/extensions/extension_constants.h"
9 #include "testing/gtest/include/gtest/gtest.h" 9 #include "testing/gtest/include/gtest/gtest.h"
10 10
(...skipping 100 matching lines...) Expand 10 before | Expand all | Expand 10 after
111 EXPECT_TRUE(handler_app.get()) << error; 111 EXPECT_TRUE(handler_app.get()) << error;
112 return handler_app; 112 return handler_app;
113 } 113 }
114 }; 114 };
115 115
116 TEST_F(ExtensionSpecialStoragePolicyTest, EmptyPolicy) { 116 TEST_F(ExtensionSpecialStoragePolicyTest, EmptyPolicy) {
117 const GURL kHttpUrl("http://foo"); 117 const GURL kHttpUrl("http://foo");
118 const GURL kExtensionUrl("chrome-extension://bar"); 118 const GURL kExtensionUrl("chrome-extension://bar");
119 119
120 scoped_refptr<ExtensionSpecialStoragePolicy> policy( 120 scoped_refptr<ExtensionSpecialStoragePolicy> policy(
121 new ExtensionSpecialStoragePolicy); 121 new ExtensionSpecialStoragePolicy(NULL));
122 122
123 ASSERT_FALSE(policy->IsStorageUnlimited(kHttpUrl)); 123 ASSERT_FALSE(policy->IsStorageUnlimited(kHttpUrl));
124 ASSERT_FALSE(policy->IsStorageUnlimited(kHttpUrl)); // test cached result 124 ASSERT_FALSE(policy->IsStorageUnlimited(kHttpUrl)); // test cached result
125 ASSERT_FALSE(policy->IsStorageUnlimited(kExtensionUrl)); 125 ASSERT_FALSE(policy->IsStorageUnlimited(kExtensionUrl));
126 ASSERT_FALSE(policy->IsStorageProtected(kHttpUrl)); 126 ASSERT_FALSE(policy->IsStorageProtected(kHttpUrl));
127 127
128 // This one is just based on the scheme. 128 // This one is just based on the scheme.
129 ASSERT_TRUE(policy->IsStorageProtected(kExtensionUrl)); 129 ASSERT_TRUE(policy->IsStorageProtected(kExtensionUrl));
130 } 130 }
131 131
132 132
133 TEST_F(ExtensionSpecialStoragePolicyTest, AppWithProtectedStorage) { 133 TEST_F(ExtensionSpecialStoragePolicyTest, AppWithProtectedStorage) {
134 scoped_refptr<Extension> extension(CreateProtectedApp()); 134 scoped_refptr<Extension> extension(CreateProtectedApp());
135 scoped_refptr<ExtensionSpecialStoragePolicy> policy( 135 scoped_refptr<ExtensionSpecialStoragePolicy> policy(
136 new ExtensionSpecialStoragePolicy); 136 new ExtensionSpecialStoragePolicy(NULL));
137 policy->GrantRightsForExtension(extension); 137 policy->GrantRightsForExtension(extension);
138 EXPECT_FALSE(policy->IsStorageUnlimited(extension->url())); 138 EXPECT_FALSE(policy->IsStorageUnlimited(extension->url()));
139 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("http://explicit/"))); 139 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("http://explicit/")));
140 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit/"))); 140 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit/")));
141 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit:6000/"))); 141 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit:6000/")));
142 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://foo.wildcards/"))); 142 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://foo.wildcards/")));
143 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://bar.wildcards/"))); 143 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://bar.wildcards/")));
144 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://not_listed/"))); 144 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://not_listed/")));
145 145
146 policy->RevokeRightsForExtension(extension); 146 policy->RevokeRightsForExtension(extension);
147 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://explicit/"))); 147 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://explicit/")));
148 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://foo.wildcards/"))); 148 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://foo.wildcards/")));
149 EXPECT_FALSE(policy->IsStorageProtected(GURL("https://bar.wildcards/"))); 149 EXPECT_FALSE(policy->IsStorageProtected(GURL("https://bar.wildcards/")));
150 } 150 }
151 151
152 TEST_F(ExtensionSpecialStoragePolicyTest, AppWithUnlimitedStorage) { 152 TEST_F(ExtensionSpecialStoragePolicyTest, AppWithUnlimitedStorage) {
153 scoped_refptr<Extension> extension(CreateUnlimitedApp()); 153 scoped_refptr<Extension> extension(CreateUnlimitedApp());
154 scoped_refptr<ExtensionSpecialStoragePolicy> policy( 154 scoped_refptr<ExtensionSpecialStoragePolicy> policy(
155 new ExtensionSpecialStoragePolicy); 155 new ExtensionSpecialStoragePolicy(NULL));
156 policy->GrantRightsForExtension(extension); 156 policy->GrantRightsForExtension(extension);
157 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit/"))); 157 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit/")));
158 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit:6000/"))); 158 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit:6000/")));
159 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://foo.wildcards/"))); 159 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://foo.wildcards/")));
160 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://foo.wildcards/"))); 160 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://foo.wildcards/")));
161 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://bar.wildcards/"))); 161 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://bar.wildcards/")));
162 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://not_listed/"))); 162 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://not_listed/")));
163 EXPECT_TRUE(policy->IsStorageUnlimited(extension->url())); 163 EXPECT_TRUE(policy->IsStorageUnlimited(extension->url()));
164 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("http://explicit/"))); 164 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("http://explicit/")));
165 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("http://explicit:6000/"))); 165 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("http://explicit:6000/")));
166 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("https://foo.wildcards/"))); 166 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("https://foo.wildcards/")));
167 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("https://bar.wildcards/"))); 167 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("https://bar.wildcards/")));
168 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("http://not_listed/"))); 168 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("http://not_listed/")));
169 169
170 policy->RevokeRightsForExtension(extension); 170 policy->RevokeRightsForExtension(extension);
171 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://explicit/"))); 171 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://explicit/")));
172 EXPECT_FALSE(policy->IsStorageProtected(GURL("https://foo.wildcards/"))); 172 EXPECT_FALSE(policy->IsStorageProtected(GURL("https://foo.wildcards/")));
173 EXPECT_FALSE(policy->IsStorageProtected(GURL("https://foo.wildcards/"))); 173 EXPECT_FALSE(policy->IsStorageProtected(GURL("https://foo.wildcards/")));
174 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://bar.wildcards/"))); 174 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://bar.wildcards/")));
175 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("http://explicit/"))); 175 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("http://explicit/")));
176 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("https://foo.wildcards/"))); 176 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("https://foo.wildcards/")));
177 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("https://bar.wildcards/"))); 177 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("https://bar.wildcards/")));
178 } 178 }
179 179
180 TEST_F(ExtensionSpecialStoragePolicyTest, OverlappingApps) { 180 TEST_F(ExtensionSpecialStoragePolicyTest, OverlappingApps) {
181 scoped_refptr<Extension> protected_app(CreateProtectedApp()); 181 scoped_refptr<Extension> protected_app(CreateProtectedApp());
182 scoped_refptr<Extension> unlimited_app(CreateUnlimitedApp()); 182 scoped_refptr<Extension> unlimited_app(CreateUnlimitedApp());
183 scoped_refptr<ExtensionSpecialStoragePolicy> policy( 183 scoped_refptr<ExtensionSpecialStoragePolicy> policy(
184 new ExtensionSpecialStoragePolicy); 184 new ExtensionSpecialStoragePolicy(NULL));
185 policy->GrantRightsForExtension(protected_app); 185 policy->GrantRightsForExtension(protected_app);
186 policy->GrantRightsForExtension(unlimited_app); 186 policy->GrantRightsForExtension(unlimited_app);
187 187
188 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit/"))); 188 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit/")));
189 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit:6000/"))); 189 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit:6000/")));
190 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://foo.wildcards/"))); 190 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://foo.wildcards/")));
191 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://foo.wildcards/"))); 191 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://foo.wildcards/")));
192 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://bar.wildcards/"))); 192 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://bar.wildcards/")));
193 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://not_listed/"))); 193 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://not_listed/")));
194 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("http://explicit/"))); 194 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("http://explicit/")));
195 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("http://explicit:6000/"))); 195 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("http://explicit:6000/")));
196 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("https://foo.wildcards/"))); 196 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("https://foo.wildcards/")));
197 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("https://bar.wildcards/"))); 197 EXPECT_TRUE(policy->IsStorageUnlimited(GURL("https://bar.wildcards/")));
198 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("http://not_listed/"))); 198 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("http://not_listed/")));
199 199
200 policy->RevokeRightsForExtension(unlimited_app); 200 policy->RevokeRightsForExtension(unlimited_app);
201 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("http://explicit/"))); 201 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("http://explicit/")));
202 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("https://foo.wildcards/"))); 202 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("https://foo.wildcards/")));
203 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("https://bar.wildcards/"))); 203 EXPECT_FALSE(policy->IsStorageUnlimited(GURL("https://bar.wildcards/")));
204 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit/"))); 204 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://explicit/")));
205 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://foo.wildcards/"))); 205 EXPECT_TRUE(policy->IsStorageProtected(GURL("http://foo.wildcards/")));
206 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://bar.wildcards/"))); 206 EXPECT_TRUE(policy->IsStorageProtected(GURL("https://bar.wildcards/")));
207 207
208 policy->RevokeRightsForExtension(protected_app); 208 policy->RevokeRightsForExtension(protected_app);
209 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://explicit/"))); 209 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://explicit/")));
210 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://foo.wildcards/"))); 210 EXPECT_FALSE(policy->IsStorageProtected(GURL("http://foo.wildcards/")));
211 EXPECT_FALSE(policy->IsStorageProtected(GURL("https://bar.wildcards/"))); 211 EXPECT_FALSE(policy->IsStorageProtected(GURL("https://bar.wildcards/")));
212 } 212 }
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698