OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "net/url_request/url_request_job_factory.h" | 5 #include "net/url_request/url_request_job_factory.h" |
6 | 6 |
7 #include "base/bind.h" | 7 #include "base/bind.h" |
8 #include "base/memory/weak_ptr.h" | 8 #include "base/memory/weak_ptr.h" |
9 #include "net/url_request/url_request_job.h" | 9 #include "net/url_request/url_request_job.h" |
10 #include "net/url_request/url_request_test_util.h" | 10 #include "net/url_request/url_request_test_util.h" |
(...skipping 69 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
80 const std::string& /* protocol */) const OVERRIDE { | 80 const std::string& /* protocol */) const OVERRIDE { |
81 return handle_all_protocols_; | 81 return handle_all_protocols_; |
82 } | 82 } |
83 | 83 |
84 mutable bool did_intercept_; | 84 mutable bool did_intercept_; |
85 mutable bool handle_all_protocols_; | 85 mutable bool handle_all_protocols_; |
86 }; | 86 }; |
87 | 87 |
88 TEST(URLRequestJobFactoryTest, NoProtocolHandler) { | 88 TEST(URLRequestJobFactoryTest, NoProtocolHandler) { |
89 TestDelegate delegate; | 89 TestDelegate delegate; |
90 scoped_refptr<URLRequestContext> request_context(new TestURLRequestContext); | 90 TestURLRequestContext request_context; |
91 TestURLRequest request(GURL("foo://bar"), &delegate); | 91 TestURLRequest request(GURL("foo://bar"), &delegate); |
92 request.set_context(request_context); | 92 request.set_context(&request_context); |
93 request.Start(); | 93 request.Start(); |
94 | 94 |
95 MessageLoop::current()->Run(); | 95 MessageLoop::current()->Run(); |
96 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); | 96 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); |
97 EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, request.status().error()); | 97 EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, request.status().error()); |
98 } | 98 } |
99 | 99 |
100 TEST(URLRequestJobFactoryTest, BasicProtocolHandler) { | 100 TEST(URLRequestJobFactoryTest, BasicProtocolHandler) { |
101 TestDelegate delegate; | 101 TestDelegate delegate; |
102 scoped_refptr<URLRequestContext> request_context(new TestURLRequestContext); | |
103 URLRequestJobFactory job_factory; | 102 URLRequestJobFactory job_factory; |
104 request_context->set_job_factory(&job_factory); | 103 TestURLRequestContext request_context; |
| 104 request_context.set_job_factory(&job_factory); |
105 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); | 105 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); |
106 TestURLRequest request(GURL("foo://bar"), &delegate); | 106 TestURLRequest request(GURL("foo://bar"), &delegate); |
107 request.set_context(request_context); | 107 request.set_context(&request_context); |
108 request.Start(); | 108 request.Start(); |
109 | 109 |
110 MessageLoop::current()->Run(); | 110 MessageLoop::current()->Run(); |
111 EXPECT_EQ(URLRequestStatus::SUCCESS, request.status().status()); | 111 EXPECT_EQ(URLRequestStatus::SUCCESS, request.status().status()); |
112 EXPECT_EQ(OK, request.status().error()); | 112 EXPECT_EQ(OK, request.status().error()); |
113 } | 113 } |
114 | 114 |
115 TEST(URLRequestJobFactoryTest, DeleteProtocolHandler) { | 115 TEST(URLRequestJobFactoryTest, DeleteProtocolHandler) { |
116 scoped_refptr<URLRequestContext> request_context(new TestURLRequestContext); | |
117 URLRequestJobFactory job_factory; | 116 URLRequestJobFactory job_factory; |
118 request_context->set_job_factory(&job_factory); | 117 TestURLRequestContext request_context; |
| 118 request_context.set_job_factory(&job_factory); |
119 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); | 119 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); |
120 job_factory.SetProtocolHandler("foo", NULL); | 120 job_factory.SetProtocolHandler("foo", NULL); |
121 } | 121 } |
122 | 122 |
123 TEST(URLRequestJobFactoryTest, BasicInterceptor) { | 123 TEST(URLRequestJobFactoryTest, BasicInterceptor) { |
124 TestDelegate delegate; | 124 TestDelegate delegate; |
125 scoped_refptr<URLRequestContext> request_context(new TestURLRequestContext); | |
126 URLRequestJobFactory job_factory; | 125 URLRequestJobFactory job_factory; |
127 request_context->set_job_factory(&job_factory); | 126 TestURLRequestContext request_context; |
| 127 request_context.set_job_factory(&job_factory); |
128 job_factory.AddInterceptor(new DummyInterceptor); | 128 job_factory.AddInterceptor(new DummyInterceptor); |
129 TestURLRequest request(GURL("http://bar"), &delegate); | 129 TestURLRequest request(GURL("http://bar"), &delegate); |
130 request.set_context(request_context); | 130 request.set_context(&request_context); |
131 request.Start(); | 131 request.Start(); |
132 | 132 |
133 MessageLoop::current()->Run(); | 133 MessageLoop::current()->Run(); |
134 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); | 134 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); |
135 EXPECT_EQ(ERR_FAILED, request.status().error()); | 135 EXPECT_EQ(ERR_FAILED, request.status().error()); |
136 } | 136 } |
137 | 137 |
138 TEST(URLRequestJobFactoryTest, InterceptorNeedsValidSchemeStill) { | 138 TEST(URLRequestJobFactoryTest, InterceptorNeedsValidSchemeStill) { |
139 TestDelegate delegate; | 139 TestDelegate delegate; |
140 scoped_refptr<URLRequestContext> request_context(new TestURLRequestContext); | |
141 URLRequestJobFactory job_factory; | 140 URLRequestJobFactory job_factory; |
142 request_context->set_job_factory(&job_factory); | 141 TestURLRequestContext request_context; |
| 142 request_context.set_job_factory(&job_factory); |
143 job_factory.AddInterceptor(new DummyInterceptor); | 143 job_factory.AddInterceptor(new DummyInterceptor); |
144 TestURLRequest request(GURL("foo://bar"), &delegate); | 144 TestURLRequest request(GURL("foo://bar"), &delegate); |
145 request.set_context(request_context); | 145 request.set_context(&request_context); |
146 request.Start(); | 146 request.Start(); |
147 | 147 |
148 MessageLoop::current()->Run(); | 148 MessageLoop::current()->Run(); |
149 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); | 149 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); |
150 EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, request.status().error()); | 150 EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, request.status().error()); |
151 } | 151 } |
152 | 152 |
153 TEST(URLRequestJobFactoryTest, InterceptorOverridesProtocolHandler) { | 153 TEST(URLRequestJobFactoryTest, InterceptorOverridesProtocolHandler) { |
154 TestDelegate delegate; | 154 TestDelegate delegate; |
155 scoped_refptr<URLRequestContext> request_context(new TestURLRequestContext); | |
156 URLRequestJobFactory job_factory; | 155 URLRequestJobFactory job_factory; |
157 request_context->set_job_factory(&job_factory); | 156 TestURLRequestContext request_context; |
| 157 request_context.set_job_factory(&job_factory); |
158 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); | 158 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); |
159 job_factory.AddInterceptor(new DummyInterceptor); | 159 job_factory.AddInterceptor(new DummyInterceptor); |
160 TestURLRequest request(GURL("foo://bar"), &delegate); | 160 TestURLRequest request(GURL("foo://bar"), &delegate); |
161 request.set_context(request_context); | 161 request.set_context(&request_context); |
162 request.Start(); | 162 request.Start(); |
163 | 163 |
164 MessageLoop::current()->Run(); | 164 MessageLoop::current()->Run(); |
165 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); | 165 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); |
166 EXPECT_EQ(ERR_FAILED, request.status().error()); | 166 EXPECT_EQ(ERR_FAILED, request.status().error()); |
167 } | 167 } |
168 | 168 |
169 TEST(URLRequestJobFactoryTest, InterceptorDoesntInterceptUnknownProtocols) { | 169 TEST(URLRequestJobFactoryTest, InterceptorDoesntInterceptUnknownProtocols) { |
170 TestDelegate delegate; | 170 TestDelegate delegate; |
171 scoped_refptr<URLRequestContext> request_context(new TestURLRequestContext); | |
172 URLRequestJobFactory job_factory; | 171 URLRequestJobFactory job_factory; |
173 request_context->set_job_factory(&job_factory); | 172 TestURLRequestContext request_context; |
| 173 request_context.set_job_factory(&job_factory); |
174 DummyInterceptor* interceptor = new DummyInterceptor; | 174 DummyInterceptor* interceptor = new DummyInterceptor; |
175 job_factory.AddInterceptor(interceptor); | 175 job_factory.AddInterceptor(interceptor); |
176 TestURLRequest request(GURL("foo://bar"), &delegate); | 176 TestURLRequest request(GURL("foo://bar"), &delegate); |
177 request.set_context(request_context); | 177 request.set_context(&request_context); |
178 request.Start(); | 178 request.Start(); |
179 | 179 |
180 MessageLoop::current()->Run(); | 180 MessageLoop::current()->Run(); |
181 EXPECT_FALSE(interceptor->did_intercept_); | 181 EXPECT_FALSE(interceptor->did_intercept_); |
182 } | 182 } |
183 | 183 |
184 TEST(URLRequestJobFactoryTest, InterceptorInterceptsHandledUnknownProtocols) { | 184 TEST(URLRequestJobFactoryTest, InterceptorInterceptsHandledUnknownProtocols) { |
185 TestDelegate delegate; | 185 TestDelegate delegate; |
186 scoped_refptr<URLRequestContext> request_context(new TestURLRequestContext); | |
187 URLRequestJobFactory job_factory; | 186 URLRequestJobFactory job_factory; |
188 request_context->set_job_factory(&job_factory); | 187 TestURLRequestContext request_context; |
| 188 request_context.set_job_factory(&job_factory); |
189 DummyInterceptor* interceptor = new DummyInterceptor; | 189 DummyInterceptor* interceptor = new DummyInterceptor; |
190 interceptor->handle_all_protocols_ = true; | 190 interceptor->handle_all_protocols_ = true; |
191 job_factory.AddInterceptor(interceptor); | 191 job_factory.AddInterceptor(interceptor); |
192 TestURLRequest request(GURL("foo://bar"), &delegate); | 192 TestURLRequest request(GURL("foo://bar"), &delegate); |
193 request.set_context(request_context); | 193 request.set_context(&request_context); |
194 request.Start(); | 194 request.Start(); |
195 | 195 |
196 MessageLoop::current()->Run(); | 196 MessageLoop::current()->Run(); |
197 EXPECT_TRUE(interceptor->did_intercept_); | 197 EXPECT_TRUE(interceptor->did_intercept_); |
198 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); | 198 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); |
199 EXPECT_EQ(ERR_FAILED, request.status().error()); | 199 EXPECT_EQ(ERR_FAILED, request.status().error()); |
200 } | 200 } |
201 | 201 |
202 TEST(URLRequestJobFactoryTest, InterceptorAffectsIsHandledProtocol) { | 202 TEST(URLRequestJobFactoryTest, InterceptorAffectsIsHandledProtocol) { |
203 DummyInterceptor* interceptor = new DummyInterceptor; | 203 DummyInterceptor* interceptor = new DummyInterceptor; |
204 URLRequestJobFactory job_factory; | 204 URLRequestJobFactory job_factory; |
205 job_factory.AddInterceptor(interceptor); | 205 job_factory.AddInterceptor(interceptor); |
206 EXPECT_FALSE(interceptor->WillHandleProtocol("anything")); | 206 EXPECT_FALSE(interceptor->WillHandleProtocol("anything")); |
207 EXPECT_FALSE(job_factory.IsHandledProtocol("anything")); | 207 EXPECT_FALSE(job_factory.IsHandledProtocol("anything")); |
208 interceptor->handle_all_protocols_ = true; | 208 interceptor->handle_all_protocols_ = true; |
209 EXPECT_TRUE(interceptor->WillHandleProtocol("anything")); | 209 EXPECT_TRUE(interceptor->WillHandleProtocol("anything")); |
210 EXPECT_TRUE(job_factory.IsHandledProtocol("anything")); | 210 EXPECT_TRUE(job_factory.IsHandledProtocol("anything")); |
211 } | 211 } |
212 | 212 |
213 } // namespace | 213 } // namespace |
214 | 214 |
215 } // namespace net | 215 } // namespace net |
OLD | NEW |