chrome/browser/supervised_user/supervised_user_test_util.h

// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_SUPERVISED_USER_SUPERVISED_USER_TEST_UTIL_H_
#define CHROME_BROWSER_SUPERVISED_USER_SUPERVISED_USER_TEST_UTIL_H_

#include <memory>
#include <string>
#include <string_view>

#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/profiles/profile_key.h"
#include "chrome/browser/signin/identity_manager_factory.h"
#include "chrome/browser/supervised_user/android/supervised_user_service_platform_delegate.h"
#include "chrome/browser/supervised_user/supervised_user_service_factory.h"
#include "chrome/browser/supervised_user/supervised_user_settings_service_factory.h"
#include "chrome/browser/sync/sync_service_factory.h"
#include "components/keyed_service/core/keyed_service_factory.h"
#include "components/supervised_user/core/browser/kids_chrome_management_url_checker_client.h"
#include "components/supervised_user/core/common/supervised_user_constants.h"
#include "components/supervised_user/test_support/supervised_user_url_filter_test_utils.h"
#include "content/public/browser/storage_partition.h"

#if BUILDFLAG(IS_ANDROID)
#include "components/supervised_user/core/browser/android/content_filters_observer_bridge.h"
#endif  // BUILDFLAG(IS_ANDROID)

struct AccountInfo;

namespace supervised_user_test_util {

// Add custodians (e.g. parents) to |profile|, which must be a supervised user.
void AddCustodians(Profile* profile);

// Updates preferences relevant to requesting extensions permissions.
void SetSupervisedUserExtensionsMayRequestPermissionsPref(Profile* profile,
                                                          bool enabled);

// Updates preferences relevant to skipping parent approval for installing
// extensions.
void SetSkipParentApprovalToInstallExtensionsPref(Profile* profile,
                                                  bool enabled);

// Sets the Geolocation content setting value.
void SetSupervisedUserGeolocationEnabledContentSetting(Profile* profile,
                                                       bool enabled);

// Populates account info with a `given_name` and other fake data needed for a
// valid `AccountInfo` structure.
void PopulateAccountInfoWithName(AccountInfo& info,
                                 const std::string& given_name);

// Updates manual block/allow list with a given host.
// e.g. SetManualFilterForHost(profile, "www.example.com", false) adds the
// given host (i.e. "www.example.com") to the blocklist and the supervised user
// will not be able to access this host. Similarly
// SetManualFilterForHost(profile, "www.example.com", true) adds the host to the
// allowlist. The supervised user will be able to access this host.
void SetManualFilterForHost(Profile* profile,
                            std::string_view host,
                            bool allowlist);

// Updates manual block/allow list with a given url.
// e.g. SetManualFilterForUrl(profile, "http://www.example.com", false) adds the
// given url to the blocklist and the supervised user
// will not be able to access this url. Similarly
// SetManualFilterForUrl(profile, "www.example.com", true) adds the url to the
// allowlist. The supervised user will be able to access this url.
void SetManualFilterForUrl(Profile* profile,
                           std::string_view url,
                           bool allowlist);

// Convenience method for browser tests emulating parent changes to web
// filtering.
void SetWebFilterType(const Profile* profile,
                      supervised_user::WebFilterType web_filter_type);

// Handy utility to use with TestingProfile::TestingFactory, that attaches to
// the testing profile a supervised user service instance with substituted url
// filtering functionalities.
template <typename URLFilter,
          typename URLFilterDelegate = supervised_user::FakeURLFilterDelegate>
std::unique_ptr<KeyedService> BuildSupervisedUserService(
    content::BrowserContext* browser_context) {
  Profile* profile = Profile::FromBrowserContext(browser_context);
  std::unique_ptr<SupervisedUserServicePlatformDelegate> platform_delegate =
      std::make_unique<SupervisedUserServicePlatformDelegate>(*profile);
  signin::IdentityManager* identity_manager =
      IdentityManagerFactory::GetForProfile(profile);
  scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory =
      profile->GetDefaultStoragePartition()
          ->GetURLLoaderFactoryForBrowserProcess();
  return std::make_unique<supervised_user::SupervisedUserService>(
      IdentityManagerFactory::GetForProfile(profile),
      profile->GetDefaultStoragePartition()
          ->GetURLLoaderFactoryForBrowserProcess(),
      *profile->GetPrefs(),
      *SupervisedUserSettingsServiceFactory::GetInstance()->GetForKey(
          profile->GetProfileKey()),
      SyncServiceFactory::GetInstance()->GetForProfile(profile),
      std::make_unique<URLFilter>(
          *profile->GetPrefs(), std::make_unique<URLFilterDelegate>(),
          std::make_unique<
              supervised_user::KidsChromeManagementURLCheckerClient>(
              identity_manager, url_loader_factory, *profile->GetPrefs(),
              platform_delegate->GetCountryCode(),
              platform_delegate->GetChannel())),
      std::make_unique<SupervisedUserServicePlatformDelegate>(*profile)
#if BUILDFLAG(IS_ANDROID)
          ,
      base::BindRepeating(
          &supervised_user::ContentFiltersObserverBridge::Create)
#endif  // BUILDFLAG(IS_ANDROID)
  );
}

}  // namespace supervised_user_test_util

#endif  // CHROME_BROWSER_SUPERVISED_USER_SUPERVISED_USER_TEST_UTIL_H_