content/browser/fileapi/mock_url_request_delegate.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef CONTENT_BROWSER_FILEAPI_MOCK_URL_REQUEST_DELEGATE_H_
   6 #define CONTENT_BROWSER_FILEAPI_MOCK_URL_REQUEST_DELEGATE_H_
   7
   8 #include "net/url_request/url_request.h"
   9
  10 namespace net {
  11 class IOBuffer;
  12 }
  13
  14 namespace content {
  15
  16 // A URL request delegate that receives the response body.
  17 class MockURLRequestDelegate : public net::URLRequest::Delegate {
  18  public:
  19   MockURLRequestDelegate();
  20   ~MockURLRequestDelegate() override;
  21
  22   void OnResponseStarted(net::URLRequest* request) override;
  23   void OnReadCompleted(net::URLRequest* request, int bytes_read) override;
  24   const std::string& response_data() const { return response_data_; }
  25
  26  private:
  27   void ReadSome(net::URLRequest* request);
  28   void ReceiveData(net::URLRequest* request, int bytes_read);
  29   void RequestComplete();
  30
  31   scoped_refptr<net::IOBuffer> io_buffer_;
  32   std::string response_data_;
  33 };
  34
  35 }  // namespace content
  36
  37 #endif  // CONTENT_BROWSER_FILEAPI_MOCK_URL_REQUEST_DELEGATE_H_