Index: content/browser/loader/test_url_loader_client.h |
diff --git a/content/browser/loader/test_url_loader_client.h b/content/browser/loader/test_url_loader_client.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..78cb46f3f278d646d46f4384b96d952833ed26ee |
--- /dev/null |
+++ b/content/browser/loader/test_url_loader_client.h |
@@ -0,0 +1,63 @@ |
+// Copyright 2016 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 CONTENT_BROWSER_LOADER_TEST_URL_LOADER_CLIENT_H_ |
+#define CONTENT_BROWSER_LOADER_TEST_URL_LOADER_CLIENT_H_ |
+ |
+#include "base/callback.h" |
+#include "base/macros.h" |
+#include "content/common/resource_request_completion_status.h" |
+#include "content/common/url_loader.mojom.h" |
+#include "content/public/common/resource_response.h" |
+#include "mojo/public/c/system/data_pipe.h" |
+#include "mojo/public/cpp/bindings/binding.h" |
+ |
+namespace content { |
+ |
+// A TestURLLoaderClient records URLLoaderClient function calls. It also calls |
+// the closure set via set_quit_closure if set, in order to make it possible to |
+// create a base::RunLoop, set its quit closure to this client and then run the |
+// RunLoop. |
+class TestURLLoaderClient final : public mojom::URLLoaderClient { |
+ public: |
+ TestURLLoaderClient(); |
+ ~TestURLLoaderClient() override; |
+ |
+ void OnReceiveResponse(const ResourceResponseHead& response_head) override; |
+ void OnStartLoadingResponseBody( |
+ mojo::ScopedDataPipeConsumerHandle body) override; |
+ void OnComplete(const ResourceRequestCompletionStatus& status) override; |
+ |
+ bool has_received_response() const { return has_received_response_; } |
+ bool has_received_completion() const { return has_received_completion_; } |
+ const ResourceResponseHead& response_head() const { return response_head_; } |
+ mojo::DataPipeConsumerHandle response_body() { return response_body_.get(); } |
+ const ResourceRequestCompletionStatus& completion_status() const { |
+ return completion_status_; |
+ } |
+ |
+ mojom::URLLoaderClientPtr CreateInterfacePtrAndBind(); |
+ void Unbind(); |
+ |
+ void RunUntilResponseReceived(); |
+ void RunUntilResponseBodyArrived(); |
+ void RunUntilComplete(); |
+ |
+ private: |
+ mojo::Binding<mojom::URLLoaderClient> binding_; |
+ ResourceResponseHead response_head_; |
+ mojo::ScopedDataPipeConsumerHandle response_body_; |
+ ResourceRequestCompletionStatus completion_status_; |
+ bool has_received_response_ = false; |
+ bool has_received_completion_ = false; |
+ base::Closure quit_closure_for_on_received_response_; |
+ base::Closure quit_closure_for_on_start_loading_response_body_; |
+ base::Closure quit_closure_for_on_complete_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(TestURLLoaderClient); |
+}; |
+ |
+} // namespace content |
+ |
+#endif // CONTENT_BROWSER_LOADER_TEST_URL_LOADER_CLIENT_H_ |