| 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(request, | 108 content::ResourceRequestInfo::AllocateForTesting(request.get(), |
| 108 ResourceType::SUB_FRAME, | 109 ResourceType::SUB_FRAME, |
| 109 &resource_context_, | 110 &resource_context_, |
| 110 render_process_id, | 111 render_process_id, |
| 111 render_frame_id, | 112 render_frame_id, |
| 112 MSG_ROUTING_NONE, | 113 MSG_ROUTING_NONE, |
| 113 false); | 114 false); |
| 114 } | 115 } |
| 115 return request; | 116 return request.Pass(); |
| 116 } | 117 } |
| 117 | 118 |
| 118 void SendResource(int resource_id) { | 119 void SendResource(int resource_id) { |
| 119 source()->SendResource(resource_id, callback_); | 120 source()->SendResource(resource_id, callback_); |
| 120 } | 121 } |
| 121 | 122 |
| 122 void SendJSWithOrigin( | 123 void SendJSWithOrigin( |
| 123 int resource_id, | 124 int resource_id, |
| 124 int render_process_id, | 125 int render_process_id, |
| 125 int render_frame_id) { | 126 int render_frame_id) { |
| (...skipping 26 matching lines...) Expand all Loading... |
| 152 net::TestURLRequestContext test_url_request_context_; | 153 net::TestURLRequestContext test_url_request_context_; |
| 153 content::MockResourceContext resource_context_; | 154 content::MockResourceContext resource_context_; |
| 154 scoped_ptr<TestIframeSource> source_; | 155 scoped_ptr<TestIframeSource> source_; |
| 155 content::URLDataSource::GotDataCallback callback_; | 156 content::URLDataSource::GotDataCallback callback_; |
| 156 scoped_refptr<InstantIOContext> instant_io_context_; | 157 scoped_refptr<InstantIOContext> instant_io_context_; |
| 157 scoped_refptr<base::RefCountedMemory> response_; | 158 scoped_refptr<base::RefCountedMemory> response_; |
| 158 }; | 159 }; |
| 159 | 160 |
| 160 TEST_F(IframeSourceTest, ShouldServiceRequest) { | 161 TEST_F(IframeSourceTest, ShouldServiceRequest) { |
| 161 scoped_ptr<net::URLRequest> request; | 162 scoped_ptr<net::URLRequest> request; |
| 162 request.reset(MockRequest("http://test/loader.js", true, | 163 request = MockRequest("http://test/loader.js", true, |
| 163 kNonInstantRendererPID, 0)); | 164 kNonInstantRendererPID, 0); |
| 164 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); | 165 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); |
| 165 request.reset(MockRequest("chrome-search://bogus/valid.js", true, | 166 request = MockRequest("chrome-search://bogus/valid.js", true, |
| 166 kInstantRendererPID, 0)); | 167 kInstantRendererPID, 0); |
| 167 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); | 168 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); |
| 168 request.reset(MockRequest("chrome-search://test/bogus.js", true, | 169 request = MockRequest("chrome-search://test/bogus.js", true, |
| 169 kInstantRendererPID, 0)); | 170 kInstantRendererPID, 0); |
| 170 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); | 171 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); |
| 171 request.reset(MockRequest("chrome-search://test/valid.js", true, | 172 request = MockRequest("chrome-search://test/valid.js", true, |
| 172 kInstantRendererPID, 0)); | 173 kInstantRendererPID, 0); |
| 173 EXPECT_TRUE(source()->ShouldServiceRequest(request.get())); | 174 EXPECT_TRUE(source()->ShouldServiceRequest(request.get())); |
| 174 request.reset(MockRequest("chrome-search://test/valid.js", true, | 175 request = MockRequest("chrome-search://test/valid.js", true, |
| 175 kNonInstantRendererPID, 0)); | 176 kNonInstantRendererPID, 0); |
| 176 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); | 177 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); |
| 177 request.reset(MockRequest("chrome-search://test/valid.js", true, | 178 request = MockRequest("chrome-search://test/valid.js", true, |
| 178 kInvalidRendererPID, 0)); | 179 kInvalidRendererPID, 0); |
| 179 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); | 180 EXPECT_FALSE(source()->ShouldServiceRequest(request.get())); |
| 180 } | 181 } |
| 181 | 182 |
| 182 TEST_F(IframeSourceTest, GetMimeType) { | 183 TEST_F(IframeSourceTest, GetMimeType) { |
| 183 // URLDataManagerBackend does not include / in path_and_query. | 184 // URLDataManagerBackend does not include / in path_and_query. |
| 184 EXPECT_EQ("text/html", source()->GetMimeType("foo.html")); | 185 EXPECT_EQ("text/html", source()->GetMimeType("foo.html")); |
| 185 EXPECT_EQ("application/javascript", source()->GetMimeType("foo.js")); | 186 EXPECT_EQ("application/javascript", source()->GetMimeType("foo.js")); |
| 186 EXPECT_EQ("text/css", source()->GetMimeType("foo.css")); | 187 EXPECT_EQ("text/css", source()->GetMimeType("foo.css")); |
| 187 EXPECT_EQ("image/png", source()->GetMimeType("foo.png")); | 188 EXPECT_EQ("image/png", source()->GetMimeType("foo.png")); |
| 188 EXPECT_EQ("", source()->GetMimeType("bogus")); | 189 EXPECT_EQ("", source()->GetMimeType("bogus")); |
| 189 } | 190 } |
| 190 | 191 |
| 191 TEST_F(IframeSourceTest, SendResource) { | 192 TEST_F(IframeSourceTest, SendResource) { |
| 192 SendResource(IDR_MOST_VISITED_TITLE_HTML); | 193 SendResource(IDR_MOST_VISITED_TITLE_HTML); |
| 193 EXPECT_FALSE(response_string().empty()); | 194 EXPECT_FALSE(response_string().empty()); |
| 194 } | 195 } |
| 195 | 196 |
| 196 TEST_F(IframeSourceTest, SendJSWithOrigin) { | 197 TEST_F(IframeSourceTest, SendJSWithOrigin) { |
| 197 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kInstantRendererPID, 0); | 198 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kInstantRendererPID, 0); |
| 198 EXPECT_FALSE(response_string().empty()); | 199 EXPECT_FALSE(response_string().empty()); |
| 199 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kNonInstantRendererPID, 0); | 200 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kNonInstantRendererPID, 0); |
| 200 EXPECT_FALSE(response_string().empty()); | 201 EXPECT_FALSE(response_string().empty()); |
| 201 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kInvalidRendererPID, 0); | 202 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS, kInvalidRendererPID, 0); |
| 202 EXPECT_TRUE(response_string().empty()); | 203 EXPECT_TRUE(response_string().empty()); |
| 203 } | 204 } |
| OLD | NEW |