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
content / public / test / test_host_resolver.cc [blame]
// 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 "content/public/test/test_host_resolver.h"
#include <algorithm>
#include "base/threading/thread.h"
#include "net/base/net_errors.h"
#include "net/base/network_interfaces.h"
#include "net/base/url_util.h"
#include "net/dns/mock_host_resolver.h"
namespace content {
namespace {
// In many cases it may be not obvious that a test makes a real DNS lookup.
// We generally don't want to rely on external DNS servers for our tests,
// so this host resolver procedure catches external queries and returns a failed
// lookup result.
class LocalHostResolverProc : public net::HostResolverProc {
public:
LocalHostResolverProc() : HostResolverProc(nullptr) {}
int Resolve(const std::string& host,
net::AddressFamily address_family,
net::HostResolverFlags host_resolver_flags,
net::AddressList* addrlist,
int* os_error) override {
// To avoid depending on external resources and to reduce (if not preclude)
// network interactions from tests, we simulate failure for non-local DNS
// queries, rather than perform them.
// If you really need to make an external DNS query, use
// net::RuleBasedHostResolverProc and its AllowDirectLookup method.
if (host != net::GetHostName() && !net::IsLocalHostname(host)) {
DVLOG(1) << "To avoid external dependencies, simulating failure for "
"external DNS lookup of "
<< host;
return net::ERR_NAME_NOT_RESOLVED;
}
return ResolveUsingPrevious(host, address_family, host_resolver_flags,
addrlist, os_error);
}
private:
~LocalHostResolverProc() override {}
};
} // namespace
TestHostResolver::TestHostResolver()
: local_resolver_(new LocalHostResolverProc()),
rule_based_resolver_(new net::RuleBasedHostResolverProc(local_resolver_)),
scoped_local_host_resolver_proc_(
new net::ScopedDefaultHostResolverProc(rule_based_resolver_.get())) {
rule_based_resolver_->AddSimulatedFailure("wpad");
}
TestHostResolver::~TestHostResolver() {}
} // namespace content