ios/chrome/test/testing_application_context.mm

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

#import "ios/chrome/test/testing_application_context.h"

#import "base/check_op.h"
#import "base/feature_list.h"
#import "base/memory/ptr_util.h"
#import "base/notreached.h"
#import "base/time/default_clock.h"
#import "base/time/default_tick_clock.h"
#import "components/application_locale_storage/application_locale_storage.h"
#import "components/network_time/network_time_tracker.h"
#import "components/os_crypt/async/browser/test_utils.h"
#import "components/variations/service/variations_service.h"
#import "ios/chrome/browser/download/model/auto_deletion/auto_deletion_service.h"
#import "ios/chrome/browser/optimization_guide/model/optimization_guide_global_state.h"
#import "ios/chrome/browser/policy/model/browser_policy_connector_ios.h"
#import "ios/chrome/browser/policy/model/configuration_policy_handler_list_factory.h"
#import "ios/chrome/browser/profile/model/ios_chrome_io_thread.h"
#import "ios/chrome/browser/promos_manager/model/features.h"
#import "ios/chrome/browser/promos_manager/model/mock_promos_manager.h"
#import "ios/chrome/browser/signin/model/account_profile_mapper.h"
#import "ios/chrome/browser/signin/model/avatar_provider.h"
#import "ios/components/security_interstitials/safe_browsing/fake_safe_browsing_service.h"
#import "ios/public/provider/chrome/browser/additional_features/additional_features_api.h"
#import "ios/public/provider/chrome/browser/push_notification/push_notification_api.h"
#import "ios/public/provider/chrome/browser/signin/signin_identity_api.h"
#import "ios/public/provider/chrome/browser/signin/signin_sso_api.h"
#import "net/url_request/url_request_context_getter.h"
#import "services/network/public/cpp/weak_wrapper_shared_url_loader_factory.h"
#import "services/network/test/test_network_connection_tracker.h"
#import "services/network/test/test_url_loader_factory.h"

TestingApplicationContext::TestingApplicationContext()
    : application_country_("us"),
      local_state_(nullptr),
      profile_manager_(nullptr),
      was_last_shutdown_clean_(false),
      test_network_connection_tracker_(
          network::TestNetworkConnectionTracker::CreateInstance()),
      variations_service_(nullptr),
      application_locale_storage_(
          std::make_unique<ApplicationLocaleStorage>()) {
  DCHECK(!GetApplicationContext());
  SetApplicationContext(this);
  application_locale_storage_->Set("en-US");
}

TestingApplicationContext::~TestingApplicationContext() {
  DCHECK_EQ(this, GetApplicationContext());
  DCHECK(!local_state_);
  SetApplicationContext(nullptr);
}

// static
TestingApplicationContext* TestingApplicationContext::GetGlobal() {
  return static_cast<TestingApplicationContext*>(GetApplicationContext());
}

void TestingApplicationContext::SetLocalState(PrefService* local_state) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  if (!local_state) {
    // The local state is owned outside of TestingApplicationContext, but
    // some of the members of TestingApplicationContext hold references to it.
    // Given our test infrastructure which tears down individual tests before
    // freeing the TestingApplicationContext, there's no good way to make the
    // local state outlive these dependencies. As a workaround, whenever
    // local state is cleared (assumedly as part of exiting the test) any
    // components owned by TestingApplicationContext that depends on the local
    // state are also freed.
    network_time_tracker_.reset();
    push_notification_service_.reset();
    optimization_guide_global_state_.reset();
  }
  local_state_ = local_state;
}

void TestingApplicationContext::SetLastShutdownClean(bool clean) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  was_last_shutdown_clean_ = clean;
}

void TestingApplicationContext::SetProfileManagerAndAccountProfileMapper(
    ProfileManagerIOS* manager,
    AccountProfileMapper* mapper) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  DCHECK(!default_account_profile_mapper_);
  DCHECK(!custom_account_profile_mapper_ || !mapper);
  DCHECK(!!manager == !!mapper);
  profile_manager_ = manager;
  custom_account_profile_mapper_ = mapper;
}

void TestingApplicationContext::SetVariationsService(
    variations::VariationsService* variations_service) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  variations_service_ = variations_service;
}

void TestingApplicationContext::SetSystemIdentityManager(
    std::unique_ptr<SystemIdentityManager> system_identity_manager) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  DCHECK(!system_identity_manager_);
  DCHECK(!default_account_profile_mapper_);
  system_identity_manager_ = std::move(system_identity_manager);
}

void TestingApplicationContext::SetIOSChromeIOThread(
    IOSChromeIOThread* ios_chrome_io_thread) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  ios_chrome_io_thread_ = ios_chrome_io_thread;
}

void TestingApplicationContext::SetSharedURLLoaderFactory(
    scoped_refptr<network::SharedURLLoaderFactory> url_loader_factory) {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  test_url_loader_factory_ = std::move(url_loader_factory);
}

void TestingApplicationContext::OnAppEnterForeground() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
}

void TestingApplicationContext::OnAppEnterBackground() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
}

void TestingApplicationContext::OnAppStartedBackgroundProcessing() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
}

void TestingApplicationContext::OnAppFinishedBackgroundProcessing() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
}

bool TestingApplicationContext::WasLastShutdownClean() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return was_last_shutdown_clean_;
}

PrefService* TestingApplicationContext::GetLocalState() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return local_state_;
}

net::URLRequestContextGetter*
TestingApplicationContext::GetSystemURLRequestContext() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return nullptr;
}

scoped_refptr<network::SharedURLLoaderFactory>
TestingApplicationContext::GetSharedURLLoaderFactory() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return test_url_loader_factory_;
}

network::mojom::NetworkContext*
TestingApplicationContext::GetSystemNetworkContext() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  NOTREACHED();
}

