Initial RemoteCommandService

chrome/browser/policy/cloud/test_request_interceptor.h

 // Copyright (c) 2013 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 CHROME_BROWSER_POLICY_CLOUD_TEST_REQUEST_INTERCEPTOR_H_
 #define CHROME_BROWSER_POLICY_CLOUD_TEST_REQUEST_INTERCEPTOR_H_
 
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/callback.h"
 #include "base/memory/ref_counted.h"
 #include "policy/proto/device_management_backend.pb.h"
 
 namespace base {
 class FilePath;
 class SequencedTaskRunner;
 }
 
 namespace net {
 class NetworkDelegate;
 class URLRequest;
 class URLRequestJob;
 }
 
 namespace policy {
 
+class TestingRemoteCommandsServer;
+
 // Intercepts all requests to the given hostname while in scope, and allows
 // queuing callbacks to handle expected requests. Must be created and destroyed
 // while the IO thread is valid.
 class TestRequestInterceptor {
  public:
   // A callback that returns a new URLRequestJob given a URLRequest.
   // This is used to queue callbacks that will handle expected requests.
   typedef base::Callback<
       net::URLRequestJob*(net::URLRequest*, net::NetworkDelegate*)> JobCallback;
 
   // Will intercept request to |hostname| made over HTTP.
   TestRequestInterceptor(
       const std::string& hostname,
       scoped_refptr<base::SequencedTaskRunner> io_task_runner);
   virtual ~TestRequestInterceptor();
 
   // Returns the number of pending callback jobs that haven't been used yet.
   size_t GetPendingSize();
 
   // |callback| will be invoked on the current loop once the next request
   // intercepted by this class has been dispatched.
   void AddRequestServicedCallback(const base::Closure& callback);
 
   // Queues |callback| to handle a request to |hostname_|. Each callback is
   // used only once, and in the order that they're pushed.
   void PushJobCallback(const JobCallback& callback);
 
+  // Explicitly add certain request types to be ignored. For valid request of
+  // given ignored types, callback from queue will NOT be consumed and a bad
+  // request response will be sent back instead.
+  void AddIgnoredRequestType(const std::string& ignored_type);
+
   // Returns a JobCallback that will fail with the given network |error|.
   static JobCallback ErrorJob(int error);
 
   // Returns a JobCallback that will fail with HTTP 400 Bad Request.
   static JobCallback BadRequestJob();
 
   // Returns a JobCallback that will process a policy register request that
   // should succeed. The request parameters are validated, and an appropriate
   // response is sent back.
   // |expected_type| is the expected type in the register request.
   // If |expect_reregister| is true then the request must have the reregister
   // flag set; otherwise the flag must be not set.
   static JobCallback RegisterJob(
       enterprise_management::DeviceRegisterRequest::Type expected_type,
       bool expect_reregister);
 
   // Returns a JobCallback that will send the contents of |file_path|.
   static JobCallback FileJob(const base::FilePath& file_path);
 
+  // Returns a JobCallback that will handle a "fetch remote command" request.
+  // |server| is a pointer to a TestingRemoteCommandsServer instance to handle
+  // the actual request, and is expected to be valid at the time the job is
+  // serviced. |expected_command_results| is the expected number of previous
+  // command results that will be sent by client.  |expected_fetched_commands|
+  // is the expected number of fetched commands by this callback.
+  static JobCallback FetchRemoteCommandsJob(TestingRemoteCommandsServer* server,
+                                            size_t expected_command_results,
+                                            size_t expected_fetched_commands);
+
  private:
   class Delegate;
 
   // Helper to execute a |task| on IO, and return only after it has completed.
   void PostToIOAndWait(const base::Closure& task);
 
   const std::string hostname_;
 
   scoped_refptr<base::SequencedTaskRunner> io_task_runner_;
 
   // Owned by URLRequestFilter. This handle is valid on IO and only while the
   // interceptor is valid.
   Delegate* delegate_;
 
   DISALLOW_COPY_AND_ASSIGN(TestRequestInterceptor);
 };
 
 }  // namespace policy
 
 #endif  // CHROME_BROWSER_POLICY_CLOUD_TEST_REQUEST_INTERCEPTOR_H_