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 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/web_applications/test/test_web_app_url_loader.h"
#include "base/containers/contains.h"
#include "base/functional/callback.h"
#include "base/task/sequenced_task_runner.h"
namespace web_app {
TestWebAppUrlLoader::TestWebAppUrlLoader() = default;
TestWebAppUrlLoader::~TestWebAppUrlLoader() = default;
void TestWebAppUrlLoader::SaveLoadUrlRequests() {
should_save_requests_ = true;
}
void TestWebAppUrlLoader::ProcessLoadUrlRequests() {
while (!pending_requests_.empty()) {
auto [url, callback] = std::move(pending_requests_.front());
pending_requests_.pop();
DCHECK(base::Contains(next_result_map_, url));
const UrlResponses& url_responses = next_result_map_[url];
DCHECK_EQ(1u, url_responses.results.size());
Result result = url_responses.results.front();
next_result_map_.erase(url);
std::move(callback).Run(result);
}
}
void TestWebAppUrlLoader::SetNextLoadUrlResult(const GURL& url, Result result) {
AddNextLoadUrlResults(url, {result});
}
void TestWebAppUrlLoader::AddNextLoadUrlResults(
const GURL& url,
const std::vector<Result>& results) {
DCHECK(!base::Contains(next_result_map_, url)) << url;
UrlResponses& responses = next_result_map_[url];
for (Result result : results)
responses.results.push(result);
}
void TestWebAppUrlLoader::LoadUrl(
content::NavigationController::LoadURLParams load_url_params,
content::WebContents* web_contents,
UrlComparison url_comparison,
ResultCallback callback) {
const GURL& url = load_url_params.url;
load_url_tracker_.Run(url, web_contents, url_comparison);
if (should_save_requests_) {
pending_requests_.emplace(url, std::move(callback));
return;
}
DCHECK(base::Contains(next_result_map_, url)) << url;
UrlResponses& responses = next_result_map_[url];
DCHECK(!responses.results.empty());
Result result = responses.results.front();
responses.results.pop();
if (responses.results.empty())
next_result_map_.erase(url);
base::SequencedTaskRunner::GetCurrentDefault()->PostTask(
FROM_HERE, base::BindOnce(std::move(callback), result));
}
TestWebAppUrlLoader::UrlResponses::UrlResponses() = default;
TestWebAppUrlLoader::UrlResponses::~UrlResponses() = default;
} // namespace web_app
|
