Add delayed retry to DeviceManagementService requests.

components/policy/core/common/cloud/device_management_service.h

 // Copyright (c) 2012 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.
 
 #ifndef COMPONENTS_POLICY_CORE_COMMON_CLOUD_DEVICE_MANAGEMENT_SERVICE_H_
 #define COMPONENTS_POLICY_CORE_COMMON_CLOUD_DEVICE_MANAGEMENT_SERVICE_H_
 
 #include <stdint.h>
 
 #include <deque>
 #include <map>
 #include <memory>
 #include <string>
 #include <vector>
 
 #include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/macros.h"
 #include "base/memory/ref_counted.h"
 #include "base/memory/weak_ptr.h"
 #include "base/strings/string_split.h"
+#include "base/threading/thread_checker.h"
 #include "components/policy/core/common/cloud/cloud_policy_constants.h"
 #include "components/policy/policy_export.h"
 #include "net/url_request/url_fetcher_delegate.h"
 #include "policy/proto/device_management_backend.pb.h"
 
 
 namespace net {
 class URLRequestContextGetter;
 }
 
@@ skipping 116 matching lines @@
 
   // Schedules a task to run |Initialize| after |delay_milliseconds| had passed.
   void ScheduleInitialization(int64_t delay_milliseconds);
 
   // Makes the service stop all requests.
   void Shutdown();
 
   // Gets the URL that the DMServer requests are sent to.
   std::string GetServerUrl();
 
+  // Sets the retry delay to a shorter time to prevent browser tests from
+  // timing out.
+  static void SetRetryDelayForTesting(long retryDelayMs);
+
  private:
   typedef std::map<const net::URLFetcher*,
                    DeviceManagementRequestJobImpl*> JobFetcherMap;
   typedef std::deque<DeviceManagementRequestJobImpl*> JobQueue;
 
   friend class DeviceManagementRequestJobImpl;
 
   // net::URLFetcherDelegate override.
   void OnURLFetchComplete(const net::URLFetcher* source) override;
 
@@ skipping 18 matching lines @@
   // The jobs we currently have in flight.
   JobFetcherMap pending_jobs_;
 
   // Jobs that are registered, but not started yet.
   JobQueue queued_jobs_;
 
   // If this service is initialized, incoming requests get fired instantly.
   // If it is not initialized, incoming requests are queued.
   bool initialized_;
 
-  // Used to create tasks to run |Initialize| delayed on the UI thread.
+  // TaskRunner used to schedule retry attempts.
+  const scoped_refptr<base::SequencedTaskRunner> task_runner_;
+
+  base::ThreadChecker thread_checker_;
+
+  // Used to create tasks which run delayed on the UI thread.
   base::WeakPtrFactory<DeviceManagementService> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(DeviceManagementService);
 };
 
 }  // namespace policy
 
 #endif  // COMPONENTS_POLICY_CORE_COMMON_CLOUD_DEVICE_MANAGEMENT_SERVICE_H_