net/url_request/url_request_job_factory_impl_unittest.cc

   1 // Copyright (c) 2012 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 "net/url_request/url_request_job_factory_impl.h"
   6
   7 #include "base/bind.h"
   8 #include "base/location.h"
   9 #include "base/memory/scoped_ptr.h"
  10 #include "base/memory/weak_ptr.h"
  11 #include "base/single_thread_task_runner.h"
  12 #include "base/thread_task_runner_handle.h"
  13 #include "net/base/request_priority.h"
  14 #include "net/url_request/url_request.h"
  15 #include "net/url_request/url_request_job.h"
  16 #include "net/url_request/url_request_test_util.h"
  17 #include "testing/gtest/include/gtest/gtest.h"
  18
  19 namespace net {
  20
  21 namespace {
  22
  23 class MockURLRequestJob : public URLRequestJob {
  24  public:
  25   MockURLRequestJob(URLRequest* request,
  26                     NetworkDelegate* network_delegate,
  27                     const URLRequestStatus& status)
  28       : URLRequestJob(request, network_delegate),
  29         status_(status),
  30         weak_factory_(this) {}
  31
  32   void Start() override {
  33     // Start reading asynchronously so that all error reporting and data
  34     // callbacks happen as they would for network requests.
  35     base::ThreadTaskRunnerHandle::Get()->PostTask(
  36         FROM_HERE,
  37         base::Bind(&MockURLRequestJob::StartAsync, weak_factory_.GetWeakPtr()));
  38   }
  39
  40  protected:
  41   ~MockURLRequestJob() override {}
  42
  43  private:
  44   void StartAsync() {
  45     SetStatus(status_);
  46     NotifyHeadersComplete();
  47   }
  48
  49   URLRequestStatus status_;
  50   base::WeakPtrFactory<MockURLRequestJob> weak_factory_;
  51 };
  52
  53 class DummyProtocolHandler : public URLRequestJobFactory::ProtocolHandler {
  54  public:
  55   URLRequestJob* MaybeCreateJob(
  56       URLRequest* request,
  57       NetworkDelegate* network_delegate) const override {
  58     return new MockURLRequestJob(
  59         request,
  60         network_delegate,
  61         URLRequestStatus(URLRequestStatus::SUCCESS, OK));
  62   }
  63 };
  64
  65 TEST(URLRequestJobFactoryTest, NoProtocolHandler) {
  66   TestDelegate delegate;
  67   TestURLRequestContext request_context;
  68   scoped_ptr<URLRequest> request(request_context.CreateRequest(
  69       GURL("foo://bar"), DEFAULT_PRIORITY, &delegate));
  70   request->Start();
  71
  72   base::MessageLoop::current()->Run();
  73   EXPECT_EQ(URLRequestStatus::FAILED, request->status().status());
  74   EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, request->status().error());
  75 }
  76
  77 TEST(URLRequestJobFactoryTest, BasicProtocolHandler) {
  78   TestDelegate delegate;
  79   URLRequestJobFactoryImpl job_factory;
  80   TestURLRequestContext request_context;
  81   request_context.set_job_factory(&job_factory);
  82   job_factory.SetProtocolHandler("foo",
  83                                  make_scoped_ptr(new DummyProtocolHandler));
  84   scoped_ptr<URLRequest> request(request_context.CreateRequest(
  85       GURL("foo://bar"), DEFAULT_PRIORITY, &delegate));
  86   request->Start();
  87
  88   base::MessageLoop::current()->Run();
  89   EXPECT_EQ(URLRequestStatus::SUCCESS, request->status().status());
  90   EXPECT_EQ(OK, request->status().error());
  91 }
  92
  93 TEST(URLRequestJobFactoryTest, DeleteProtocolHandler) {
  94   URLRequestJobFactoryImpl job_factory;
  95   TestURLRequestContext request_context;
  96   request_context.set_job_factory(&job_factory);
  97   job_factory.SetProtocolHandler("foo",
  98                                  make_scoped_ptr(new DummyProtocolHandler));
  99   job_factory.SetProtocolHandler("foo", nullptr);
 100 }
 101
 102 }  // namespace
 103
 104 }  // namespace net