1 // Copyright 2012 The Chromium Authors
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.h"
6
7 #include "base/functional/bind.h"
8 #include "base/location.h"
9 #include "base/memory/weak_ptr.h"
10 #include "base/run_loop.h"
11 #include "base/task/single_thread_task_runner.h"
12 #include "base/test/task_environment.h"
13 #include "net/base/request_priority.h"
14 #include "net/test/gtest_util.h"
15 #include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
16 #include "net/url_request/url_request.h"
17 #include "net/url_request/url_request_context.h"
18 #include "net/url_request/url_request_context_builder.h"
19 #include "net/url_request/url_request_job.h"
20 #include "net/url_request/url_request_test_util.h"
21 #include "testing/gmock/include/gmock/gmock.h"
22 #include "testing/gtest/include/gtest/gtest.h"
23
24 using net::test::IsError;
25 using net::test::IsOk;
26
27 namespace net {
28
29 namespace {
30
31 class MockURLRequestJob : public URLRequestJob {
32 public:
MockURLRequestJob(URLRequest * request)33 explicit MockURLRequestJob(URLRequest* request) : URLRequestJob(request) {}
34
35 ~MockURLRequestJob() override = default;
36
Start()37 void Start() override {
38 // Start reading asynchronously so that all error reporting and data
39 // callbacks happen as they would for network requests.
40 base::SingleThreadTaskRunner::GetCurrentDefault()->PostTask(
41 FROM_HERE, base::BindOnce(&MockURLRequestJob::StartAsync,
42 weak_factory_.GetWeakPtr()));
43 }
44
45 private:
StartAsync()46 void StartAsync() { NotifyHeadersComplete(); }
47
48 base::WeakPtrFactory<MockURLRequestJob> weak_factory_{this};
49 };
50
51 class DummyProtocolHandler : public URLRequestJobFactory::ProtocolHandler {
52 public:
CreateJob(URLRequest * request) const53 std::unique_ptr<URLRequestJob> CreateJob(URLRequest* request) const override {
54 return std::make_unique<MockURLRequestJob>(request);
55 }
56 };
57
TEST(URLRequestJobFactoryTest,NoProtocolHandler)58 TEST(URLRequestJobFactoryTest, NoProtocolHandler) {
59 base::test::TaskEnvironment task_environment(
60 base::test::TaskEnvironment::MainThreadType::IO);
61 TestDelegate delegate;
62 auto request_context = CreateTestURLRequestContextBuilder()->Build();
63 std::unique_ptr<URLRequest> request(
64 request_context->CreateRequest(GURL("foo://bar"), DEFAULT_PRIORITY,
65 &delegate, TRAFFIC_ANNOTATION_FOR_TESTS));
66 request->Start();
67
68 delegate.RunUntilComplete();
69 EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, delegate.request_status());
70 }
71
TEST(URLRequestJobFactoryTest,BasicProtocolHandler)72 TEST(URLRequestJobFactoryTest, BasicProtocolHandler) {
73 base::test::TaskEnvironment task_environment(
74 base::test::TaskEnvironment::MainThreadType::IO);
75 TestDelegate delegate;
76 auto context_builder = CreateTestURLRequestContextBuilder();
77 context_builder->SetProtocolHandler("foo",
78 std::make_unique<DummyProtocolHandler>());
79 auto request_context = context_builder->Build();
80 std::unique_ptr<URLRequest> request(
81 request_context->CreateRequest(GURL("foo://bar"), DEFAULT_PRIORITY,
82 &delegate, TRAFFIC_ANNOTATION_FOR_TESTS));
83 request->Start();
84
85 delegate.RunUntilComplete();
86 EXPECT_EQ(OK, delegate.request_status());
87 }
88
89 } // namespace
90
91 } // namespace net
92