Index: components/domain_reliability/monitor_unittest.cc |
diff --git a/components/domain_reliability/monitor_unittest.cc b/components/domain_reliability/monitor_unittest.cc |
index 6e7b9134fe434c3e2ea42bc3a75b82297120adc0..a120db51e625b58cf6614d39714dacaaedf7844c 100644 |
--- a/components/domain_reliability/monitor_unittest.cc |
+++ b/components/domain_reliability/monitor_unittest.cc |
@@ -11,6 +11,8 @@ |
#include "base/bind.h" |
#include "base/memory/scoped_ptr.h" |
#include "base/message_loop/message_loop_proxy.h" |
+#include "base/prefs/pref_registry_simple.h" |
+#include "base/prefs/testing_pref_service.h" |
#include "base/test/test_simple_task_runner.h" |
#include "components/domain_reliability/baked_in_configs.h" |
#include "components/domain_reliability/beacon.h" |
@@ -34,6 +36,8 @@ typedef std::vector<DomainReliabilityBeacon> BeaconVector; |
static const size_t kAlwaysReportIndex = 0u; |
static const size_t kNeverReportIndex = 1u; |
+static const char* kPrefName = "reporting_enabled"; |
+ |
scoped_refptr<net::HttpResponseHeaders> MakeHttpResponseHeaders( |
const std::string& headers) { |
return scoped_refptr<net::HttpResponseHeaders>( |
@@ -54,7 +58,13 @@ class DomainReliabilityMonitorTest : public testing::Test { |
time_(new MockTime()), |
monitor_("test-reporter", scoped_ptr<MockableTime>(time_)), |
context_(NULL) { |
- monitor_.Init(url_request_context_getter_); |
+ // This will block uploads, but there's no test for whether uploads happen. |
+ // TODO(ttuttle): Test that uploads actually happen. |
+ pref_service_.registry()->RegisterBooleanPref(kPrefName, false); |
+ pref_service_.SetUserPref(kPrefName, new base::FundamentalValue(true)); |
+ monitor_.InitReportingPref( |
+ &pref_service_, kPrefName, network_task_runner_); |
+ monitor_.InitURLRequestContext(url_request_context_getter_); |
context_ = monitor_.AddContextForTesting(MakeTestConfig()); |
} |
@@ -111,6 +121,7 @@ class DomainReliabilityMonitorTest : public testing::Test { |
scoped_refptr<base::TestSimpleTaskRunner> network_task_runner_; |
scoped_refptr<net::URLRequestContextGetter> url_request_context_getter_; |
MockTime* time_; |
+ TestingPrefServiceSimple pref_service_; |
DomainReliabilityMonitor monitor_; |
DomainReliabilityContext* context_; |
DomainReliabilityMonitor::RequestInfo request_; |