Terry Mancey
170d7b5f8b
Adds IsIneligibleToStart and IsInitialized accessors to AdsService along with an OnAdsServiceIneligibleToStart observer notification. AdsServiceWaiter gains wait helpers that check whether a lifecycle event has already fired before blocking, so tests can safely observe startup and shutdown regardless of timing.
62 lines
1.9 KiB
C++
62 lines
1.9 KiB
C++
/* Copyright (c) 2024 The Brave Authors. All rights reserved.
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
* You can obtain one at https://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "brave/browser/brave_ads/ads_service_waiter.h"
|
|
|
|
#include "brave/components/brave_ads/core/browser/service/ads_service.h"
|
|
|
|
namespace brave_ads::test {
|
|
|
|
AdsServiceWaiter::AdsServiceWaiter(AdsService& ads_service)
|
|
: ads_service_(ads_service) {
|
|
observation_.Observe(&ads_service);
|
|
}
|
|
|
|
AdsServiceWaiter::~AdsServiceWaiter() = default;
|
|
|
|
void AdsServiceWaiter::WaitForOnAdsServiceIneligibleToStart() {
|
|
// `OnAdsServiceIneligibleToStart` will not fire again if the service was
|
|
// marked ineligible before observation began.
|
|
if (!ads_service_->IsIneligibleToStart()) {
|
|
on_ads_service_ineligible_to_start_run_loop_.Run();
|
|
}
|
|
}
|
|
|
|
void AdsServiceWaiter::WaitForOnDidInitializeAdsService() {
|
|
// `OnDidInitializeAdsService` will not fire again if the service was already
|
|
// initialized before observation began.
|
|
if (!ads_service_->IsInitialized()) {
|
|
on_did_initialize_ads_service_run_loop_.Run();
|
|
}
|
|
}
|
|
|
|
void AdsServiceWaiter::WaitForOnDidShutdownAdsService() {
|
|
on_did_shutdown_ads_service_run_loop_.Run();
|
|
}
|
|
|
|
void AdsServiceWaiter::WaitForOnDidClearAdsServiceData() {
|
|
on_did_clear_ads_service_data_run_loop_.Run();
|
|
}
|
|
|
|
///////////////////////////////////////////////////////////////////////////////
|
|
|
|
void AdsServiceWaiter::OnAdsServiceIneligibleToStart() {
|
|
on_ads_service_ineligible_to_start_run_loop_.Quit();
|
|
}
|
|
|
|
void AdsServiceWaiter::OnDidInitializeAdsService() {
|
|
on_did_initialize_ads_service_run_loop_.Quit();
|
|
}
|
|
|
|
void AdsServiceWaiter::OnDidShutdownAdsService() {
|
|
on_did_shutdown_ads_service_run_loop_.Quit();
|
|
}
|
|
|
|
void AdsServiceWaiter::OnDidClearAdsServiceData() {
|
|
on_did_clear_ads_service_data_run_loop_.Quit();
|
|
}
|
|
|
|
} // namespace brave_ads::test
|