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 <memory> | 7 #include <memory> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/memory/ref_counted_memory.h" | 10 #include "base/memory/ref_counted_memory.h" |
11 #include "base/message_loop/message_loop.h" | 11 #include "base/message_loop/message_loop.h" |
12 #include "chrome/browser/search/instant_io_context.h" | 12 #include "chrome/browser/search/instant_io_context.h" |
13 #include "chrome/grit/browser_resources.h" | 13 #include "chrome/grit/browser_resources.h" |
14 #include "content/public/browser/browser_thread.h" | 14 #include "content/public/browser/browser_thread.h" |
15 #include "content/public/browser/resource_request_info.h" | 15 #include "content/public/browser/resource_request_info.h" |
16 #include "content/public/common/previews_state.h" | 16 #include "content/public/common/previews_state.h" |
17 #include "content/public/test/mock_resource_context.h" | 17 #include "content/public/test/mock_resource_context.h" |
18 #include "content/public/test/test_browser_thread_bundle.h" | 18 #include "content/public/test/test_browser_thread_bundle.h" |
19 #include "ipc/ipc_message.h" | 19 #include "ipc/ipc_message.h" |
20 #include "net/base/request_priority.h" | 20 #include "net/base/request_priority.h" |
| 21 #include "net/traffic_annotation/network_traffic_annotation_test_helper.h" |
21 #include "net/url_request/url_request.h" | 22 #include "net/url_request/url_request.h" |
22 #include "net/url_request/url_request_context.h" | 23 #include "net/url_request/url_request_context.h" |
23 #include "net/url_request/url_request_test_util.h" | 24 #include "net/url_request/url_request_test_util.h" |
24 #include "testing/gtest/include/gtest/gtest.h" | 25 #include "testing/gtest/include/gtest/gtest.h" |
25 #include "url/gurl.h" | 26 #include "url/gurl.h" |
26 | 27 |
27 const int kNonInstantRendererPID = 0; | 28 const int kNonInstantRendererPID = 0; |
28 const char kNonInstantOrigin[] = "http://evil"; | 29 const char kNonInstantOrigin[] = "http://evil"; |
29 const int kInstantRendererPID = 1; | 30 const int kInstantRendererPID = 1; |
30 const char kInstantOrigin[] = "chrome-search://instant"; | 31 const char kInstantOrigin[] = "chrome-search://instant"; |
(...skipping 54 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
85 if (response_.get()) { | 86 if (response_.get()) { |
86 return std::string(response_->front_as<char>(), response_->size()); | 87 return std::string(response_->front_as<char>(), response_->size()); |
87 } | 88 } |
88 return ""; | 89 return ""; |
89 } | 90 } |
90 | 91 |
91 std::unique_ptr<net::URLRequest> MockRequest(const std::string& url, | 92 std::unique_ptr<net::URLRequest> MockRequest(const std::string& url, |
92 int render_process_id) { | 93 int render_process_id) { |
93 std::unique_ptr<net::URLRequest> request( | 94 std::unique_ptr<net::URLRequest> request( |
94 resource_context_.GetRequestContext()->CreateRequest( | 95 resource_context_.GetRequestContext()->CreateRequest( |
95 GURL(url), net::DEFAULT_PRIORITY, NULL)); | 96 GURL(url), net::DEFAULT_PRIORITY, NULL, |
| 97 TRAFFIC_ANNOTATION_FOR_TESTS)); |
96 content::ResourceRequestInfo::AllocateForTesting( | 98 content::ResourceRequestInfo::AllocateForTesting( |
97 request.get(), content::RESOURCE_TYPE_SUB_FRAME, &resource_context_, | 99 request.get(), content::RESOURCE_TYPE_SUB_FRAME, &resource_context_, |
98 render_process_id, MSG_ROUTING_NONE, MSG_ROUTING_NONE, | 100 render_process_id, MSG_ROUTING_NONE, MSG_ROUTING_NONE, |
99 /*is_main_frame=*/false, | 101 /*is_main_frame=*/false, |
100 /*parent_is_main_frame=*/false, | 102 /*parent_is_main_frame=*/false, |
101 /*allow_download=*/true, | 103 /*allow_download=*/true, |
102 /*is_async=*/false, content::PREVIEWS_OFF); | 104 /*is_async=*/false, content::PREVIEWS_OFF); |
103 return request; | 105 return request; |
104 } | 106 } |
105 | 107 |
(...skipping 77 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
183 source()->set_origin(kInstantOrigin); | 185 source()->set_origin(kInstantOrigin); |
184 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS); | 186 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS); |
185 EXPECT_FALSE(response_string().empty()); | 187 EXPECT_FALSE(response_string().empty()); |
186 source()->set_origin(kNonInstantOrigin); | 188 source()->set_origin(kNonInstantOrigin); |
187 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS); | 189 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS); |
188 EXPECT_FALSE(response_string().empty()); | 190 EXPECT_FALSE(response_string().empty()); |
189 source()->set_origin(std::string()); | 191 source()->set_origin(std::string()); |
190 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS); | 192 SendJSWithOrigin(IDR_MOST_VISITED_TITLE_JS); |
191 EXPECT_TRUE(response_string().empty()); | 193 EXPECT_TRUE(response_string().empty()); |
192 } | 194 } |
OLD | NEW |