Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(747)

Side by Side Diff: net/nqe/network_quality_estimator_util_unittest.cc

Issue 2936823002: NQE: Exclude network observations from private networks (Closed)
Patch Set: bengr, mmenke comments Created 3 years, 6 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 // Copyright 2017 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "net/nqe/network_quality_estimator_util.h"
6
7 #include <memory>
8
9 #include "base/memory/ptr_util.h"
10 #include "net/base/net_errors.h"
11 #include "net/base/test_completion_callback.h"
12 #include "net/dns/mock_host_resolver.h"
13 #include "net/log/test_net_log.h"
14 #include "testing/gtest/include/gtest/gtest.h"
15 #include "url/gurl.h"
16
17 namespace net {
18
19 namespace nqe {
20
21 namespace internal {
22
23 namespace {
24
25 // Verify that the cached network qualities from the prefs are not used if the
26 // reading of the network quality prefs is not enabled..
27 TEST(NetworkQualityEstimatorUtilTest, ReservedHost) {
28 std::unique_ptr<BoundTestNetLog> net_log =
29 base::MakeUnique<BoundTestNetLog>();
30 BoundTestNetLog* net_log_ptr = net_log.get();
31
32 MockCachingHostResolver mock_host_resolver;
33
34 scoped_refptr<net::RuleBasedHostResolverProc> rules(
35 new net::RuleBasedHostResolverProc(nullptr));
36
37 // example1.com resolves to a private IP address.
38 rules->AddRule("example1.com", "127.0.0.3");
39
40 // example2.com resolves to a public IP address.
41 rules->AddRule("example2.com", "27.0.0.3");
42
43 mock_host_resolver.set_rules(rules.get());
44
45 {
46 // Resolve example1.com so that the resolution entry is cached.
47 TestCompletionCallback callback;
48 std::unique_ptr<HostResolver::Request> request;
49 AddressList ignored;
50 int rv = mock_host_resolver.Resolve(
51 HostResolver::RequestInfo(HostPortPair("example1.com", 443)),
52 DEFAULT_PRIORITY, &ignored, callback.callback(), &request,
53 NetLogWithSource());
54 EXPECT_EQ(ERR_IO_PENDING, rv);
55 EXPECT_EQ(OK, callback.WaitForResult());
56 }
57
58 {
59 // Resolve example2.com so that the resolution entry is cached.
60 TestCompletionCallback callback;
61 std::unique_ptr<HostResolver::Request> request;
62 AddressList ignored;
63 int rv = mock_host_resolver.Resolve(
64 HostResolver::RequestInfo(HostPortPair("example2.com", 443)),
65 DEFAULT_PRIORITY, &ignored, callback.callback(), &request,
66 NetLogWithSource());
67 EXPECT_EQ(ERR_IO_PENDING, rv);
68 EXPECT_EQ(OK, callback.WaitForResult());
69 }
70
71 EXPECT_EQ(2u, mock_host_resolver.num_resolve());
72
73 EXPECT_TRUE(IsPrivateHost(&mock_host_resolver, HostPortPair("localhost", 443),
74 net_log_ptr->bound()));
mmenke 2017/06/16 15:03:58 Also check localhost6?
tbansal1 2017/06/17 01:54:01 Done.
75 EXPECT_TRUE(IsPrivateHost(&mock_host_resolver, HostPortPair("127.0.0.1", 80),
76 net_log_ptr->bound()));
mmenke 2017/06/16 15:03:58 Also suggest 0.0.0.0 (It's used as an alias for 12
tbansal1 2017/06/17 01:54:01 Done.
77
78 EXPECT_TRUE(IsPrivateHost(&mock_host_resolver,
79 HostPortPair("192.168.0.1", 443),
80 net_log_ptr->bound()));
81 EXPECT_EQ(0u, mock_host_resolver.num_resolve_from_cache());
82
83 EXPECT_FALSE(IsPrivateHost(&mock_host_resolver,
84 HostPortPair("92.168.0.1", 443),
85 net_log_ptr->bound()));
86 EXPECT_EQ(1u, mock_host_resolver.num_resolve_from_cache());
mmenke 2017/06/16 15:03:58 Suggest checking some IPv6 hosts (::1, sure there
tbansal1 2017/06/17 01:54:01 Done.
87
88 EXPECT_TRUE(IsPrivateHost(&mock_host_resolver,
89 HostPortPair("example1.com", 443),
90 net_log_ptr->bound()));
91 EXPECT_EQ(2u, mock_host_resolver.num_resolve_from_cache());
92
93 EXPECT_FALSE(IsPrivateHost(&mock_host_resolver,
94 HostPortPair("example2.com", 443),
95 net_log_ptr->bound()));
mmenke 2017/06/16 15:03:58 Test with a hostname that's not cached, so Resolve
tbansal1 2017/06/17 01:54:01 Done.
96 EXPECT_EQ(3u, mock_host_resolver.num_resolve_from_cache());
97
98 // IsPrivateHost() should have queried only the resolver's cache.
99 EXPECT_EQ(2u, mock_host_resolver.num_resolve());
100 }
101
102 } // namespace
103
104 } // namespace internal
105
106 } // namespace nqe
107
108 } // namespace net
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698