| 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 70 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 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 TestURLRequestContext request_context; | 90 TestURLRequestContext request_context; |
| 91 TestURLRequest request(GURL("foo://bar"), &delegate); | 91 TestURLRequest request(GURL("foo://bar"), &delegate, &request_context); |
| 92 request.set_context(&request_context); | |
| 93 request.Start(); | 92 request.Start(); |
| 94 | 93 |
| 95 MessageLoop::current()->Run(); | 94 MessageLoop::current()->Run(); |
| 96 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); | 95 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); |
| 97 EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, request.status().error()); | 96 EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, request.status().error()); |
| 98 } | 97 } |
| 99 | 98 |
| 100 TEST(URLRequestJobFactoryTest, BasicProtocolHandler) { | 99 TEST(URLRequestJobFactoryTest, BasicProtocolHandler) { |
| 101 TestDelegate delegate; | 100 TestDelegate delegate; |
| 102 URLRequestJobFactory job_factory; | 101 URLRequestJobFactory job_factory; |
| 103 TestURLRequestContext request_context; | 102 TestURLRequestContext request_context; |
| 104 request_context.set_job_factory(&job_factory); | 103 request_context.set_job_factory(&job_factory); |
| 105 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); | 104 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); |
| 106 TestURLRequest request(GURL("foo://bar"), &delegate); | 105 TestURLRequest request(GURL("foo://bar"), &delegate, &request_context); |
| 107 request.set_context(&request_context); | |
| 108 request.Start(); | 106 request.Start(); |
| 109 | 107 |
| 110 MessageLoop::current()->Run(); | 108 MessageLoop::current()->Run(); |
| 111 EXPECT_EQ(URLRequestStatus::SUCCESS, request.status().status()); | 109 EXPECT_EQ(URLRequestStatus::SUCCESS, request.status().status()); |
| 112 EXPECT_EQ(OK, request.status().error()); | 110 EXPECT_EQ(OK, request.status().error()); |
| 113 } | 111 } |
| 114 | 112 |
| 115 TEST(URLRequestJobFactoryTest, DeleteProtocolHandler) { | 113 TEST(URLRequestJobFactoryTest, DeleteProtocolHandler) { |
| 116 URLRequestJobFactory job_factory; | 114 URLRequestJobFactory job_factory; |
| 117 TestURLRequestContext request_context; | 115 TestURLRequestContext request_context; |
| 118 request_context.set_job_factory(&job_factory); | 116 request_context.set_job_factory(&job_factory); |
| 119 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); | 117 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); |
| 120 job_factory.SetProtocolHandler("foo", NULL); | 118 job_factory.SetProtocolHandler("foo", NULL); |
| 121 } | 119 } |
| 122 | 120 |
| 123 TEST(URLRequestJobFactoryTest, BasicInterceptor) { | 121 TEST(URLRequestJobFactoryTest, BasicInterceptor) { |
| 124 TestDelegate delegate; | 122 TestDelegate delegate; |
| 125 URLRequestJobFactory job_factory; | 123 URLRequestJobFactory job_factory; |
| 126 TestURLRequestContext request_context; | 124 TestURLRequestContext request_context; |
| 127 request_context.set_job_factory(&job_factory); | 125 request_context.set_job_factory(&job_factory); |
| 128 job_factory.AddInterceptor(new DummyInterceptor); | 126 job_factory.AddInterceptor(new DummyInterceptor); |
| 129 TestURLRequest request(GURL("http://bar"), &delegate); | 127 TestURLRequest request(GURL("http://bar"), &delegate, &request_context); |
| 130 request.set_context(&request_context); | |
| 131 request.Start(); | 128 request.Start(); |
| 132 | 129 |
| 133 MessageLoop::current()->Run(); | 130 MessageLoop::current()->Run(); |
| 134 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); | 131 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); |
| 135 EXPECT_EQ(ERR_FAILED, request.status().error()); | 132 EXPECT_EQ(ERR_FAILED, request.status().error()); |
| 136 } | 133 } |
| 137 | 134 |
| 138 TEST(URLRequestJobFactoryTest, InterceptorNeedsValidSchemeStill) { | 135 TEST(URLRequestJobFactoryTest, InterceptorNeedsValidSchemeStill) { |
| 139 TestDelegate delegate; | 136 TestDelegate delegate; |
| 140 URLRequestJobFactory job_factory; | 137 URLRequestJobFactory job_factory; |
| 141 TestURLRequestContext request_context; | 138 TestURLRequestContext request_context; |
| 142 request_context.set_job_factory(&job_factory); | 139 request_context.set_job_factory(&job_factory); |
| 143 job_factory.AddInterceptor(new DummyInterceptor); | 140 job_factory.AddInterceptor(new DummyInterceptor); |
| 144 TestURLRequest request(GURL("foo://bar"), &delegate); | 141 TestURLRequest request(GURL("foo://bar"), &delegate, &request_context); |
| 145 request.set_context(&request_context); | |
| 146 request.Start(); | 142 request.Start(); |
| 147 | 143 |
| 148 MessageLoop::current()->Run(); | 144 MessageLoop::current()->Run(); |
| 149 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); | 145 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); |
| 150 EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, request.status().error()); | 146 EXPECT_EQ(ERR_UNKNOWN_URL_SCHEME, request.status().error()); |
| 151 } | 147 } |
| 152 | 148 |
| 153 TEST(URLRequestJobFactoryTest, InterceptorOverridesProtocolHandler) { | 149 TEST(URLRequestJobFactoryTest, InterceptorOverridesProtocolHandler) { |
| 154 TestDelegate delegate; | 150 TestDelegate delegate; |
| 155 URLRequestJobFactory job_factory; | 151 URLRequestJobFactory job_factory; |
| 156 TestURLRequestContext request_context; | 152 TestURLRequestContext request_context; |
| 157 request_context.set_job_factory(&job_factory); | 153 request_context.set_job_factory(&job_factory); |
| 158 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); | 154 job_factory.SetProtocolHandler("foo", new DummyProtocolHandler); |
| 159 job_factory.AddInterceptor(new DummyInterceptor); | 155 job_factory.AddInterceptor(new DummyInterceptor); |
| 160 TestURLRequest request(GURL("foo://bar"), &delegate); | 156 TestURLRequest request(GURL("foo://bar"), &delegate, &request_context); |
| 161 request.set_context(&request_context); | |
| 162 request.Start(); | 157 request.Start(); |
| 163 | 158 |
| 164 MessageLoop::current()->Run(); | 159 MessageLoop::current()->Run(); |
| 165 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); | 160 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); |
| 166 EXPECT_EQ(ERR_FAILED, request.status().error()); | 161 EXPECT_EQ(ERR_FAILED, request.status().error()); |
| 167 } | 162 } |
| 168 | 163 |
| 169 TEST(URLRequestJobFactoryTest, InterceptorDoesntInterceptUnknownProtocols) { | 164 TEST(URLRequestJobFactoryTest, InterceptorDoesntInterceptUnknownProtocols) { |
| 170 TestDelegate delegate; | 165 TestDelegate delegate; |
| 171 URLRequestJobFactory job_factory; | 166 URLRequestJobFactory job_factory; |
| 172 TestURLRequestContext request_context; | 167 TestURLRequestContext request_context; |
| 173 request_context.set_job_factory(&job_factory); | 168 request_context.set_job_factory(&job_factory); |
| 174 DummyInterceptor* interceptor = new DummyInterceptor; | 169 DummyInterceptor* interceptor = new DummyInterceptor; |
| 175 job_factory.AddInterceptor(interceptor); | 170 job_factory.AddInterceptor(interceptor); |
| 176 TestURLRequest request(GURL("foo://bar"), &delegate); | 171 TestURLRequest request(GURL("foo://bar"), &delegate, &request_context); |
| 177 request.set_context(&request_context); | |
| 178 request.Start(); | 172 request.Start(); |
| 179 | 173 |
| 180 MessageLoop::current()->Run(); | 174 MessageLoop::current()->Run(); |
| 181 EXPECT_FALSE(interceptor->did_intercept_); | 175 EXPECT_FALSE(interceptor->did_intercept_); |
| 182 } | 176 } |
| 183 | 177 |
| 184 TEST(URLRequestJobFactoryTest, InterceptorInterceptsHandledUnknownProtocols) { | 178 TEST(URLRequestJobFactoryTest, InterceptorInterceptsHandledUnknownProtocols) { |
| 185 TestDelegate delegate; | 179 TestDelegate delegate; |
| 186 URLRequestJobFactory job_factory; | 180 URLRequestJobFactory job_factory; |
| 187 TestURLRequestContext request_context; | 181 TestURLRequestContext request_context; |
| 188 request_context.set_job_factory(&job_factory); | 182 request_context.set_job_factory(&job_factory); |
| 189 DummyInterceptor* interceptor = new DummyInterceptor; | 183 DummyInterceptor* interceptor = new DummyInterceptor; |
| 190 interceptor->handle_all_protocols_ = true; | 184 interceptor->handle_all_protocols_ = true; |
| 191 job_factory.AddInterceptor(interceptor); | 185 job_factory.AddInterceptor(interceptor); |
| 192 TestURLRequest request(GURL("foo://bar"), &delegate); | 186 TestURLRequest request(GURL("foo://bar"), &delegate, &request_context); |
| 193 request.set_context(&request_context); | |
| 194 request.Start(); | 187 request.Start(); |
| 195 | 188 |
| 196 MessageLoop::current()->Run(); | 189 MessageLoop::current()->Run(); |
| 197 EXPECT_TRUE(interceptor->did_intercept_); | 190 EXPECT_TRUE(interceptor->did_intercept_); |
| 198 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); | 191 EXPECT_EQ(URLRequestStatus::FAILED, request.status().status()); |
| 199 EXPECT_EQ(ERR_FAILED, request.status().error()); | 192 EXPECT_EQ(ERR_FAILED, request.status().error()); |
| 200 } | 193 } |
| 201 | 194 |
| 202 TEST(URLRequestJobFactoryTest, InterceptorAffectsIsHandledProtocol) { | 195 TEST(URLRequestJobFactoryTest, InterceptorAffectsIsHandledProtocol) { |
| 203 DummyInterceptor* interceptor = new DummyInterceptor; | 196 DummyInterceptor* interceptor = new DummyInterceptor; |
| 204 URLRequestJobFactory job_factory; | 197 URLRequestJobFactory job_factory; |
| 205 job_factory.AddInterceptor(interceptor); | 198 job_factory.AddInterceptor(interceptor); |
| 206 EXPECT_FALSE(interceptor->WillHandleProtocol("anything")); | 199 EXPECT_FALSE(interceptor->WillHandleProtocol("anything")); |
| 207 EXPECT_FALSE(job_factory.IsHandledProtocol("anything")); | 200 EXPECT_FALSE(job_factory.IsHandledProtocol("anything")); |
| 208 interceptor->handle_all_protocols_ = true; | 201 interceptor->handle_all_protocols_ = true; |
| 209 EXPECT_TRUE(interceptor->WillHandleProtocol("anything")); | 202 EXPECT_TRUE(interceptor->WillHandleProtocol("anything")); |
| 210 EXPECT_TRUE(job_factory.IsHandledProtocol("anything")); | 203 EXPECT_TRUE(job_factory.IsHandledProtocol("anything")); |
| 211 } | 204 } |
| 212 | 205 |
| 213 } // namespace | 206 } // namespace |
| 214 | 207 |
| 215 } // namespace net | 208 } // namespace net |
| OLD | NEW |