Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 "chrome/browser/extensions/api/content_settings/content_settings_store. h" | 5 #include "chrome/browser/extensions/api/content_settings/content_settings_store. h" |
| 6 | 6 |
| 7 #include <stdint.h> | 7 #include <stdint.h> |
| 8 | 8 |
| 9 #include <memory> | 9 #include <memory> |
| 10 | 10 |
| 11 #include "base/memory/ptr_util.h" | |
| 12 #include "base/values.h" | |
| 13 #include "chrome/browser/extensions/api/content_settings/content_settings_api_co nstants.h" | |
| 14 #include "components/content_settings/core/browser/content_settings_registry.h" | |
| 11 #include "components/content_settings/core/browser/content_settings_rule.h" | 15 #include "components/content_settings/core/browser/content_settings_rule.h" |
| 12 #include "components/content_settings/core/browser/content_settings_utils.h" | 16 #include "components/content_settings/core/browser/content_settings_utils.h" |
| 13 #include "components/content_settings/core/test/content_settings_test_utils.h" | 17 #include "components/content_settings/core/test/content_settings_test_utils.h" |
| 14 #include "testing/gmock/include/gmock/gmock.h" | 18 #include "testing/gmock/include/gmock/gmock.h" |
| 15 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
| 16 #include "url/gurl.h" | 20 #include "url/gurl.h" |
| 17 | 21 |
| 18 using ::testing::Mock; | 22 using ::testing::Mock; |
| 19 | 23 |
| 20 namespace extensions { | 24 namespace extensions { |
| 21 | 25 |
| 26 namespace keys = content_settings_api_constants; | |
| 27 | |
| 22 namespace { | 28 namespace { |
| 23 | 29 |
| 24 void CheckRule(const content_settings::Rule& rule, | 30 void CheckRule(const content_settings::Rule& rule, |
| 25 const ContentSettingsPattern& primary_pattern, | 31 const ContentSettingsPattern& primary_pattern, |
| 26 const ContentSettingsPattern& secondary_pattern, | 32 const ContentSettingsPattern& secondary_pattern, |
| 27 ContentSetting setting) { | 33 ContentSetting setting) { |
| 28 EXPECT_EQ(primary_pattern.ToString(), rule.primary_pattern.ToString()); | 34 EXPECT_EQ(primary_pattern.ToString(), rule.primary_pattern.ToString()); |
| 29 EXPECT_EQ(secondary_pattern.ToString(), rule.secondary_pattern.ToString()); | 35 EXPECT_EQ(secondary_pattern.ToString(), rule.secondary_pattern.ToString()); |
| 30 EXPECT_EQ(setting, content_settings::ValueToContentSetting(rule.value.get())); | 36 EXPECT_EQ(setting, content_settings::ValueToContentSetting(rule.value.get())); |
| 31 } | 37 } |
| (...skipping 216 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... | |
| 248 CheckRule(rules[0], pattern_2, pattern_2, CONTENT_SETTING_BLOCK); | 254 CheckRule(rules[0], pattern_2, pattern_2, CONTENT_SETTING_BLOCK); |
| 249 | 255 |
| 250 // Uninstall second extension. | 256 // Uninstall second extension. |
| 251 store()->UnregisterExtension(ext_id_2); | 257 store()->UnregisterExtension(ext_id_2); |
| 252 | 258 |
| 253 rules = GetSettingsForOneTypeFromStore(store(), CONTENT_SETTINGS_TYPE_COOKIES, | 259 rules = GetSettingsForOneTypeFromStore(store(), CONTENT_SETTINGS_TYPE_COOKIES, |
| 254 std::string(), incognito); | 260 std::string(), incognito); |
| 255 ASSERT_EQ(0u, rules.size()); | 261 ASSERT_EQ(0u, rules.size()); |
| 256 } | 262 } |
| 257 | 263 |
| 264 TEST_F(ContentSettingsStoreTest, SetFromList) { | |
| 265 // Force creation of ContentSettingsRegistry, so that the string to content | |
| 266 // setting type lookup can succeed. | |
| 267 content_settings::ContentSettingsRegistry::GetInstance(); | |
| 268 | |
| 269 ::testing::StrictMock<MockContentSettingsStoreObserver> observer; | |
| 270 store()->AddObserver(&observer); | |
| 271 | |
| 272 GURL url("http://www.youtube.com"); | |
| 273 | |
| 274 EXPECT_EQ(CONTENT_SETTING_DEFAULT, | |
| 275 GetContentSettingFromStore(store(), | |
| 276 url, | |
| 277 url, | |
| 278 CONTENT_SETTINGS_TYPE_COOKIES, | |
| 279 std::string(), | |
| 280 false)); | |
| 281 | |
| 282 // Register first extension | |
| 283 std::string ext_id("my_extension"); | |
| 284 RegisterExtension(ext_id); | |
| 285 | |
| 286 // Set setting via a list | |
| 287 ContentSettingsPattern pattern = | |
| 288 ContentSettingsPattern::FromURL(GURL("http://www.youtube.com")); | |
| 289 EXPECT_CALL(observer, OnContentSettingChanged(ext_id, false)); | |
| 290 | |
| 291 // Build a preference list in JSON format. | |
| 292 base::ListValue pref_list; | |
| 293 // {"primaryPattern": pattern, "secondaryPattern": pattern, "type": "cookies", | |
| 294 // "setting": "allow"} | |
| 295 auto dict_value = base::MakeUnique<base::DictionaryValue>(); | |
| 296 dict_value->SetString(keys::kPrimaryPatternKey, pattern.ToString()); | |
| 297 dict_value->SetString(keys::kSecondaryPatternKey, pattern.ToString()); | |
| 298 dict_value->SetString(keys::kContentSettingsTypeKey, "cookies"); | |
| 299 dict_value->SetString(keys::kContentSettingKey, "allow"); | |
| 300 pref_list.Append(std::move(dict_value)); | |
| 301 // Test content settings types that have been removed. Should be ignored. | |
| 302 // {"primaryPattern": pattern, "secondaryPattern": pattern, | |
| 303 // "type": "fullscreen", "setting": "allow"} | |
| 304 dict_value = base::MakeUnique<base::DictionaryValue>(); | |
| 305 dict_value->SetString(keys::kPrimaryPatternKey, pattern.ToString()); | |
| 306 dict_value->SetString(keys::kSecondaryPatternKey, pattern.ToString()); | |
| 307 dict_value->SetString(keys::kContentSettingsTypeKey, "fullscreen"); | |
| 308 dict_value->SetString(keys::kContentSettingKey, "allow"); | |
| 309 pref_list.Append(std::move(dict_value)); | |
| 310 // {"primaryPattern": pattern, "secondaryPattern": pattern, | |
| 311 // "type": "mouselock", "setting": "allow"} | |
| 312 dict_value = base::MakeUnique<base::DictionaryValue>(); | |
| 313 dict_value->SetString(keys::kPrimaryPatternKey, pattern.ToString()); | |
| 314 dict_value->SetString(keys::kSecondaryPatternKey, pattern.ToString()); | |
| 315 dict_value->SetString(keys::kContentSettingsTypeKey, "mouselock"); | |
| 316 dict_value->SetString(keys::kContentSettingKey, "allow"); | |
| 317 pref_list.Append(std::move(dict_value)); | |
| 318 | |
| 319 store()->SetExtensionContentSettingFromList(ext_id, &pref_list, | |
| 320 kExtensionPrefsScopeRegular); | |
| 321 Mock::VerifyAndClear(&observer); | |
| 322 | |
| 323 EXPECT_EQ(CONTENT_SETTING_ALLOW, | |
| 324 GetContentSettingFromStore(store(), | |
| 325 url, | |
| 326 url, | |
| 327 CONTENT_SETTINGS_TYPE_COOKIES, | |
| 328 std::string(), | |
| 329 false)); | |
| 330 // Fullscreen and mouselock should remain unchanged. | |
|
benwells
2016/10/21 05:19:04
Nit: Would it be clearer to say "not be stored" in
Matt Giuca
2016/10/21 05:21:43
This getter isn't directly getting the data that w
Matt Giuca
2016/10/21 05:31:32
Done.
| |
| 331 EXPECT_EQ(CONTENT_SETTING_DEFAULT, | |
| 332 GetContentSettingFromStore(store(), | |
| 333 url, | |
| 334 url, | |
| 335 CONTENT_SETTINGS_TYPE_FULLSCREEN, | |
| 336 std::string(), | |
| 337 false)); | |
| 338 EXPECT_EQ(CONTENT_SETTING_DEFAULT, | |
| 339 GetContentSettingFromStore(store(), | |
| 340 url, | |
| 341 url, | |
| 342 CONTENT_SETTINGS_TYPE_MOUSELOCK, | |
| 343 std::string(), | |
| 344 false)); | |
| 345 | |
| 346 store()->RemoveObserver(&observer); | |
| 347 } | |
| 348 | |
| 258 } // namespace extensions | 349 } // namespace extensions |
| OLD | NEW |