| // Copyright (c) 2012 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. |
| |
| #include "net/url_request/url_request_job_factory_impl.h" |
| |
| #include "base/bind.h" |
| #include "base/location.h" |
| #include "base/memory/scoped_ptr.h" |
| #include "base/memory/weak_ptr.h" |
| #include "base/single_thread_task_runner.h" |
| #include "base/thread_task_runner_handle.h" |
| #include "net/base/request_priority.h" |
| #include "net/url_request/url_request.h" |
| #include "net/url_request/url_request_job.h" |
| #include "net/url_request/url_request_test_util.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace net { |
| |
| namespace { |
| |
| class MockURLRequestJob : public URLRequestJob { |
| public: |
| MockURLRequestJob(URLRequest* request, |
| NetworkDelegate* network_delegate, |
| const URLRequestStatus& status) |
| : URLRequestJob(request, network_delegate), |
| status_(status), |
| weak_factory_(this) {} |
| |
| void Start() override { |
| // Start reading asynchronously so that all error reporting and data |
| // callbacks happen as they would for network requests. |
| base::ThreadTaskRunnerHandle::Get()->PostTask( |
| FROM_HERE, |
| base::Bind(&MockURLRequestJob::StartAsync, weak_factory_.GetWeakPtr())); |
| } |
| |
| protected: |
| ~MockURLRequestJob() override {} |
| |
| private: |
| void StartAsync() { |
| SetStatus(status_); |
| NotifyHeadersComplete(); |
| } |
| |
| URLRequestStatus status_; |
| base::WeakPtrFactory<MockURLRequestJob> weak_factory_; |
| }; |
| |
| class DummyProtocolHandler : public URLRequestJobFactory::ProtocolHandler { |
| public: |
| URLRequestJob* MaybeCreateJob( |
| URLRequest* request, |
| NetworkDelegate* network_delegate) const override { |
| return new MockURLRequestJob( |
| request, |
| network_delegate, |
| URLRequestStatus(URLRequestStatus::SUCCESS, OK)); |
| } |
| }; |
| |
| TEST(URLRequestJobFactoryTest, NoProtocolHandler) { |
| TestDelegate delegate; |
| TestURLRequestContext request_context; |
| scoped_ptr<URLRequest> request(request_context.CreateRequest( |
| GURL("foo://bar"), DEFAULT_PRIORITY, &delegate)); |
| request->Start(); |
| |
| base::MessageLoop::current()->Run(); |
| EXPECT_EQ(URLRequestStatus::FAILED, request->status().status()); |
| EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, request->status().error()); |
| } |
| |
| TEST(URLRequestJobFactoryTest, BasicProtocolHandler) { |
| TestDelegate delegate; |
| URLRequestJobFactoryImpl job_factory; |
| TestURLRequestContext request_context; |
| request_context.set_job_factory(&job_factory); |
| job_factory.SetProtocolHandler("foo", |
| make_scoped_ptr(new DummyProtocolHandler)); |
| scoped_ptr<URLRequest> request(request_context.CreateRequest( |
| GURL("foo://bar"), DEFAULT_PRIORITY, &delegate)); |
| request->Start(); |
| |
| base::MessageLoop::current()->Run(); |
| EXPECT_EQ(URLRequestStatus::SUCCESS, request->status().status()); |
| EXPECT_EQ(OK, request->status().error()); |
| } |
| |
| TEST(URLRequestJobFactoryTest, DeleteProtocolHandler) { |
| URLRequestJobFactoryImpl job_factory; |
| TestURLRequestContext request_context; |
| request_context.set_job_factory(&job_factory); |
| job_factory.SetProtocolHandler("foo", |
| make_scoped_ptr(new DummyProtocolHandler)); |
| job_factory.SetProtocolHandler("foo", nullptr); |
| } |
| |
| } // namespace |
| |
| } // namespace net |