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 1ac1b088f2b653cfb57d5bdb268ab8d7b6ba5b2a..ef3403ff31e4a06a5f36a9b28251e743279f9e63 100644 |
| --- a/content/browser/frame_host/render_frame_host_manager_unittest.cc |
| +++ b/content/browser/frame_host/render_frame_host_manager_unittest.cc |
| @@ -38,6 +38,7 @@ |
| #include "content/public/test/mock_render_process_host.h" |
| #include "content/public/test/test_notification_tracker.h" |
| #include "content/public/test/test_utils.h" |
| +#include "content/test/browser_side_navigation_test_utils.h" |
| #include "content/test/test_content_browser_client.h" |
| #include "content/test/test_content_client.h" |
| #include "content/test/test_render_frame_host.h" |
| @@ -70,7 +71,8 @@ class RenderFrameHostManagerTestWebUIControllerFactory |
| : public WebUIControllerFactory { |
| public: |
| RenderFrameHostManagerTestWebUIControllerFactory() |
| - : should_create_webui_(false) { |
| + : should_create_webui_(false), type_(1) { |
| + CHECK_NE(reinterpret_cast<WebUI::TypeID>(type_), WebUI::kNoWebUI); |
| } |
| ~RenderFrameHostManagerTestWebUIControllerFactory() override {} |
| @@ -78,16 +80,34 @@ class RenderFrameHostManagerTestWebUIControllerFactory |
| should_create_webui_ = should_create_webui; |
| } |
| + // This method simulates the expectation that different WebUI instance types |
| + // would be created. The |type| value will be returned by GetWebUIType casted |
| + // to WebUI::TypeID. |
| + // As WebUI::TypeID is a typedef to void pointer, factory implementations |
| + // return values that they know to be unique to their respective cases. So |
| + // values set here should be safe if kept very low (just above zero). |
| + void set_webui_type(uintptr_t type) { |
| + CHECK_NE(reinterpret_cast<WebUI::TypeID>(type), WebUI::kNoWebUI); |
| + type_ = type; |
| + } |
| + |
| // WebUIFactory implementation. |
| WebUIController* CreateWebUIControllerForURL(WebUI* web_ui, |
| const GURL& url) const override { |
| - if (!(should_create_webui_ && HasWebUIScheme(url))) |
| - return NULL; |
| - return new WebUIController(web_ui); |
| + // If WebUI creation is enabled for the test and this is a WebUI URL, |
| + // returns a new instance. |
| + if (should_create_webui_ && HasWebUIScheme(url)) |
|
nasko
2015/11/18 23:48:03
Uh, this is weird, but probably ok at the content/
carlosk
2015/11/19 16:49:14
Acknowledged. Note this is not a regression and fo
|
| + return new WebUIController(web_ui); |
| + return nullptr; |
| } |
| WebUI::TypeID GetWebUIType(BrowserContext* browser_context, |
| const GURL& url) const override { |
| + // If WebUI creation is enabled for the test and this is a WebUI URL, |
| + // returns a mock WebUI type. |
| + if (should_create_webui_ && HasWebUIScheme(url)) { |
| + return reinterpret_cast<WebUI::TypeID>(type_); |
|
nasko
2015/11/18 23:48:04
Can we avoid adding this? Can't we just come up wi
carlosk
2015/11/19 16:49:14
A change like that will force tests to use actual
nasko
2015/11/19 19:45:24
Why would we have tests that don't use actual URLs
carlosk
2015/11/20 12:21:10
Sorry: actual *WebUI* URLs (chrome://gpu, chrome:/
|
| + } |
| return WebUI::kNoWebUI; |
| } |
| @@ -103,6 +123,7 @@ class RenderFrameHostManagerTestWebUIControllerFactory |
| private: |
| bool should_create_webui_; |
| + uintptr_t type_; |
| DISALLOW_COPY_AND_ASSIGN(RenderFrameHostManagerTestWebUIControllerFactory); |
| }; |
| @@ -280,6 +301,8 @@ class RenderFrameHostManagerTest : public RenderViewHostImplTestHarness { |
| factory_.set_should_create_webui(should_create_webui); |
| } |
| + void set_webui_type(int type) { factory_.set_webui_type(type); } |
| + |
| void NavigateActiveAndCommit(const GURL& url) { |
| // Note: we navigate the active RenderFrameHost because previous navigations |
| // won't have committed yet, so NavigateAndCommit does the wrong thing |
| @@ -411,12 +434,28 @@ class RenderFrameHostManagerTest : public RenderViewHostImplTestHarness { |
| FrameNavigationEntry* frame_entry = entry.root_node()->frame_entry.get(); |
| if (base::CommandLine::ForCurrentProcess()->HasSwitch( |
| switches::kEnableBrowserSideNavigation)) { |
| + NavigationControllerImpl* controller = |
| + static_cast<NavigationControllerImpl*>(manager->current_frame_host() |
| + ->frame_tree_node() |
| + ->navigator() |
| + ->GetController()); |
| + // TODO(carlosk): This implementation below will not work with restore |
| + // navigations. Method GetNavigationType should be exposed from |
| + // navigator_impl.cc and used here to determine FrameMsg_Navigate_Type. |
| + CHECK(entry.restore_type() == NavigationEntryImpl::RESTORE_NONE); |
| scoped_ptr<NavigationRequest> navigation_request = |
| NavigationRequest::CreateBrowserInitiated( |
| manager->frame_tree_node_, frame_entry->url(), |
| frame_entry->referrer(), *frame_entry, entry, |
| FrameMsg_Navigate_Type::NORMAL, false, base::TimeTicks::Now(), |
| - static_cast<NavigationControllerImpl*>(&controller())); |
| + controller); |
| + |
| + // Simulates request creation that triggers the 1st internal call to |
| + // GetFrameHostForNavigation. |
| + manager->DidCreateNavigationRequest(*navigation_request); |
| + |
| + // And also simulates the 2nd and final call to GetFrameHostForNavigation |
| + // that determines the final frame that will commit the navigation. |
| TestRenderFrameHost* frame_host = static_cast<TestRenderFrameHost*>( |
| manager->GetFrameHostForNavigation(*navigation_request)); |
| CHECK(frame_host); |
| @@ -1089,7 +1128,7 @@ TEST_F(RenderFrameHostManagerTest, WebUI) { |
| RenderFrameHostImpl* initial_rfh = manager->current_frame_host(); |
| EXPECT_FALSE(manager->current_host()->IsRenderViewLive()); |
| - EXPECT_FALSE(manager->web_ui()); |
| + EXPECT_FALSE(manager->current_frame_host()->web_ui()); |
| EXPECT_TRUE(initial_rfh); |
| const GURL kUrl("chrome://foo"); |
| @@ -1116,13 +1155,8 @@ TEST_F(RenderFrameHostManagerTest, WebUI) { |
| // The Web UI is committed immediately because the RenderViewHost has not been |
| // used yet. UpdateStateForNavigate() took the short cut path. |
| - if (base::CommandLine::ForCurrentProcess()->HasSwitch( |
| - switches::kEnableBrowserSideNavigation)) { |
| - EXPECT_FALSE(manager->speculative_web_ui()); |
| - } else { |
| - EXPECT_FALSE(manager->pending_web_ui()); |
| - } |
| - EXPECT_TRUE(manager->web_ui()); |
| + EXPECT_FALSE(manager->GetNavigatingWebUI()); |
| + EXPECT_TRUE(manager->current_frame_host()->web_ui()); |
| // Commit. |
| manager->DidNavigateFrame(host, true); |
| @@ -1191,12 +1225,8 @@ TEST_F(RenderFrameHostManagerTest, WebUIInNewTab) { |
| // No cross-process transition happens because we are already in the right |
| // SiteInstance. We should grant bindings immediately. |
| EXPECT_EQ(host2, manager2->current_frame_host()); |
| - if (base::CommandLine::ForCurrentProcess()->HasSwitch( |
| - switches::kEnableBrowserSideNavigation)) { |
| - EXPECT_TRUE(manager2->speculative_web_ui()); |
| - } else { |
| - EXPECT_TRUE(manager2->pending_web_ui()); |
| - } |
| + EXPECT_TRUE(manager2->GetNavigatingWebUI()); |
| + EXPECT_FALSE(host2->web_ui()); |
| EXPECT_TRUE( |
| host2->render_view_host()->GetEnabledBindings() & BINDINGS_POLICY_WEB_UI); |
| @@ -1210,16 +1240,14 @@ TEST_F(RenderFrameHostManagerTest, WebUIWasReused) { |
| // Navigate to a WebUI page. |
| const GURL kUrl1("chrome://foo"); |
| contents()->NavigateAndCommit(kUrl1); |
| - RenderFrameHostManager* manager = |
| - main_test_rfh()->frame_tree_node()->render_manager(); |
| - WebUIImpl* web_ui = manager->web_ui(); |
| + WebUIImpl* web_ui = main_test_rfh()->web_ui(); |
| EXPECT_TRUE(web_ui); |
| // Navigate to another WebUI page which should be same-site and keep the |
| // current WebUI. |
| const GURL kUrl2("chrome://foo/bar"); |
| contents()->NavigateAndCommit(kUrl2); |
| - EXPECT_EQ(web_ui, manager->web_ui()); |
| + EXPECT_EQ(web_ui, main_test_rfh()->web_ui()); |
| } |
| // Tests that a WebUI is correctly cleaned up when navigating from a chrome:// |
| @@ -1230,12 +1258,12 @@ TEST_F(RenderFrameHostManagerTest, WebUIWasCleared) { |
| // Navigate to a WebUI page. |
| const GURL kUrl1("chrome://foo"); |
| contents()->NavigateAndCommit(kUrl1); |
| - EXPECT_TRUE(main_test_rfh()->frame_tree_node()->render_manager()->web_ui()); |
| + EXPECT_TRUE(main_test_rfh()->web_ui()); |
| // Navigate to a non-WebUI page. |
| const GURL kUrl2("http://www.google.com"); |
| contents()->NavigateAndCommit(kUrl2); |
| - EXPECT_FALSE(main_test_rfh()->frame_tree_node()->render_manager()->web_ui()); |
| + EXPECT_FALSE(main_test_rfh()->web_ui()); |
| } |
| // Tests that we don't end up in an inconsistent state if a page does a back and |
| @@ -2662,4 +2690,238 @@ TEST_F(RenderFrameHostManagerTest, |
| VerifyPageFocusMessage(hostC->GetProcess(), true, proxy->GetRoutingID()); |
| } |
| +// Checks that a restore navigation to a WebUI works. |
| +TEST_F(RenderFrameHostManagerTest, RestoreNavigationToWebUI) { |
| + set_should_create_webui(true); |
| + |
| + const GURL init_kUrl("chrome://foo/"); |
|
nasko
2015/11/18 23:48:03
This variable name doesn't really follow any singl
carlosk
2015/11/19 16:49:14
Done.
|
| + SiteInstanceImpl* init_instance = |
| + static_cast<SiteInstanceImpl*>(SiteInstance::Create(browser_context())); |
| + init_instance->SetSite(init_kUrl); |
| + scoped_ptr<TestWebContents> web_contents( |
| + TestWebContents::Create(browser_context(), init_instance)); |
| + RenderFrameHostManager* manager = web_contents->GetRenderManagerForTesting(); |
| + NavigationControllerImpl& controller = web_contents->GetController(); |
| + |
| + // Setup a restored entry. |
| + std::vector<scoped_ptr<NavigationEntry>> entries; |
| + scoped_ptr<NavigationEntry> new_entry = |
| + NavigationControllerImpl::CreateNavigationEntry( |
| + init_kUrl, Referrer(), ui::PAGE_TRANSITION_TYPED, false, |
| + std::string(), browser_context()); |
| + new_entry->SetPageID(0); |
| + entries.push_back(new_entry.Pass()); |
| + controller.Restore( |
| + 0, NavigationController::RESTORE_LAST_SESSION_EXITED_CLEANLY, &entries); |
| + ASSERT_EQ(0u, entries.size()); |
| + ASSERT_EQ(1, controller.GetEntryCount()); |
| + |
| + RenderFrameHostImpl* initial_host = manager->current_frame_host(); |
| + ASSERT_TRUE(initial_host); |
| + EXPECT_FALSE(initial_host->IsRenderFrameLive()); |
| + EXPECT_FALSE(initial_host->web_ui()); |
| + |
| + // Navigation request for a same-session restore navigation. |
|
nasko
2015/11/18 23:48:03
What is "same-session" navigation?
carlosk
2015/11/19 16:49:14
This comment was outdated. It meant this should be
|
| + NavigationEntryImpl entry(nullptr /* instance */, 0 /* page_id */, init_kUrl, |
| + Referrer(), base::string16() /* title */, |
| + ui::PAGE_TRANSITION_RELOAD, |
| + false /* is_renderer_init */); |
| + entry.set_restore_type( |
| + NavigationEntryImpl::RESTORE_LAST_SESSION_EXITED_CLEANLY); |
| + NavigateToEntry(manager, entry); |
| + |
| + // As the initial renderer was not live, the new RenderFrameHost should be |
| + // made immediately active at request time. |
| + EXPECT_FALSE(GetPendingFrameHost(manager)); |
| + TestRenderFrameHost* host = |
|
nasko
2015/11/18 23:48:04
nit: If you are calling the first RFH init_host, t
carlosk
2015/11/19 16:49:14
Done.
|
| + static_cast<TestRenderFrameHost*>(manager->current_frame_host()); |
| + ASSERT_TRUE(host); |
| + EXPECT_EQ(host, initial_host); |
| + EXPECT_TRUE(host->IsRenderFrameLive()); |
| + WebUIImpl* web_ui = manager->GetNavigatingWebUI(); |
| + EXPECT_TRUE(web_ui); |
| + EXPECT_FALSE(host->web_ui()); |
| + |
| + manager->DidNavigateFrame(host, true); |
| + EXPECT_EQ(host, manager->current_frame_host()); |
| + EXPECT_EQ(web_ui, host->web_ui()); |
| +} |
| + |
| +// Creates a test class for PlzNavigate tests. |
| +class RenderFrameHostManagerTestWithBrowserSideNavigation |
| + : public RenderFrameHostManagerTest { |
| + public: |
| + void SetUp() override { |
| + EnableBrowserSideNavigation(); |
| + RenderFrameHostManagerTest::SetUp(); |
| + } |
| +}; |
| + |
| +// PlzNavigate: Tests that the correct intermediary and final navigation states |
| +// are reached when navigating from a renderer that is not live to a WebUI URL. |
| +TEST_F(RenderFrameHostManagerTestWithBrowserSideNavigation, |
| + NavigateFromDeadRendererToWebUI) { |
| + set_should_create_webui(true); |
| + RenderViewHostChangedObserver change_observer(contents()); |
|
nasko
2015/11/18 23:48:04
Why do we care about RVH? I'd rather not add more
carlosk
2015/11/19 16:49:14
I don't think it makes sense to track the RFH eith
|
| + RenderFrameHostManager* manager = contents()->GetRenderManagerForTesting(); |
| + |
| + RenderFrameHostImpl* initial_host = manager->current_frame_host(); |
| + ASSERT_TRUE(initial_host); |
| + EXPECT_FALSE(initial_host->IsRenderFrameLive()); |
| + |
| + // Navigation request. |
| + const GURL kUrl("chrome://foo"); |
| + NavigationEntryImpl entry(nullptr /* instance */, -1 /* page_id */, kUrl, |
| + Referrer(), base::string16() /* title */, |
| + ui::PAGE_TRANSITION_TYPED, |
| + false /* is_renderer_init */); |
| + FrameNavigationEntry* frame_entry = entry.root_node()->frame_entry.get(); |
| + scoped_ptr<NavigationRequest> navigation_request = |
| + NavigationRequest::CreateBrowserInitiated( |
| + contents()->GetFrameTree()->root(), frame_entry->url(), |
| + frame_entry->referrer(), *frame_entry, entry, |
| + FrameMsg_Navigate_Type::NORMAL, false, base::TimeTicks::Now(), |
| + static_cast<NavigationControllerImpl*>(&controller())); |
| + manager->DidCreateNavigationRequest(*navigation_request); |
| + |
| + // As the initial renderer was not live, the new RenderFrameHost should be |
| + // made immediately active at request time. |
| + EXPECT_TRUE(change_observer.DidHostChange()); |
| + EXPECT_FALSE(GetPendingFrameHost(manager)); |
| + TestRenderFrameHost* host = |
| + static_cast<TestRenderFrameHost*>(manager->current_frame_host()); |
| + ASSERT_TRUE(host); |
| + EXPECT_NE(host, initial_host); |
| + EXPECT_TRUE(host->IsRenderFrameLive()); |
| + WebUIImpl* web_ui = host->web_ui(); |
| + EXPECT_TRUE(web_ui); |
| + |
| + // Prepare to commit, update the navigating RenderFrameHost. |
| + EXPECT_EQ(host, manager->GetFrameHostForNavigation(*navigation_request)); |
| + |
| + // No pending RenderFrameHost as the current one should be reused. |
| + EXPECT_FALSE(GetPendingFrameHost(manager)); |
| + |
| + // But there should be a pending WebUI set to re-use the current one. |
|
nasko
2015/11/18 23:48:03
nit: reuse
carlosk
2015/11/19 16:49:14
Done.
|
| + EXPECT_EQ(web_ui, host->web_ui()); |
|
nasko
2015/11/18 23:48:04
If there should be a pending WebUI, why aren't you
carlosk
2015/11/19 16:49:14
Yes, this was indeed missing here and all over. I
|
| + |
| + // The RenderFrameHost committed. |
| + manager->DidNavigateFrame(host, true); |
| + EXPECT_FALSE(change_observer.DidHostChange()); |
| + EXPECT_EQ(host, manager->current_frame_host()); |
| + EXPECT_FALSE(GetPendingFrameHost(manager)); |
| + EXPECT_EQ(web_ui, host->web_ui()); |
| +} |
| + |
| +// PlzNavigate: Tests that the correct intermediary and final navigation states |
| +// are reached when navigating same-site between two WebUIs of the same type. |
| +TEST_F(RenderFrameHostManagerTestWithBrowserSideNavigation, |
| + NavigateSameSiteBetweenWebUIs) { |
| + set_should_create_webui(true); |
| + NavigateActiveAndCommit(GURL("chrome://foo")); |
| + |
| + RenderFrameHostManager* manager = contents()->GetRenderManagerForTesting(); |
| + TestRenderFrameHost* host = |
| + static_cast<TestRenderFrameHost*>(manager->current_frame_host()); |
| + EXPECT_TRUE(host->IsRenderFrameLive()); |
| + WebUIImpl* web_ui = host->web_ui(); |
| + EXPECT_TRUE(web_ui); |
| + |
| + RenderViewHostChangedObserver change_observer(contents()); |
|
nasko
2015/11/18 23:48:03
Same comment as above. If we care about such host
carlosk
2015/11/19 16:49:14
Done.
|
| + |
| + // Navigation request. No change in the returned WebUI type. |
| + const GURL kUrl("chrome://foo/bar"); |
| + NavigationEntryImpl entry(nullptr /* instance */, -1 /* page_id */, kUrl, |
| + Referrer(), base::string16() /* title */, |
| + ui::PAGE_TRANSITION_TYPED, |
| + false /* is_renderer_init */); |
| + FrameNavigationEntry* frame_entry = entry.root_node()->frame_entry.get(); |
| + scoped_ptr<NavigationRequest> navigation_request = |
| + NavigationRequest::CreateBrowserInitiated( |
| + contents()->GetFrameTree()->root(), frame_entry->url(), |
| + frame_entry->referrer(), *frame_entry, entry, |
| + FrameMsg_Navigate_Type::NORMAL, false, base::TimeTicks::Now(), |
| + static_cast<NavigationControllerImpl*>(&controller())); |
| + manager->DidCreateNavigationRequest(*navigation_request); |
| + |
| + // The current WebUI should still be in place and there should be a new |
| + // pending WebUI instance in the current RenderFrameHost. |
| + EXPECT_FALSE(change_observer.DidHostChange()); |
| + EXPECT_FALSE(GetPendingFrameHost(manager)); |
| + EXPECT_EQ(web_ui, host->web_ui()); |
| + |
| + // Prepare to commit, update the navigating RenderFrameHost. |
| + EXPECT_EQ(host, manager->GetFrameHostForNavigation(*navigation_request)); |
| + |
| + EXPECT_FALSE(GetPendingFrameHost(manager)); |
| + EXPECT_EQ(web_ui, host->web_ui()); |
| + |
| + // The RenderFrameHost committed. |
| + manager->DidNavigateFrame(host, true); |
| + EXPECT_FALSE(change_observer.DidHostChange()); |
| + EXPECT_EQ(web_ui, host->web_ui()); |
| +} |
| + |
| +// PlzNavigate: Tests that the correct intermediary and final navigation states |
| +// are reached when navigating cross-site between two different WebUI types. |
| +TEST_F(RenderFrameHostManagerTestWithBrowserSideNavigation, |
| + NavigateCrossSiteBetweenWebUIs) { |
| + // Cross-site navigations will always cause the change of the WebUI instance |
| + // but for consistency sake different types will be set for each navigation. |
| + set_should_create_webui(true); |
| + set_webui_type(1); |
| + NavigateActiveAndCommit(GURL("chrome://foo")); |
| + |
| + RenderFrameHostManager* manager = contents()->GetRenderManagerForTesting(); |
| + TestRenderFrameHost* host = |
| + static_cast<TestRenderFrameHost*>(manager->current_frame_host()); |
| + EXPECT_TRUE(host->IsRenderFrameLive()); |
| + EXPECT_TRUE(host->web_ui()); |
| + |
| + RenderViewHostChangedObserver change_observer(contents()); |
| + set_webui_type(2); |
| + |
| + // Navigation request. |
| + const GURL kUrl("chrome://bar"); |
| + NavigationEntryImpl entry(nullptr /* instance */, -1 /* page_id */, kUrl, |
| + Referrer(), base::string16() /* title */, |
| + ui::PAGE_TRANSITION_TYPED, |
| + false /* is_renderer_init */); |
| + FrameNavigationEntry* frame_entry = entry.root_node()->frame_entry.get(); |
| + scoped_ptr<NavigationRequest> navigation_request = |
| + NavigationRequest::CreateBrowserInitiated( |
| + contents()->GetFrameTree()->root(), frame_entry->url(), |
| + frame_entry->referrer(), *frame_entry, entry, |
| + FrameMsg_Navigate_Type::NORMAL, false, base::TimeTicks::Now(), |
| + static_cast<NavigationControllerImpl*>(&controller())); |
| + manager->DidCreateNavigationRequest(*navigation_request); |
| + |
| + // The current WebUI should still be in place and there should be a new |
| + // active WebUI instance in the speculative RenderFrameHost. |
| + EXPECT_FALSE(change_observer.DidHostChange()); |
| + TestRenderFrameHost* speculative_host = |
| + static_cast<TestRenderFrameHost*>(GetPendingFrameHost(manager)); |
| + EXPECT_TRUE(speculative_host); |
| + EXPECT_TRUE(manager->current_frame_host()->web_ui()); |
| + WebUIImpl* next_web_ui = speculative_host->web_ui(); |
| + EXPECT_TRUE(next_web_ui); |
| + EXPECT_NE(next_web_ui, manager->current_frame_host()->web_ui()); |
| + |
| + // Prepare to commit, update the navigating RenderFrameHost. |
| + EXPECT_EQ(speculative_host, |
| + manager->GetFrameHostForNavigation(*navigation_request)); |
| + |
| + EXPECT_EQ(speculative_host, GetPendingFrameHost(manager)); |
| + EXPECT_TRUE(manager->current_frame_host()->web_ui()); |
| + EXPECT_NE(next_web_ui, manager->current_frame_host()->web_ui()); |
| + EXPECT_EQ(next_web_ui, speculative_host->web_ui()); |
| + |
| + // The RenderFrameHost committed. |
| + manager->DidNavigateFrame(speculative_host, true); |
| + EXPECT_EQ(speculative_host, manager->current_frame_host()); |
| + EXPECT_TRUE(change_observer.DidHostChange()); |
| + EXPECT_EQ(next_web_ui, manager->current_frame_host()->web_ui()); |
| + EXPECT_FALSE(GetPendingFrameHost(manager)); |
| +} |
| + |
| } // namespace content |