| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "chrome/browser/search/iframe_source.h" | 5 #include "chrome/browser/search/iframe_source.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/memory/ref_counted_memory.h" | 8 #include "base/memory/ref_counted_memory.h" |
| 9 #include "base/memory/scoped_ptr.h" | 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
| (...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 86 | 86 |
| 87 TestIframeSource* source() { return source_.get(); } | 87 TestIframeSource* source() { return source_.get(); } |
| 88 | 88 |
| 89 std::string response_string() { | 89 std::string response_string() { |
| 90 if (response_.get()) { | 90 if (response_.get()) { |
| 91 return std::string(response_->front_as<char>(), response_->size()); | 91 return std::string(response_->front_as<char>(), response_->size()); |
| 92 } | 92 } |
| 93 return ""; | 93 return ""; |
| 94 } | 94 } |
| 95 | 95 |
| 96 net::URLRequest* MockRequest( | 96 scoped_ptr<net::URLRequest> MockRequest( |
| 97 const std::string& url, | 97 const std::string& url, |
| 98 bool allocate_info, | 98 bool allocate_info, |
| 99 int render_process_id, | 99 int render_process_id, |
| 100 int render_frame_id) { | 100 int render_frame_id) { |
| 101 net::URLRequest* request = | 101 scoped_ptr<net::URLRequest> request( |
| 102 new net::URLRequest(GURL(url), | 102 resource_context_.GetRequestContext()->CreateRequest( |
| 103 net::DEFAULT_PRIORITY, | 103 GURL(url), |
| 104 NULL, | 104 net::DEFAULT_PRIORITY, |
| 105 resource_context_.GetRequestContext()); | 105 NULL, |
| 106 NULL)); |
| 106 if (allocate_info) { | 107 if (allocate_info) { |
| 107 content::ResourceRequestInfo::AllocateForTesting( | 108 content::ResourceRequestInfo::AllocateForTesting( |
| 108 request, | 109 request.get(), |
| 109 content::RESOURCE_TYPE_SUB_FRAME, | 110 content::RESOURCE_TYPE_SUB_FRAME, |
| 110 &resource_context_, | 111 &resource_context_, |
| 111 render_process_id, | 112 render_process_id, |
| 112 render_frame_id, | 113 render_frame_id, |
| 113 MSG_ROUTING_NONE, | 114 MSG_ROUTING_NONE, |
| 114 false); | 115 false); |
| 115 } | 116 } |
| 116 return request; | 117 return request.Pass(); |
| 117 } | 118 } |
| 118 | 119 |
| 119 void SendResource(int resource_id) { | 120 void SendResource(int resource_id) { |
| 120 source()->SendResource(resource_id, callback_); | 121 source()->SendResource(resource_id, callback_); |
| 121 } | 122 } |
| 122 | 123 |
| 123 void SendJSWithOrigin( | 124 void SendJSWithOrigin( |
| 124 int resource_id, | 125 int resource_id, |
| 125 int render_process_id, | 126 int render_process_id, |
| 126 int render_frame_id) { | 127 int render_frame_id) { |
| (...skipping 26 matching lines...) Expand all Loading... |
| 153 net::TestURLRequestContext test_url_request_context_; | 154 net::TestURLRequestContext test_url_request_context_; |
| 154 content::MockResourceContext resource_context_; | 155 content::MockResourceContext resource_context_; |
| 155 scoped_ptr<TestIframeSource> source_; | 156 scoped_ptr<TestIframeSource> source_; |
| 156 content::URLDataSource::GotDataCallback callback_; | 157 content::URLDataSource::GotDataCallback callback_; |
| 157 scoped_refptr<InstantIOContext> instant_io_context_; | 158 scoped_refptr<InstantIOContext> instant_io_context_; |
| 158 scoped_refptr<base::RefCountedMemory> response_; | 159 scoped_refptr<base::RefCountedMemory> response_; |
| 159 }; | 160 }; |
| 160 | 161 |
| 161 TEST_F(IframeSourceTest, ShouldServiceRequest) { | 162 TEST_F(IframeSourceTest, ShouldServiceRequest) { |
| 162 scoped_ptr<net::URLRequest> request; | 163 scoped_ptr<net::URLRequest> request; |
| 163 request.reset(MockRequest("http://test/loader.js", true, | 164 request = MockRequest("http://test/loader.js", true, |
| 164 kNonInstantRendererPID, 0)); | 165 kNonInstantRendererPID, 0); |
| 165 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); | 166 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); |
| 166 request.reset(MockRequest("chrome-search://bogus/valid.js", true, | 167 request = MockRequest("chrome-search://bogus/valid.js", true, |
| 167 kInstantRendererPID, 0)); | 168 kInstantRendererPID, 0); |
| 168 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); | 169 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); |
| 169 request.reset(MockRequest("chrome-search://test/bogus.js", true, | 170 request = MockRequest("chrome-search://test/bogus.js", true, |
| 170 kInstantRendererPID, 0)); | 171 kInstantRendererPID, 0); |
| 171 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); | 172 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); |
| 172 request.reset(MockRequest("chrome-search://test/valid.js", true, | 173 request = MockRequest("chrome-search://test/valid.js", true, |
| 173 kInstantRendererPID, 0)); | 174 kInstantRendererPID, 0); |
| 174 EXPECT_TRUE(source()->ShouldServiceRequest(request.get())); | 175 EXPECT_TRUE(source()->ShouldServiceRequest(request.get())); |
| 175 request.reset(MockRequest("chrome-search://test/valid.js", true, | 176 request = MockRequest("chrome-search://test/valid.js", true, |
| 176 kNonInstantRendererPID, 0)); | 177 kNonInstantRendererPID, 0); |
| 177 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); | 178 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); |
| 178 request.reset(MockRequest("chrome-search://test/valid.js", true, | 179 request = MockRequest("chrome-search://test/valid.js", true, |
| 179 kInvalidRendererPID, 0)); | 180 kInvalidRendererPID, 0); |
| 180 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); | 181 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); |
| 181 } | 182 } |
| 182 | 183 |
| 183 TEST_F(IframeSourceTest, GetMimeType) { | 184 TEST_F(IframeSourceTest, GetMimeType) { |
| 184 // URLDataManagerBackend does not include / in path_and_query. | 185 // URLDataManagerBackend does not include / in path_and_query. |
| 185 EXPECT_EQ("text/html", source()->GetMimeType("foo.html")); | 186 EXPECT_EQ("text/html", source()->GetMimeType("foo.html")); |
| 186 EXPECT_EQ("application/javascript", source()->GetMimeType("foo.js")); | 187 EXPECT_EQ("application/javascript", source()->GetMimeType("foo.js")); |
| 187 EXPECT_EQ("text/css", source()->GetMimeType("foo.css")); | 188 EXPECT_EQ("text/css", source()->GetMimeType("foo.css")); |
| 188 EXPECT_EQ("image/png", source()->GetMimeType("foo.png")); | 189 EXPECT_EQ("image/png", source()->GetMimeType("foo.png")); |
| 189 EXPECT_EQ("", source()->GetMimeType("bogus")); | 190 EXPECT_EQ("", source()->GetMimeType("bogus")); |
| 190 } | 191 } |
| 191 | 192 |
| 192 TEST_F(IframeSourceTest, SendResource) { | 193 TEST_F(IframeSourceTest, SendResource) { |
| 193 SendResource(IDR_MOST_VISITED_TITLE_HTML); | 194 SendResource(IDR_MOST_VISITED_TITLE_HTML); |
| 194 EXPECT_FALSE(response_string().empty()); | 195 EXPECT_FALSE(response_string().empty()); |
| 195 } | 196 } |
| 196 | 197 |
| 197 TEST_F(IframeSourceTest, SendJSWithOrigin) { | 198 TEST_F(IframeSourceTest, SendJSWithOrigin) { |
| 198 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kInstantRendererPID, 0); | 199 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kInstantRendererPID, 0); |
| 199 EXPECT_FALSE(response_string().empty()); | 200 EXPECT_FALSE(response_string().empty()); |
| 200 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kNonInstantRendererPID, 0); | 201 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kNonInstantRendererPID, 0); |
| 201 EXPECT_FALSE(response_string().empty()); | 202 EXPECT_FALSE(response_string().empty()); |
| 202 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kInvalidRendererPID, 0); | 203 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kInvalidRendererPID, 0); |
| 203 EXPECT_TRUE(response_string().empty()); | 204 EXPECT_TRUE(response_string().empty()); |
| 204 } | 205 } |
| OLD | NEW |