Domain Reliability: Don't crash on shutdown with uploads pending

components/domain_reliability/service.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/domain_reliability/service.h"
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/location.h"
 #include "base/single_thread_task_runner.h"
 #include "base/task_runner_util.h"
 #include "base/threading/thread_task_runner_handle.h"
 #include "components/domain_reliability/monitor.h"
 #include "net/url_request/url_request_context_getter.h"
 #include "url/gurl.h"
 
 namespace domain_reliability {
 
 namespace {
 
+void AddContextForTestingOnNetworkTaskRunner(
+    base::WeakPtr<DomainReliabilityMonitor> monitor,
+    std::unique_ptr<const DomainReliabilityConfig> config) {
+  if (!monitor)
+    return;
+
+  monitor->AddContextForTesting(std::move(config));
+}
+
 std::unique_ptr<base::Value> GetWebUIDataOnNetworkTaskRunner(
     base::WeakPtr<DomainReliabilityMonitor> monitor) {
   if (!monitor) {
     base::DictionaryValue* dict = new base::DictionaryValue();
     dict->SetString("error", "no_monitor");
     return std::unique_ptr<base::Value>(dict);
   }
 
   return monitor->GetWebUIData();
 }
@@ skipping 45 matching lines @@
                         callback) const override {
     DCHECK(network_task_runner_.get());
 
     PostTaskAndReplyWithResult(
         network_task_runner_.get(),
         FROM_HERE,
         base::Bind(&GetWebUIDataOnNetworkTaskRunner, monitor_),
         callback);
   }
 
+  void SetDiscardUploadsForTesting(bool discard_uploads) override {
+    DCHECK(network_task_runner_.get());
+
+    network_task_runner_->PostTask(
+        FROM_HERE,
+        base::Bind(&DomainReliabilityMonitor::SetDiscardUploads,
+                   monitor_,
+                   discard_uploads));
+  }
+
+  void AddContextForTesting(
+      std::unique_ptr<const DomainReliabilityConfig> config) override {
+    DCHECK(network_task_runner_.get());
+
+    network_task_runner_->PostTask(
+        FROM_HERE,
+        base::Bind(&AddContextForTestingOnNetworkTaskRunner,
+                   monitor_,
+                   base::Passed(&config)));
+  }
+
+  void ForceUploadsForTesting() override {
+    DCHECK(network_task_runner_.get());
+
+    network_task_runner_->PostTask(
+        FROM_HERE,
+        base::Bind(&DomainReliabilityMonitor::ForceUploadsForTesting,
+                   monitor_));
+  }
+
  private:
   std::string upload_reporter_string_;
   base::WeakPtr<DomainReliabilityMonitor> monitor_;
   scoped_refptr<base::SingleThreadTaskRunner> network_task_runner_;
 };
 
 // static
 DomainReliabilityService* DomainReliabilityService::Create(
     const std::string& upload_reporter_string) {
   return new DomainReliabilityServiceImpl(upload_reporter_string);
 }
 
 DomainReliabilityService::~DomainReliabilityService() {}
 
 DomainReliabilityService::DomainReliabilityService() {}
 
 }  // namespace domain_reliability