| // Copyright (c) 2009 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/dom_ui/new_tab_ui.h" |
| #include "chrome/browser/renderer_host/test_render_view_host.h" |
| #include "chrome/common/url_constants.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| class DOMUITest : public RenderViewHostTestHarness { |
| public: |
| DOMUITest() {} |
| |
| // Tests navigating with a DOM UI from a fresh (nothing pending or committed) |
| // state, through pending, committed, then another navigation. The first page |
| // ID that we should use is passed as a parameter. We'll use the next two |
| // values. This must be increasing for the life of the tests. |
| static void DoNavigationTest(TabContents* contents, int page_id) { |
| NavigationController* controller = &contents->controller(); |
| |
| // Start a pending load. |
| GURL new_tab_url(chrome::kChromeUINewTabURL); |
| controller->LoadURL(new_tab_url, GURL(), PageTransition::LINK); |
| |
| // The navigation entry should be pending with no committed entry. |
| ASSERT_TRUE(controller->pending_entry()); |
| ASSERT_FALSE(controller->GetLastCommittedEntry()); |
| |
| // Check the things the pending DOM UI should have set. |
| EXPECT_FALSE(contents->ShouldDisplayURL()); |
| EXPECT_FALSE(contents->ShouldDisplayFavIcon()); |
| EXPECT_TRUE(contents->IsBookmarkBarAlwaysVisible()); |
| EXPECT_TRUE(contents->FocusLocationBarByDefault()); |
| |
| // Now commit the load. |
| static_cast<TestRenderViewHost*>( |
| contents->render_view_host())->SendNavigate(page_id, new_tab_url); |
| |
| // The same flags should be set as before now that the load has committed. |
| EXPECT_FALSE(contents->ShouldDisplayURL()); |
| EXPECT_FALSE(contents->ShouldDisplayFavIcon()); |
| EXPECT_TRUE(contents->IsBookmarkBarAlwaysVisible()); |
| EXPECT_TRUE(contents->FocusLocationBarByDefault()); |
| |
| // Start a pending navigation to a regular page. |
| GURL next_url("http://google.com/"); |
| controller->LoadURL(next_url, GURL(), PageTransition::LINK); |
| |
| // Check the flags. Some should reflect the new page (URL, title), some |
| // should reflect the old one (bookmark bar) until it has committed. |
| EXPECT_TRUE(contents->ShouldDisplayURL()); |
| EXPECT_TRUE(contents->ShouldDisplayFavIcon()); |
| EXPECT_TRUE(contents->IsBookmarkBarAlwaysVisible()); |
| EXPECT_FALSE(contents->FocusLocationBarByDefault()); |
| |
| // Commit the regular page load. Note that we must send it to the "pending" |
| // RenderViewHost if there is one, since this transition will also cause a |
| // process transition, and our RVH pointer will be the "committed" one. |
| // In the second call to this function from DOMUIToStandard, it won't |
| // actually be pending, which is the point of this test. |
| if (contents->render_manager()->pending_render_view_host()) { |
| static_cast<TestRenderViewHost*>( |
| contents->render_manager()->pending_render_view_host())->SendNavigate( |
| page_id + 1, next_url); |
| } else { |
| static_cast<TestRenderViewHost*>( |
| contents->render_view_host())->SendNavigate(page_id + 1, next_url); |
| } |
| |
| // The state should now reflect a regular page. |
| EXPECT_TRUE(contents->ShouldDisplayURL()); |
| EXPECT_TRUE(contents->ShouldDisplayFavIcon()); |
| EXPECT_FALSE(contents->IsBookmarkBarAlwaysVisible()); |
| EXPECT_FALSE(contents->FocusLocationBarByDefault()); |
| } |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(DOMUITest); |
| }; |
| |
| // Tests that the New Tab Page flags are correctly set and propogated by |
| // TabContents when we first navigate to a DOM UI page, then to a standard |
| // non-DOM-UI page. |
| TEST_F(DOMUITest, DOMUIToStandard) { |
| DoNavigationTest(contents(), 1); |
| |
| // Test the case where we're not doing the initial navigation. This is |
| // slightly different than the very-first-navigation case since the |
| // SiteInstance will be the same (the original TabContents must still be |
| // alive), which will trigger different behavior in RenderViewHostManager. |
| TestTabContents contents2(profile_.get(), NULL); |
| |
| DoNavigationTest(&contents2, 101); |
| } |
| |
| TEST_F(DOMUITest, DOMUIToDOMUI) { |
| // Do a load (this state is tested above). |
| GURL new_tab_url(chrome::kChromeUINewTabURL); |
| controller().LoadURL(new_tab_url, GURL(), PageTransition::LINK); |
| rvh()->SendNavigate(1, new_tab_url); |
| |
| // Start another pending load of the new tab page. |
| controller().LoadURL(new_tab_url, GURL(), PageTransition::LINK); |
| rvh()->SendNavigate(2, new_tab_url); |
| |
| // The flags should be the same as the non-pending state. |
| EXPECT_FALSE(contents()->ShouldDisplayURL()); |
| EXPECT_FALSE(contents()->ShouldDisplayFavIcon()); |
| EXPECT_TRUE(contents()->IsBookmarkBarAlwaysVisible()); |
| EXPECT_TRUE(contents()->FocusLocationBarByDefault()); |
| } |
| |
| TEST_F(DOMUITest, StandardToDOMUI) { |
| // Start a pending navigation to a regular page. |
| GURL std_url("http://google.com/"); |
| controller().LoadURL(std_url, GURL(), PageTransition::LINK); |
| |
| // The state should now reflect the default. |
| EXPECT_TRUE(contents()->ShouldDisplayURL()); |
| EXPECT_TRUE(contents()->ShouldDisplayFavIcon()); |
| EXPECT_FALSE(contents()->IsBookmarkBarAlwaysVisible()); |
| EXPECT_FALSE(contents()->FocusLocationBarByDefault()); |
| |
| // Commit the load, the state should be the same. |
| rvh()->SendNavigate(1, std_url); |
| EXPECT_TRUE(contents()->ShouldDisplayURL()); |
| EXPECT_TRUE(contents()->ShouldDisplayFavIcon()); |
| EXPECT_FALSE(contents()->IsBookmarkBarAlwaysVisible()); |
| EXPECT_FALSE(contents()->FocusLocationBarByDefault()); |
| |
| // Start a pending load for a DOMUI. |
| GURL new_tab_url(chrome::kChromeUINewTabURL); |
| controller().LoadURL(new_tab_url, GURL(), PageTransition::LINK); |
| EXPECT_FALSE(contents()->ShouldDisplayURL()); |
| EXPECT_TRUE(contents()->ShouldDisplayFavIcon()); |
| EXPECT_FALSE(contents()->IsBookmarkBarAlwaysVisible()); |
| EXPECT_TRUE(contents()->FocusLocationBarByDefault()); |
| |
| // Committing DOM UI is tested above. |
| } |