Chromium Code Reviews| OLD | NEW |
|---|---|
| 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 Loading... | |
| 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 Loading... | |
| 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 |
| OLD | NEW |