Make ResourceHandler::OnWillRead able to complete asynchronously.

content/browser/loader/test_resource_handler.h

 // 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_RESOURCE_HANDLER_H_
 #define CONTENT_BROWSER_LOADER_TEST_RESOURCE_HANDLER_H_
 
 #include <memory>
 #include <string>
 
@@ skipping 34 matching lines @@
   // ResourceHandler implementation:
   void OnRequestRedirected(
       const net::RedirectInfo& redirect_info,
       ResourceResponse* response,
       std::unique_ptr<ResourceController> controller) override;
   void OnResponseStarted(
       ResourceResponse* response,
       std::unique_ptr<ResourceController> controller) override;
   void OnWillStart(const GURL& url,
                    std::unique_ptr<ResourceController> controller) override;
-  bool OnWillRead(scoped_refptr<net::IOBuffer>* buf,
-                  int* buf_size) override;
+  void OnWillRead(scoped_refptr<net::IOBuffer>* buf,
+                  int* buf_size,
+                  std::unique_ptr<ResourceController> controller) override;
   void OnReadCompleted(int bytes_read,
                        std::unique_ptr<ResourceController> controller) override;
   void OnResponseCompleted(
       const net::URLRequestStatus& status,
       std::unique_ptr<ResourceController> controller) override;
   void OnDataDownloaded(int bytes_downloaded) override;
 
   void Resume();
   void CancelWithError(net::Error error_code);
 
@@ skipping 28 matching lines @@
 
   void set_defer_on_will_start(bool defer_on_will_start) {
     defer_on_will_start_ = defer_on_will_start;
   }
   void set_defer_on_request_redirected(bool defer_on_request_redirected) {
     defer_on_request_redirected_ = defer_on_request_redirected;
   }
   void set_defer_on_response_started(bool defer_on_response_started) {
     defer_on_response_started_ = defer_on_response_started;
   }
+  // Only the next OnWillRead call will set |defer| to true.
+  void set_defer_on_will_read(bool defer_on_will_read) {
+    defer_on_will_read_ = defer_on_will_read;
+  }
   // Only the next OnReadCompleted call will set |defer| to true.
   void set_defer_on_read_completed(bool defer_on_read_completed) {
     defer_on_read_completed_ = defer_on_read_completed;
   }
   // The final-byte read will set |defer| to true.
   void set_defer_on_read_eof(bool defer_on_read_eof) {
     defer_on_read_eof_ = defer_on_read_eof;
   }
   void set_defer_on_response_completed(bool defer_on_response_completed) {
     defer_on_response_completed_ = defer_on_response_completed;
@@ skipping 12 matching lines @@
 
   // Return the number of times the corresponding method was invoked.
 
   int on_will_start_called() const { return on_will_start_called_; }
   int on_request_redirected_called() const {
     return on_request_redirected_called_;
   }
   int on_response_started_called() const { return on_response_started_called_; }
   int on_will_read_called() const { return on_will_read_called_; }
   int on_read_completed_called() const { return on_read_completed_called_; }
-  int on_read_eof() const { return on_read_eof_; }
+  int on_read_eof_called() const { return on_read_eof_called_; }
   int on_response_completed_called() const {
     return on_response_completed_called_;
   }
 
   // URL passed to OnResponseStarted, if it was called.
   const GURL& start_url() const { return start_url_; }
 
   ResourceResponse* resource_response() { return resource_response_.get(); };
 
   int total_bytes_downloaded() const { return total_bytes_downloaded_; }
@@ skipping 26 matching lines @@
   bool on_will_start_result_ = true;
   bool on_request_redirected_result_ = true;
   bool on_response_started_result_ = true;
   bool on_will_read_result_ = true;
   bool on_read_completed_result_ = true;
   bool on_read_eof_result_ = true;
 
   bool defer_on_will_start_ = false;
   bool defer_on_request_redirected_ = false;
   bool defer_on_response_started_ = false;
+  bool defer_on_will_read_ = false;
   bool defer_on_read_completed_ = false;
   bool defer_on_read_eof_ = false;
   bool defer_on_response_completed_ = false;
 
   bool expect_on_data_downloaded_ = false;
 
   bool expect_eof_read_ = true;
 
   int on_will_start_called_ = 0;
   int on_request_redirected_called_ = 0;
   int on_response_started_called_ = 0;
   int on_will_read_called_ = 0;
   int on_read_completed_called_ = 0;
-  int on_read_eof_ = 0;
+  int on_read_eof_called_ = 0;
   int on_response_completed_called_ = 0;
 
   GURL start_url_;
   scoped_refptr<ResourceResponse> resource_response_;
   int total_bytes_downloaded_ = 0;
   std::string body_;
   net::URLRequestStatus final_status_ =
       net::URLRequestStatus::FromError(net::ERR_UNEXPECTED);
   bool canceled_ = false;
 
+  // Pointers to the parent's read buffer and size. Only non-NULL during
+  // OnWillRead call, until cancellation or resumption.
+  scoped_refptr<net::IOBuffer>* parent_read_buffer_ = nullptr;
+  int* parent_read_buffer_size_ = nullptr;
+
   // Tracks recursive calls, which aren't allowed.
   int call_depth_ = 0;
 
   std::unique_ptr<base::RunLoop> deferred_run_loop_;
 
   base::RunLoop response_complete_run_loop_;
 
   base::WeakPtrFactory<TestResourceHandler> weak_ptr_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(TestResourceHandler);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_BROWSER_LOADER_TEST_RESOURCE_HANDLER_H_