Index: content/renderer/savable_resources_browsertest.cc |
diff --git a/content/renderer/savable_resources_browsertest.cc b/content/renderer/savable_resources_browsertest.cc |
index 8bf7a53926d427bc50643fc74383d58bdaaaddb1..ea991997b6f063c198c8c093ea8a07a0cec2a96e 100644 |
--- a/content/renderer/savable_resources_browsertest.cc |
+++ b/content/renderer/savable_resources_browsertest.cc |
@@ -5,14 +5,16 @@ |
#include "base/bind.h" |
#include "base/command_line.h" |
#include "base/files/file_path.h" |
+#include "content/public/browser/render_frame_host.h" |
#include "content/public/browser/web_contents.h" |
#include "content/public/common/content_switches.h" |
-#include "content/public/renderer/render_view.h" |
+#include "content/public/renderer/render_frame.h" |
#include "content/public/test/content_browser_test.h" |
#include "content/public/test/content_browser_test_utils.h" |
#include "content/renderer/savable_resources.h" |
#include "content/shell/browser/shell.h" |
#include "net/base/filename_util.h" |
+#include "third_party/WebKit/public/web/WebLocalFrame.h" |
namespace content { |
@@ -37,28 +39,22 @@ class SavableResourcesTest : public ContentBrowserTest { |
// Load the test file. |
NavigateToURL(shell(), file_url); |
- PostTaskToInProcessRendererAndWait( |
- base::Bind(&SavableResourcesTest::CheckResources, |
- base::Unretained(this), |
- page_file_path, |
- expected_resources_set, |
- file_url, |
- shell()->web_contents()->GetRoutingID())); |
+ PostTaskToInProcessRendererAndWait(base::Bind( |
+ &SavableResourcesTest::CheckResources, base::Unretained(this), |
+ page_file_path, expected_resources_set, file_url, |
+ shell()->web_contents()->GetMainFrame()->GetRoutingID())); |
} |
void CheckResources(const base::FilePath& page_file_path, |
const std::set<GURL>& expected_resources_set, |
const GURL& file_url, |
- int render_view_id) { |
+ int render_frame_routing_id) { |
// Get all savable resource links for the page. |
std::vector<GURL> resources_list; |
std::vector<GURL> referrer_urls_list; |
std::vector<blink::WebReferrerPolicy> referrer_policies_list; |
- std::vector<GURL> frames_list; |
- SavableResourcesResult result(&resources_list, |
- &referrer_urls_list, |
- &referrer_policies_list, |
- &frames_list); |
+ SavableResourcesResult result(&resources_list, &referrer_urls_list, |
+ &referrer_policies_list); |
const char* savable_schemes[] = { |
"http", |
@@ -67,23 +63,16 @@ class SavableResourcesTest : public ContentBrowserTest { |
NULL |
}; |
- RenderView* render_view = RenderView::FromRoutingID(render_view_id); |
+ RenderFrame* render_frame = |
+ RenderFrame::FromRoutingID(render_frame_routing_id); |
- ASSERT_TRUE(GetAllSavableResourceLinksForCurrentPage( |
- render_view->GetWebView(), file_url, &result, savable_schemes)); |
+ ASSERT_TRUE(GetSavableResourceLinksForFrame( |
+ render_frame->GetWebFrame(), |
+ &result, savable_schemes)); |
// Check all links of sub-resource |
- for (std::vector<GURL>::const_iterator cit = resources_list.begin(); |
- cit != resources_list.end(); ++cit) { |
- ASSERT_TRUE(expected_resources_set.find(*cit) != |
- expected_resources_set.end()); |
- } |
- |
- // Check all links of frame. |
- for (std::vector<GURL>::const_iterator cit = frames_list.begin(); |
- cit != frames_list.end(); ++cit) { |
- ASSERT_TRUE(expected_resources_set.find(*cit) != |
- expected_resources_set.end()); |
+ for (const auto& resource : resources_list) { |
+ ASSERT_TRUE(expected_resources_set.count(resource) != 0); |
} |
} |
}; |