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

Side by Side Diff: components/domain_reliability/monitor_unittest.cc

Issue 615313006: Domain Reliability: Get correct reporting pref once, on startup (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Fix ordering on SetDiscardUploads Created 6 years, 2 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
1 // Copyright 2014 The Chromium Authors. All rights reserved. 1 // Copyright 2014 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include "components/domain_reliability/monitor.h" 5 #include "components/domain_reliability/monitor.h"
6 6
7 #include <map> 7 #include <map>
8 #include <string> 8 #include <string>
9 #include <vector> 9 #include <vector>
10 10
11 #include "base/bind.h" 11 #include "base/bind.h"
12 #include "base/memory/scoped_ptr.h" 12 #include "base/memory/scoped_ptr.h"
13 #include "base/message_loop/message_loop_proxy.h" 13 #include "base/message_loop/message_loop_proxy.h"
14 #include "base/prefs/pref_registry_simple.h"
15 #include "base/prefs/testing_pref_service.h"
16 #include "base/test/test_simple_task_runner.h" 14 #include "base/test/test_simple_task_runner.h"
17 #include "components/domain_reliability/baked_in_configs.h" 15 #include "components/domain_reliability/baked_in_configs.h"
18 #include "components/domain_reliability/beacon.h" 16 #include "components/domain_reliability/beacon.h"
19 #include "components/domain_reliability/config.h" 17 #include "components/domain_reliability/config.h"
20 #include "components/domain_reliability/test_util.h" 18 #include "components/domain_reliability/test_util.h"
21 #include "net/base/host_port_pair.h" 19 #include "net/base/host_port_pair.h"
22 #include "net/base/load_flags.h" 20 #include "net/base/load_flags.h"
23 #include "net/http/http_response_headers.h" 21 #include "net/http/http_response_headers.h"
24 #include "net/http/http_util.h" 22 #include "net/http/http_util.h"
25 #include "net/url_request/url_request_context_getter.h" 23 #include "net/url_request/url_request_context_getter.h"
26 #include "net/url_request/url_request_status.h" 24 #include "net/url_request/url_request_status.h"
27 #include "net/url_request/url_request_test_util.h" 25 #include "net/url_request/url_request_test_util.h"
28 #include "testing/gtest/include/gtest/gtest.h" 26 #include "testing/gtest/include/gtest/gtest.h"
29 27
30 namespace domain_reliability { 28 namespace domain_reliability {
31 29
32 namespace { 30 namespace {
33 31
34 typedef std::vector<DomainReliabilityBeacon> BeaconVector; 32 typedef std::vector<DomainReliabilityBeacon> BeaconVector;
35 33
36 static const size_t kAlwaysReportIndex = 0u; 34 static const size_t kAlwaysReportIndex = 0u;
37 static const size_t kNeverReportIndex = 1u; 35 static const size_t kNeverReportIndex = 1u;
38 36
39 static const char* kPrefName = "reporting_enabled";
40
41 scoped_refptr<net::HttpResponseHeaders> MakeHttpResponseHeaders( 37 scoped_refptr<net::HttpResponseHeaders> MakeHttpResponseHeaders(
42 const std::string& headers) { 38 const std::string& headers) {
43 return scoped_refptr<net::HttpResponseHeaders>( 39 return scoped_refptr<net::HttpResponseHeaders>(
44 new net::HttpResponseHeaders(net::HttpUtil::AssembleRawHeaders( 40 new net::HttpResponseHeaders(net::HttpUtil::AssembleRawHeaders(
45 headers.c_str(), headers.length()))); 41 headers.c_str(), headers.length())));
46 } 42 }
47 43
48 } // namespace 44 } // namespace
49 45
50 class DomainReliabilityMonitorTest : public testing::Test { 46 class DomainReliabilityMonitorTest : public testing::Test {
51 protected: 47 protected:
52 typedef DomainReliabilityMonitor::RequestInfo RequestInfo; 48 typedef DomainReliabilityMonitor::RequestInfo RequestInfo;
53 49
54 DomainReliabilityMonitorTest() 50 DomainReliabilityMonitorTest()
55 : pref_task_runner_(new base::TestSimpleTaskRunner()), 51 : pref_task_runner_(new base::TestSimpleTaskRunner()),
56 network_task_runner_(new base::TestSimpleTaskRunner()), 52 network_task_runner_(new base::TestSimpleTaskRunner()),
57 url_request_context_getter_( 53 url_request_context_getter_(
58 new net::TestURLRequestContextGetter(network_task_runner_)), 54 new net::TestURLRequestContextGetter(network_task_runner_)),
59 time_(new MockTime()), 55 time_(new MockTime()),
60 pref_service_(CreatePrefService()),
61 monitor_("test-reporter", 56 monitor_("test-reporter",
62 pref_task_runner_, 57 pref_task_runner_,
63 network_task_runner_, 58 network_task_runner_,
64 pref_service_.get(),
65 kPrefName,
66 scoped_ptr<MockableTime>(time_)), 59 scoped_ptr<MockableTime>(time_)),
67 context_(NULL) { 60 context_(NULL) {
68 monitor_.MoveToNetworkThread(); 61 monitor_.MoveToNetworkThread();
69 monitor_.InitURLRequestContext(url_request_context_getter_); 62 monitor_.InitURLRequestContext(url_request_context_getter_);
davidben 2014/10/03 00:12:25 Shouldn't this also call SetDiscardUploads? It loo
Deprecated (see juliatuttle) 2014/10/03 00:42:09 Done.
70 context_ = monitor_.AddContextForTesting(MakeTestConfig()); 63 context_ = monitor_.AddContextForTesting(MakeTestConfig());
71 } 64 }
72 65
73 static PrefService* CreatePrefService() {
74 TestingPrefServiceSimple* prefs = new TestingPrefServiceSimple();
75 prefs->registry()->RegisterBooleanPref(kPrefName, false);
76 prefs->SetUserPref(kPrefName, new base::FundamentalValue(true));
77 return prefs;
78 }
79
80 static RequestInfo MakeRequestInfo() { 66 static RequestInfo MakeRequestInfo() {
81 RequestInfo request; 67 RequestInfo request;
82 request.status = net::URLRequestStatus(); 68 request.status = net::URLRequestStatus();
83 request.status.set_status(net::URLRequestStatus::SUCCESS); 69 request.status.set_status(net::URLRequestStatus::SUCCESS);
84 request.status.set_error(net::OK); 70 request.status.set_error(net::OK);
85 request.response_info.socket_address = 71 request.response_info.socket_address =
86 net::HostPortPair::FromString("12.34.56.78:80"); 72 net::HostPortPair::FromString("12.34.56.78:80");
87 request.response_info.headers = MakeHttpResponseHeaders( 73 request.response_info.headers = MakeHttpResponseHeaders(
88 "HTTP/1.1 200 OK\n\n"); 74 "HTTP/1.1 200 OK\n\n");
89 request.response_info.network_accessed = true; 75 request.response_info.network_accessed = true;
(...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after
124 } 110 }
125 111
126 DomainReliabilityContext* CreateAndAddContext(const std::string& domain) { 112 DomainReliabilityContext* CreateAndAddContext(const std::string& domain) {
127 return monitor_.AddContextForTesting(MakeTestConfigWithDomain(domain)); 113 return monitor_.AddContextForTesting(MakeTestConfigWithDomain(domain));
128 } 114 }
129 115
130 scoped_refptr<base::TestSimpleTaskRunner> pref_task_runner_; 116 scoped_refptr<base::TestSimpleTaskRunner> pref_task_runner_;
131 scoped_refptr<base::TestSimpleTaskRunner> network_task_runner_; 117 scoped_refptr<base::TestSimpleTaskRunner> network_task_runner_;
132 scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_; 118 scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_;
133 MockTime* time_; 119 MockTime* time_;
134 scoped_ptr<PrefService> pref_service_;
135 DomainReliabilityMonitor monitor_; 120 DomainReliabilityMonitor monitor_;
136 DomainReliabilityContext* context_; 121 DomainReliabilityContext* context_;
137 DomainReliabilityMonitor::RequestInfo request_; 122 DomainReliabilityMonitor::RequestInfo request_;
138 }; 123 };
139 124
140 namespace { 125 namespace {
141 126
142 TEST_F(DomainReliabilityMonitorTest, Create) { 127 TEST_F(DomainReliabilityMonitorTest, Create) {
143 EXPECT_EQ(0u, CountPendingBeacons()); 128 EXPECT_EQ(0u, CountPendingBeacons());
144 EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 0u)); 129 EXPECT_TRUE(CheckRequestCounts(kAlwaysReportIndex, 0u, 0u));
(...skipping 251 matching lines...) Expand 10 before | Expand all | Expand 10 after
396 request.url = GURL("http://test.wildcard/always_report"); 381 request.url = GURL("http://test.wildcard/always_report");
397 OnRequestLegComplete(request); 382 OnRequestLegComplete(request);
398 383
399 EXPECT_TRUE(CheckRequestCounts(context1, kAlwaysReportIndex, 1u, 0u)); 384 EXPECT_TRUE(CheckRequestCounts(context1, kAlwaysReportIndex, 1u, 0u));
400 EXPECT_TRUE(CheckRequestCounts(context2, kAlwaysReportIndex, 0u, 0u)); 385 EXPECT_TRUE(CheckRequestCounts(context2, kAlwaysReportIndex, 0u, 0u));
401 } 386 }
402 387
403 } // namespace 388 } // namespace
404 389
405 } // namespace domain_reliability 390 } // namespace domain_reliability
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698