Index: chrome/browser/policy/test_request_interceptor.h |
diff --git a/chrome/browser/policy/test_request_interceptor.h b/chrome/browser/policy/test_request_interceptor.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..e9d968b045d3512614dc3466dd6c0b20a13b9fdf |
--- /dev/null |
+++ b/chrome/browser/policy/test_request_interceptor.h |
@@ -0,0 +1,69 @@ |
+// 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_TEST_REQUEST_INTERCEPTOR_H_ |
+#define CHROME_BROWSER_POLICY_TEST_REQUEST_INTERCEPTOR_H_ |
+ |
+#include <string> |
+ |
+#include "base/basictypes.h" |
+#include "base/callback.h" |
+ |
+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. |
+ explicit TestRequestInterceptor(const std::string& hostname); |
+ virtual ~TestRequestInterceptor(); |
+ |
+ // Returns the number of pending callback jobs that haven't been used yet. |
+ size_t GetPendingSize() const; |
+ |
+ // 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. |
+ // If |expect_reregister| is true then the request must have the reregister |
+ // flag set; otherwise the flag must be not set. |
+ static JobCallback RegisterJob(bool expect_reregister); |
+ |
+ private: |
+ class Delegate; |
+ |
+ const std::string hostname_; |
+ |
+ // 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_TEST_REQUEST_INTERCEPTOR_H_ |