content/browser/loader/buffered_resource_handler_unittest.cc

   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 #include "content/browser/loader/buffered_resource_handler.h"
   6
   7 #include "base/logging.h"
   8 #include "base/macros.h"
   9 #include "base/memory/scoped_ptr.h"
  10 #include "content/browser/loader/resource_dispatcher_host_impl.h"
  11 #include "content/public/browser/resource_controller.h"
  12 #include "content/public/browser/resource_dispatcher_host_delegate.h"
  13 #include "content/public/browser/resource_request_info.h"
  14 #include "content/public/common/resource_response.h"
  15 #include "content/public/test/test_browser_thread_bundle.h"
  16 #include "content/public/test/test_utils.h"
  17 #include "content/test/fake_plugin_service.h"
  18 #include "net/url_request/url_request_context.h"
  19 #include "testing/gtest/include/gtest/gtest.h"
  20 #include "url/gurl.h"
  21
  22 namespace content {
  23
  24 namespace {
  25
  26 class TestResourceHandler : public ResourceHandler {
  27  public:
  28   TestResourceHandler() : ResourceHandler(nullptr) {}
  29
  30   void SetController(ResourceController* controller) override {}
  31
  32   bool OnUploadProgress(uint64 position, uint64 size) override {
  33     NOTREACHED();
  34     return false;
  35   }
  36
  37   bool OnRequestRedirected(const net::RedirectInfo& redirect_info,
  38                            ResourceResponse* response,
  39                            bool* defer) override {
  40     NOTREACHED();
  41     return false;
  42   }
  43
  44   bool OnResponseStarted(ResourceResponse* response, bool* defer) override {
  45     return false;
  46   }
  47
  48   bool OnWillStart(const GURL& url, bool* defer) override {
  49     NOTREACHED();
  50     return false;
  51   }
  52
  53   bool OnBeforeNetworkStart(const GURL& url, bool* defer) override {
  54     NOTREACHED();
  55     return false;
  56   }
  57
  58   bool OnWillRead(scoped_refptr<net::IOBuffer>* buf,
  59                   int* buf_size,
  60                   int min_size) override {
  61     NOTREACHED();
  62     return false;
  63   }
  64
  65   bool OnReadCompleted(int bytes_read, bool* defer) override {
  66     NOTREACHED();
  67     return false;
  68   }
  69
  70   void OnResponseCompleted(const net::URLRequestStatus& status,
  71                            const std::string& security_info,
  72                            bool* defer) override {
  73   }
  74
  75   void OnDataDownloaded(int bytes_downloaded) override {
  76     NOTREACHED();
  77   }
  78
  79  private:
  80   DISALLOW_COPY_AND_ASSIGN(TestResourceHandler);
  81 };
  82
  83 class TestResourceDispatcherHost : public ResourceDispatcherHostImpl {
  84  public:
  85   explicit TestResourceDispatcherHost(bool stream_has_handler)
  86       : stream_has_handler_(stream_has_handler),
  87         intercepted_as_stream_(false) {}
  88
  89   bool intercepted_as_stream() const { return intercepted_as_stream_; }
  90
  91   scoped_ptr<ResourceHandler> CreateResourceHandlerForDownload(
  92       net::URLRequest* request,
  93       bool is_content_initiated,
  94       bool must_download,
  95       uint32 id,
  96       scoped_ptr<DownloadSaveInfo> save_info,
  97       const DownloadUrlParameters::OnStartedCallback& started_cb) override {
  98     return scoped_ptr<ResourceHandler>(new TestResourceHandler).Pass();
  99   }
 100
 101   scoped_ptr<ResourceHandler> MaybeInterceptAsStream(
 102       net::URLRequest* request,
 103       ResourceResponse* response,
 104       std::string* payload) override {
 105     if (stream_has_handler_) {
 106       intercepted_as_stream_ = true;
 107       return scoped_ptr<ResourceHandler>(new TestResourceHandler).Pass();
 108     } else {
 109       return scoped_ptr<ResourceHandler>();
 110     }
 111   }
 112
 113  private:
 114   // Whether the URL request should be intercepted as a stream.
 115   bool stream_has_handler_;
 116
 117   // Whether the URL request has been intercepted as a stream.
 118   bool intercepted_as_stream_;
 119 };
 120
 121 class TestResourceDispatcherHostDelegate
 122     : public ResourceDispatcherHostDelegate {
 123  public:
 124   TestResourceDispatcherHostDelegate(bool must_download)
 125       : must_download_(must_download) {
 126   }
 127
 128   bool ShouldForceDownloadResource(const GURL& url,
 129                                    const std::string& mime_type) override {
 130     return must_download_;
 131   }
 132
 133  private:
 134   const bool must_download_;
 135 };
 136
 137 class TestResourceController : public ResourceController {
 138  public:
 139   void Cancel() override {}
 140
 141   void CancelAndIgnore() override {
 142     NOTREACHED();
 143   }
 144
 145   void CancelWithError(int error_code) override {
 146     NOTREACHED();
 147   }
 148
 149   void Resume() override {
 150     NOTREACHED();
 151   }
 152 };
 153
 154 class BufferedResourceHandlerTest : public testing::Test {
 155  public:
 156   BufferedResourceHandlerTest() : stream_has_handler_(false) {}
 157
 158   void set_stream_has_handler(bool stream_has_handler) {
 159     stream_has_handler_ = stream_has_handler;
 160   }
 161
 162   bool TestStreamIsIntercepted(bool allow_download,
 163                                bool must_download,
 164                                ResourceType request_resource_type);
 165
 166  private:
 167   // Whether the URL request should be intercepted as a stream.
 168   bool stream_has_handler_;
 169
 170   TestBrowserThreadBundle thread_bundle_;
 171 };
 172
 173 bool BufferedResourceHandlerTest::TestStreamIsIntercepted(
 174     bool allow_download,
 175     bool must_download,
 176     ResourceType request_resource_type) {
 177   net::URLRequestContext context;
 178   scoped_ptr<net::URLRequest> request(context.CreateRequest(
 179       GURL("http://www.google.com"), net::DEFAULT_PRIORITY, nullptr, nullptr));
 180   bool is_main_frame = request_resource_type == RESOURCE_TYPE_MAIN_FRAME;
 181   ResourceRequestInfo::AllocateForTesting(
 182       request.get(),
 183       request_resource_type,
 184       nullptr,          // context
 185       0,                // render_process_id
 186       0,                // render_view_id
 187       0,                // render_frame_id
 188       is_main_frame,    // is_main_frame
 189       false,            // parent_is_main_frame
 190       allow_download,   // allow_download
 191       true);            // is_async
 192
 193   TestResourceDispatcherHost host(stream_has_handler_);
 194   TestResourceDispatcherHostDelegate host_delegate(must_download);
 195   host.SetDelegate(&host_delegate);
 196
 197   FakePluginService plugin_service;
 198   scoped_ptr<ResourceHandler> buffered_handler(
 199       new BufferedResourceHandler(
 200           scoped_ptr<ResourceHandler>(new TestResourceHandler()).Pass(),
 201           &host,
 202           &plugin_service,
 203           request.get()));
 204   TestResourceController resource_controller;
 205   buffered_handler->SetController(&resource_controller);
 206
 207   scoped_refptr<ResourceResponse> response(new ResourceResponse);
 208   // The MIME type isn't important but it shouldn't be empty.
 209   response->head.mime_type = "application/pdf";
 210
 211   bool defer = false;
 212   buffered_handler->OnResponseStarted(response.get(), &defer);
 213
 214   content::RunAllPendingInMessageLoop();
 215
 216   return host.intercepted_as_stream();
 217 }
 218
 219 // Test that stream requests are correctly intercepted under the right
 220 // circumstances.
 221 TEST_F(BufferedResourceHandlerTest, StreamHandling) {
 222   bool allow_download;
 223   bool must_download;
 224   ResourceType resource_type;
 225
 226   // Ensure the stream is handled by MaybeInterceptAsStream in the
 227   // ResourceDispatcherHost.
 228   set_stream_has_handler(true);
 229
 230   // Main frame request with no download allowed. Stream shouldn't be
 231   // intercepted.
 232   allow_download = false;
 233   must_download = false;
 234   resource_type = RESOURCE_TYPE_MAIN_FRAME;
 235   EXPECT_FALSE(
 236       TestStreamIsIntercepted(allow_download, must_download, resource_type));
 237
 238   // Main frame request with download allowed. Stream should be intercepted.
 239   allow_download = true;
 240   must_download = false;
 241   resource_type = RESOURCE_TYPE_MAIN_FRAME;
 242   EXPECT_TRUE(
 243       TestStreamIsIntercepted(allow_download, must_download, resource_type));
 244
 245   // Main frame request with download forced. Stream shouldn't be intercepted.
 246   allow_download = true;
 247   must_download = true;
 248   resource_type = RESOURCE_TYPE_MAIN_FRAME;
 249   EXPECT_FALSE(
 250       TestStreamIsIntercepted(allow_download, must_download, resource_type));
 251
 252   // Sub-resource request with download not allowed. Stream shouldn't be
 253   // intercepted.
 254   allow_download = false;
 255   must_download = false;
 256   resource_type = RESOURCE_TYPE_SUB_RESOURCE;
 257   EXPECT_FALSE(
 258       TestStreamIsIntercepted(allow_download, must_download, resource_type));
 259
 260   // Object request with download not allowed. Stream should be intercepted.
 261   allow_download = false;
 262   must_download = false;
 263   resource_type = RESOURCE_TYPE_OBJECT;
 264   EXPECT_TRUE(
 265       TestStreamIsIntercepted(allow_download, must_download, resource_type));
 266
 267   // Test the cases where the stream isn't handled by MaybeInterceptAsStream
 268   // in the ResourceDispatcherHost.
 269   set_stream_has_handler(false);
 270
 271   allow_download = false;
 272   must_download = false;
 273   resource_type = RESOURCE_TYPE_OBJECT;
 274   EXPECT_FALSE(
 275       TestStreamIsIntercepted(allow_download, must_download, resource_type));
 276
 277   allow_download = true;
 278   must_download = false;
 279   resource_type = RESOURCE_TYPE_MAIN_FRAME;
 280   EXPECT_FALSE(
 281       TestStreamIsIntercepted(allow_download, must_download, resource_type));
 282 }
 283
 284 }  // namespace
 285
 286 }  // namespace content