// 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 { // 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(); // 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); // 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); 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_