ApplicationLocaleStorage*
TestingApplicationContext::GetApplicationLocaleStorage() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  CHECK(application_locale_storage_);
  return application_locale_storage_.get();
}

const std::string& TestingApplicationContext::GetApplicationCountry() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  DCHECK(!application_country_.empty());
  return application_country_;
}

ProfileManagerIOS* TestingApplicationContext::GetProfileManager() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return profile_manager_;
}

metrics_services_manager::MetricsServicesManager*
TestingApplicationContext::GetMetricsServicesManager() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return nullptr;
}

metrics::MetricsService* TestingApplicationContext::GetMetricsService() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return nullptr;
}

signin::ActivePrimaryAccountsMetricsRecorder*
TestingApplicationContext::GetActivePrimaryAccountsMetricsRecorder() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return nullptr;
}

ukm::UkmRecorder* TestingApplicationContext::GetUkmRecorder() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return nullptr;
}

variations::VariationsService*
TestingApplicationContext::GetVariationsService() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return variations_service_;
}

net::NetLog* TestingApplicationContext::GetNetLog() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return nullptr;
}

net_log::NetExportFileWriter*
TestingApplicationContext::GetNetExportFileWriter() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return nullptr;
}

network_time::NetworkTimeTracker*
TestingApplicationContext::GetNetworkTimeTracker() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  if (!network_time_tracker_) {
    DCHECK(local_state_);
    network_time_tracker_.reset(new network_time::NetworkTimeTracker(
        base::WrapUnique(new base::DefaultClock),
        base::WrapUnique(new base::DefaultTickClock), local_state_, nullptr,
        std::nullopt));
  }
  return network_time_tracker_.get();
}

IOSChromeIOThread* TestingApplicationContext::GetIOSChromeIOThread() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return ios_chrome_io_thread_.get();
}

gcm::GCMDriver* TestingApplicationContext::GetGCMDriver() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return nullptr;
}

component_updater::ComponentUpdateService*
TestingApplicationContext::GetComponentUpdateService() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return nullptr;
}

SafeBrowsingService* TestingApplicationContext::GetSafeBrowsingService() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  if (!fake_safe_browsing_service_) {
    fake_safe_browsing_service_ =
        base::MakeRefCounted<FakeSafeBrowsingService>();
  }
  return fake_safe_browsing_service_.get();
}

network::NetworkConnectionTracker*
TestingApplicationContext::GetNetworkConnectionTracker() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return test_network_connection_tracker_.get();
}

BrowserPolicyConnectorIOS*
TestingApplicationContext::GetBrowserPolicyConnector() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);

  if (!browser_policy_connector_.get()) {
    browser_policy_connector_ = std::make_unique<BrowserPolicyConnectorIOS>(
        base::BindRepeating(&BuildPolicyHandlerList, true));
  }

  return browser_policy_connector_.get();
}

id<SingleSignOnService> TestingApplicationContext::GetSingleSignOnService() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  if (!single_sign_on_service_) {
    single_sign_on_service_ = ios::provider::CreateSSOService();
    DCHECK(single_sign_on_service_);
  }
  return single_sign_on_service_;
}

signin::AvatarProvider* TestingApplicationContext::GetIdentityAvatarProvider() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  if (!resized_avatar_caches_) {
    resized_avatar_caches_ = std::make_unique<signin::AvatarProvider>();
  }
  return resized_avatar_caches_.get();
}

SystemIdentityManager* TestingApplicationContext::GetSystemIdentityManager() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  if (!system_identity_manager_) {
    system_identity_manager_ =
        ios::provider::CreateSystemIdentityManager(GetSingleSignOnService());
  }
  return system_identity_manager_.get();
}

AccountProfileMapper* TestingApplicationContext::GetAccountProfileMapper() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  if (custom_account_profile_mapper_) {
    return custom_account_profile_mapper_;
  }
  if (!default_account_profile_mapper_) {
    default_account_profile_mapper_ = std::make_unique<AccountProfileMapper>(
        GetSystemIdentityManager(), /*profile_manager=*/nullptr,
        /*local_state=*/nullptr);
  }
  return default_account_profile_mapper_.get();
}

IncognitoSessionTracker*
TestingApplicationContext::GetIncognitoSessionTracker() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  return nullptr;
}

PushNotificationService*
TestingApplicationContext::GetPushNotificationService() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  if (!push_notification_service_) {
    push_notification_service_ = ios::provider::CreatePushNotificationService();
    DCHECK(push_notification_service_);
  }

  return push_notification_service_.get();
}

os_crypt_async::OSCryptAsync* TestingApplicationContext::GetOSCryptAsync() {
  if (!os_crypt_async_) {
    os_crypt_async_ = os_crypt_async::GetTestOSCryptAsyncForTesting();
  }
  return os_crypt_async_.get();
}

AdditionalFeaturesController*
TestingApplicationContext::GetAdditionalFeaturesController() {
  DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  if (!additional_features_controller_) {
    additional_features_controller_ =
        ios::provider::CreateAdditionalFeaturesController();
  }
  return additional_features_controller_.get();
}

auto_deletion::AutoDeletionService*
TestingApplicationContext::GetAutoDeletionService() {
  if (!auto_deletion_service_) {
    auto_deletion_service_ =
        std::make_unique<auto_deletion::AutoDeletionService>(GetLocalState());
  }
  return auto_deletion_service_.get();
}

optimization_guide::OptimizationGuideGlobalState*
TestingApplicationContext::GetOptimizationGuideGlobalState() {
  if (!optimization_guide_global_state_) {
    optimization_guide_global_state_ =
        std::make_unique<optimization_guide::OptimizationGuideGlobalState>();
  }
  return optimization_guide_global_state_.get();
}