privacy_sandbox_test_util.cc 4.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293
  1. // Copyright 2021 The Chromium Authors. All rights reserved.
  2. // Use of this source code is governed by a BSD-style license that can be
  3. // found in the LICENSE file.
  4. #include "components/privacy_sandbox/privacy_sandbox_test_util.h"
  5. #include "base/feature_list.h"
  6. #include "components/content_settings/core/browser/cookie_settings.h"
  7. #include "components/content_settings/core/browser/host_content_settings_map.h"
  8. #include "components/content_settings/core/common/pref_names.h"
  9. #include "components/content_settings/core/test/content_settings_mock_provider.h"
  10. #include "components/content_settings/core/test/content_settings_test_utils.h"
  11. #include "components/privacy_sandbox/privacy_sandbox_features.h"
  12. #include "components/privacy_sandbox/privacy_sandbox_prefs.h"
  13. #include "components/sync_preferences/testing_pref_service_syncable.h"
  14. namespace privacy_sandbox_test_util {
  15. MockPrivacySandboxObserver::MockPrivacySandboxObserver() = default;
  16. MockPrivacySandboxObserver::~MockPrivacySandboxObserver() = default;
  17. MockPrivacySandboxSettingsDelegate::MockPrivacySandboxSettingsDelegate() =
  18. default;
  19. MockPrivacySandboxSettingsDelegate::~MockPrivacySandboxSettingsDelegate() =
  20. default;
  21. void SetupTestState(
  22. sync_preferences::TestingPrefServiceSyncable* testing_pref_service,
  23. HostContentSettingsMap* map,
  24. bool privacy_sandbox_enabled,
  25. bool block_third_party_cookies,
  26. ContentSetting default_cookie_setting,
  27. const std::vector<CookieContentSettingException>& user_cookie_exceptions,
  28. ContentSetting managed_cookie_setting,
  29. const std::vector<CookieContentSettingException>&
  30. managed_cookie_exceptions) {
  31. // Setup block-third-party-cookies settings.
  32. testing_pref_service->SetUserPref(
  33. prefs::kCookieControlsMode,
  34. std::make_unique<base::Value>(static_cast<int>(
  35. block_third_party_cookies
  36. ? content_settings::CookieControlsMode::kBlockThirdParty
  37. : content_settings::CookieControlsMode::kOff)));
  38. // Setup cookie content settings.
  39. auto user_provider = std::make_unique<content_settings::MockProvider>();
  40. auto managed_provider = std::make_unique<content_settings::MockProvider>();
  41. if (default_cookie_setting != kNoSetting) {
  42. user_provider->SetWebsiteSetting(
  43. ContentSettingsPattern::Wildcard(), ContentSettingsPattern::Wildcard(),
  44. ContentSettingsType::COOKIES, base::Value(default_cookie_setting));
  45. }
  46. for (const auto& exception : user_cookie_exceptions) {
  47. user_provider->SetWebsiteSetting(
  48. ContentSettingsPattern::FromString(exception.primary_pattern),
  49. ContentSettingsPattern::FromString(exception.secondary_pattern),
  50. ContentSettingsType::COOKIES, base::Value(exception.content_setting));
  51. }
  52. if (managed_cookie_setting != kNoSetting) {
  53. managed_provider->SetWebsiteSetting(
  54. ContentSettingsPattern::Wildcard(), ContentSettingsPattern::Wildcard(),
  55. ContentSettingsType::COOKIES, base::Value(managed_cookie_setting));
  56. }
  57. for (const auto& exception : managed_cookie_exceptions) {
  58. managed_provider->SetWebsiteSetting(
  59. ContentSettingsPattern::FromString(exception.primary_pattern),
  60. ContentSettingsPattern::FromString(exception.secondary_pattern),
  61. ContentSettingsType::COOKIES, base::Value(exception.content_setting));
  62. }
  63. content_settings::TestUtils::OverrideProvider(
  64. map, std::move(user_provider), HostContentSettingsMap::DEFAULT_PROVIDER);
  65. content_settings::TestUtils::OverrideProvider(
  66. map, std::move(managed_provider),
  67. HostContentSettingsMap::POLICY_PROVIDER);
  68. // Only adjust the Privacy Sandbox preference which should be being consulted
  69. // based on feature state.
  70. if (base::FeatureList::IsEnabled(privacy_sandbox::kPrivacySandboxSettings3)) {
  71. testing_pref_service->SetUserPref(
  72. prefs::kPrivacySandboxApisEnabledV2,
  73. std::make_unique<base::Value>(privacy_sandbox_enabled));
  74. } else {
  75. testing_pref_service->SetUserPref(
  76. prefs::kPrivacySandboxApisEnabled,
  77. std::make_unique<base::Value>(privacy_sandbox_enabled));
  78. }
  79. }
  80. } // namespace privacy_sandbox_test_util