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
|
// Copyright 2023 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/test/supervised_user/embedded_test_server_setup_mixin.h"
#include <algorithm>
#include <string>
#include <string_view>
#include <vector>
#include "base/containers/span.h"
#include "base/logging.h"
#include "base/strings/strcat.h"
#include "base/strings/string_split.h"
#include "base/strings/string_util.h"
#include "chrome/test/base/mixin_based_in_process_browser_test.h"
#include "chrome/test/supervised_user/child_account_test_utils.h"
#include "content/public/test/browser_test_utils.h"
#include "net/dns/mock_host_resolver.h"
#include "net/test/embedded_test_server/embedded_test_server.h"
#include "services/network/public/cpp/network_switches.h"
namespace supervised_user {
namespace {
std::vector<std::string> SplitHostList(std::string_view host_list) {
return base::SplitString(host_list, ",", base::TRIM_WHITESPACE,
base::SPLIT_WANT_NONEMPTY);
}
} // namespace
EmbeddedTestServerSetupMixin::EmbeddedTestServerSetupMixin(
InProcessBrowserTestMixinHost& host,
InProcessBrowserTest* test_base,
raw_ptr<net::EmbeddedTestServer> server,
const Options& options)
: InProcessBrowserTestMixin(&host),
test_base_(test_base),
embedded_test_server_(server),
resolver_rules_map_host_list_(
SplitHostList(options.resolver_rules_map_host_list)) {
CHECK(server) << "This mixin requires an embedded test server";
}
EmbeddedTestServerSetupMixin::~EmbeddedTestServerSetupMixin() = default;
void EmbeddedTestServerSetupMixin::SetUp() {
CHECK(embedded_test_server_->InitializeAndListen());
}
void EmbeddedTestServerSetupMixin::SetUpCommandLine(
base::CommandLine* command_line) {
std::ranges::for_each(
resolver_rules_map_host_list_, [&](const std::string& host) {
AddHostResolverRule(command_line, host, *embedded_test_server_);
});
LOG(INFO) << "Embedded test server is listening on "
<< embedded_test_server_->host_port_pair().ToString() << ".";
LOG(INFO) << "Following hosts will be mapped to it: ";
for (const std::string& host_pattern : resolver_rules_map_host_list_) {
LOG(INFO) << "\t" << host_pattern;
}
}
void EmbeddedTestServerSetupMixin::SetUpOnMainThread() {
embedded_test_server_->ServeFilesFromSourceDirectory("chrome/test/data");
embedded_test_server_->StartAcceptingConnections();
}
void EmbeddedTestServerSetupMixin::TearDownOnMainThread() {
CHECK(embedded_test_server_->ShutdownAndWaitUntilComplete());
}
} // namespace supervised_user
|
