| Index: extensions/browser/preload_check_test_util.cc
|
| diff --git a/extensions/browser/preload_check_test_util.cc b/extensions/browser/preload_check_test_util.cc
|
| index f5becf43ae4a7173c854c744f33c495fd2992565..482236fbed4e03526fdc3f87a298756c94edd676 100644
|
| --- a/extensions/browser/preload_check_test_util.cc
|
| +++ b/extensions/browser/preload_check_test_util.cc
|
| @@ -20,14 +20,6 @@ namespace extensions {
|
| PreloadCheckRunner::PreloadCheckRunner() : called_(false) {}
|
| PreloadCheckRunner::~PreloadCheckRunner() {}
|
|
|
| -void PreloadCheckRunner::OnCheckComplete(PreloadCheck::Errors errors) {
|
| - ASSERT_FALSE(called_);
|
| - called_ = true;
|
| - errors_ = errors;
|
| - if (run_loop_)
|
| - run_loop_->Quit();
|
| -}
|
| -
|
| void PreloadCheckRunner::Run(PreloadCheck* check) {
|
| check->Start(GetCallback());
|
| }
|
| @@ -36,9 +28,18 @@ void PreloadCheckRunner::RunUntilComplete(PreloadCheck* check) {
|
| Run(check);
|
| ASSERT_FALSE(called_);
|
|
|
| + WaitForComplete();
|
| + ASSERT_TRUE(called_);
|
| +}
|
| +
|
| +void PreloadCheckRunner::WaitForComplete() {
|
| run_loop_ = base::MakeUnique<base::RunLoop>();
|
| run_loop_->Run();
|
| - ASSERT_TRUE(called_);
|
| +}
|
| +
|
| +void PreloadCheckRunner::WaitForIdle() {
|
| + run_loop_ = base::MakeUnique<base::RunLoop>();
|
| + run_loop_->RunUntilIdle();
|
| }
|
|
|
| PreloadCheck::ResultCallback PreloadCheckRunner::GetCallback() {
|
| @@ -46,23 +47,24 @@ PreloadCheck::ResultCallback PreloadCheckRunner::GetCallback() {
|
| base::Unretained(this));
|
| }
|
|
|
| -void PreloadCheckRunner::WaitForIdle() {
|
| - run_loop_ = base::MakeUnique<base::RunLoop>();
|
| - run_loop_->RunUntilIdle();
|
| +void PreloadCheckRunner::OnCheckComplete(PreloadCheck::Errors errors) {
|
| + ASSERT_FALSE(called_);
|
| + called_ = true;
|
| + errors_ = errors;
|
| +
|
| + if (run_loop_)
|
| + run_loop_->Quit();
|
| }
|
|
|
| // PreloadCheckStub:
|
| -PreloadCheckStub::PreloadCheckStub()
|
| - : PreloadCheck(nullptr), is_async_(false), weak_ptr_factory_(this) {}
|
| +PreloadCheckStub::PreloadCheckStub(const Errors& errors)
|
| + : PreloadCheck(nullptr), errors_(errors), weak_ptr_factory_(this) {}
|
|
|
| PreloadCheckStub::~PreloadCheckStub() {}
|
|
|
| -void PreloadCheckStub::AddError(Error error) {
|
| - errors_.insert(error);
|
| -}
|
| -
|
| void PreloadCheckStub::Start(ResultCallback callback) {
|
| DCHECK(!callback.is_null());
|
| + started_ = true;
|
| if (is_async_) {
|
| // TODO(michaelpg): Bind the callback directly and remove RunCallback
|
| // once crbug.com/704027 is addressed.
|
| @@ -79,8 +81,4 @@ void PreloadCheckStub::RunCallback(ResultCallback callback) {
|
| std::move(callback).Run(errors_);
|
| }
|
|
|
| -base::string16 PreloadCheckStub::GetErrorMessage() const {
|
| - return message_;
|
| -}
|
| -
|
| } // namespace extensions
|
|
|