1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
// Copyright 2017 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "extensions/browser/preload_check_test_util.h"
#include "base/check.h"
#include "base/functional/bind.h"
#include "base/functional/callback_helpers.h"
#include "base/run_loop.h"
#include "base/task/single_thread_task_runner.h"
#include "extensions/common/extension.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace extensions {
// PreloadCheckRunner:
PreloadCheckRunner::PreloadCheckRunner() : called_(false) {}
PreloadCheckRunner::~PreloadCheckRunner() = default;
void PreloadCheckRunner::Run(PreloadCheck* check) {
check->Start(GetCallback());
}
void PreloadCheckRunner::RunUntilComplete(PreloadCheck* check) {
Run(check);
ASSERT_FALSE(called_);
WaitForComplete();
ASSERT_TRUE(called_);
}
void PreloadCheckRunner::WaitForComplete() {
run_loop_ = std::make_unique<base::RunLoop>();
run_loop_->Run();
}
void PreloadCheckRunner::WaitForIdle() {
run_loop_ = std::make_unique<base::RunLoop>();
run_loop_->RunUntilIdle();
}
PreloadCheck::ResultCallback PreloadCheckRunner::GetCallback() {
return base::BindOnce(&PreloadCheckRunner::OnCheckComplete,
base::Unretained(this));
}
void PreloadCheckRunner::OnCheckComplete(const PreloadCheck::Errors& errors) {
ASSERT_FALSE(called_);
called_ = true;
errors_ = errors;
if (run_loop_) {
run_loop_->Quit();
}
}
// PreloadCheckStub:
PreloadCheckStub::PreloadCheckStub(const Errors& errors)
: PreloadCheck(nullptr), errors_(errors) {}
PreloadCheckStub::~PreloadCheckStub() = default;
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.
base::SingleThreadTaskRunner::GetCurrentDefault()->PostTask(
FROM_HERE,
base::BindOnce(&PreloadCheckStub::RunCallback,
weak_ptr_factory_.GetWeakPtr(), std::move(callback)));
} else {
std::move(callback).Run(errors_);
}
}
void PreloadCheckStub::RunCallback(ResultCallback callback) {
std::move(callback).Run(errors_);
}
} // namespace extensions
|
