Chromium Code Reviews| Index: content/browser/frame_host/render_frame_host_manager_unittest.cc |
| diff --git a/content/browser/frame_host/render_frame_host_manager_unittest.cc b/content/browser/frame_host/render_frame_host_manager_unittest.cc |
| index 9615c1375b7f59f41be01480e52dbf5de4b9c7ba..ca7d7ec92ece3611f26011973171c6aac08f7c43 100644 |
| --- a/content/browser/frame_host/render_frame_host_manager_unittest.cc |
| +++ b/content/browser/frame_host/render_frame_host_manager_unittest.cc |
| @@ -503,7 +503,8 @@ TEST_F(RenderFrameHostManagerTest, NewTabPageProcesses) { |
| // Navigate our first tab to the chrome url and then to the destination, |
| // ensuring we grant bindings to the chrome URL. |
| NavigateActiveAndCommit(kChromeUrl); |
| - EXPECT_TRUE(active_rvh()->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| + EXPECT_TRUE(contents()->GetMainFrame()->GetEnabledBindings() & |
|
Charlie Reis
2017/01/18 22:18:43
nit: main_rfh() might be more concise?
Sam McNally
2017/01/19 05:30:37
Done.
|
| + BINDINGS_POLICY_WEB_UI); |
| NavigateActiveAndCommit(kDestUrl); |
| EXPECT_FALSE(contents()->GetPendingMainFrame()); |
| @@ -1032,8 +1033,7 @@ TEST_F(RenderFrameHostManagerTest, WebUI) { |
| // Commit. |
| manager->DidNavigateFrame(host, true); |
| - EXPECT_TRUE( |
| - host->render_view_host()->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| + EXPECT_TRUE(host->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| } |
| // Tests that we can open a WebUI link in a new tab from a WebUI page and still |
| @@ -1067,15 +1067,13 @@ TEST_F(RenderFrameHostManagerTest, WebUIInNewTab) { |
| // It should already have bindings. |
| EXPECT_EQ(host1, GetPendingFrameHost(manager1)); |
| EXPECT_NE(host1, manager1->current_frame_host()); |
| - EXPECT_TRUE( |
| - host1->render_view_host()->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| + EXPECT_TRUE(host1->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| // Commit and ensure we still have bindings. |
| manager1->DidNavigateFrame(host1, true); |
| SiteInstance* webui_instance = host1->GetSiteInstance(); |
| EXPECT_EQ(host1, manager1->current_frame_host()); |
| - EXPECT_TRUE( |
| - host1->render_view_host()->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| + EXPECT_TRUE(host1->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| // Now simulate clicking a link that opens in a new tab. |
| std::unique_ptr<TestWebContents> web_contents2( |
| @@ -1100,8 +1098,7 @@ TEST_F(RenderFrameHostManagerTest, WebUIInNewTab) { |
| EXPECT_EQ(host2, manager2->current_frame_host()); |
| EXPECT_TRUE(manager2->GetNavigatingWebUI()); |
| EXPECT_FALSE(host2->web_ui()); |
| - EXPECT_TRUE( |
| - host2->render_view_host()->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| + EXPECT_TRUE(host2->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| manager2->DidNavigateFrame(host2, true); |
| } |
| @@ -1556,7 +1553,8 @@ TEST_F(RenderFrameHostManagerTest, EnableWebUIWithSwappedOutOpener) { |
| // Ensure the RVH has WebUI bindings. |
| TestRenderViewHost* rvh1 = test_rvh(); |
| - EXPECT_TRUE(rvh1->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| + EXPECT_TRUE(rvh1->GetMainFrame()->GetEnabledBindings() & |
| + BINDINGS_POLICY_WEB_UI); |
| // Create a new tab and simulate it being the opener for the main |
| // tab. It should be in the same SiteInstance. |
| @@ -1582,7 +1580,8 @@ TEST_F(RenderFrameHostManagerTest, EnableWebUIWithSwappedOutOpener) { |
| EXPECT_FALSE(opener1_rvh->is_active()); |
| // Ensure the new RVH has WebUI bindings. |
| - EXPECT_TRUE(rvh2->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| + EXPECT_TRUE(rvh2->GetMainFrame()->GetEnabledBindings() & |
| + BINDINGS_POLICY_WEB_UI); |
| } |
| // Test that we reuse the same guest SiteInstance if we navigate across sites. |
| @@ -2142,8 +2141,7 @@ TEST_F(RenderFrameHostManagerTestWithSiteIsolation, |
| RenderFrameHostManager* main_rfhm = contents()->GetRenderManagerForTesting(); |
| RenderFrameHostImpl* webui_rfh = NavigateToEntry(main_rfhm, webui_entry); |
| EXPECT_EQ(webui_rfh, GetPendingFrameHost(main_rfhm)); |
| - EXPECT_TRUE(webui_rfh->render_view_host()->GetEnabledBindings() & |
| - BINDINGS_POLICY_WEB_UI); |
| + EXPECT_TRUE(webui_rfh->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| // Before it commits, do a cross-process navigation in a subframe. This |
| // should not grant WebUI bindings to the subframe's RVH. |
| @@ -2153,8 +2151,7 @@ TEST_F(RenderFrameHostManagerTestWithSiteIsolation, |
| base::string16() /* title */, ui::PAGE_TRANSITION_LINK, |
| false /* is_renderer_init */); |
| RenderFrameHostImpl* bar_rfh = NavigateToEntry(subframe_rfhm, subframe_entry); |
| - EXPECT_FALSE(bar_rfh->render_view_host()->GetEnabledBindings() & |
| - BINDINGS_POLICY_WEB_UI); |
| + EXPECT_FALSE(bar_rfh->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| } |
| // Test that opener proxies are created properly with a cycle on the opener |