1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "base/strings/utf_string_conversions.h" 6 #include "chrome/app/chrome_command_ids.h" 7 #include "chrome/browser/ui/browser.h" 8 #include "chrome/browser/ui/browser_commands.h" 9 #include "chrome/browser/ui/tabs/tab_strip_model.h" 10 #include "chrome/common/url_constants.h" 11 #include "chrome/test/base/in_process_browser_test.h" 12 #include "chrome/test/base/ui_test_utils.h" 13 #include "content/public/browser/navigation_entry.h" 14 #include "content/public/browser/notification_service.h" 15 #include "content/public/browser/notification_types.h" 16 #include "content/public/browser/render_view_host.h" 17 #include "content/public/browser/web_contents.h" 18 #include "content/public/test/browser_test_utils.h" 19 #include "net/test/embedded_test_server/embedded_test_server.h" 20 #include "url/gurl.h" 21 22 namespace { 23 const char kTestHtml[] = "/viewsource/test.html"; 24 const char kTestMedia[] = "files/media/pink_noise_140ms.wav"; 25 } 26 27 typedef InProcessBrowserTest ViewSourceTest; 28 29 // This test renders a page in view-source and then checks to see if the title 30 // set in the html was set successfully (it shouldn't because we rendered the 31 // page in view source). 32 // Flaky; see http://crbug.com/72201. 33 IN_PROC_BROWSER_TEST_F(ViewSourceTest, DoesBrowserRenderInViewSource) { 34 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); 35 36 // First we navigate to our view-source test page. 37 GURL url(content::kViewSourceScheme + std::string(":") + 38 embedded_test_server()->GetURL(kTestHtml).spec()); 39 ui_test_utils::NavigateToURL(browser(), url); 40 41 // Check that the title didn't get set. It should not be there (because we 42 // are in view-source mode). 43 EXPECT_NE(ASCIIToUTF16("foo"), 44 browser()->tab_strip_model()->GetActiveWebContents()->GetTitle()); 45 } 46 47 // This test renders a page normally and then renders the same page in 48 // view-source mode. This is done since we had a problem at one point during 49 // implementation of the view-source: prefix being consumed (removed from the 50 // URL) if the URL was not changed (apart from adding the view-source prefix) 51 IN_PROC_BROWSER_TEST_F(ViewSourceTest, DoesBrowserConsumeViewSourcePrefix) { 52 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); 53 54 // First we navigate to google.html. 55 GURL url(embedded_test_server()->GetURL(kTestHtml)); 56 ui_test_utils::NavigateToURL(browser(), url); 57 58 // Then we navigate to the same url but with the "view-source:" prefix. 59 GURL url_viewsource(content::kViewSourceScheme + std::string(":") + 60 url.spec()); 61 ui_test_utils::NavigateToURL(browser(), url_viewsource); 62 63 // The URL should still be prefixed with "view-source:". 64 EXPECT_EQ(url_viewsource.spec(), 65 browser()->tab_strip_model()->GetActiveWebContents()-> 66 GetURL().spec()); 67 } 68 69 // Make sure that when looking at the actual page, we can select "View Source" 70 // from the menu. 71 IN_PROC_BROWSER_TEST_F(ViewSourceTest, ViewSourceInMenuEnabledOnANormalPage) { 72 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); 73 74 GURL url(embedded_test_server()->GetURL(kTestHtml)); 75 ui_test_utils::NavigateToURL(browser(), url); 76 77 EXPECT_TRUE(chrome::CanViewSource(browser())); 78 } 79 80 // For page that is media content, make sure that we cannot select "View Source" 81 // See http://crbug.com/83714 82 IN_PROC_BROWSER_TEST_F(ViewSourceTest, ViewSourceInMenuDisabledOnAMediaPage) { 83 ASSERT_TRUE(test_server()->Start()); 84 85 GURL url(test_server()->GetURL(kTestMedia)); 86 ui_test_utils::NavigateToURL(browser(), url); 87 88 const char* mime_type = browser()->tab_strip_model()->GetActiveWebContents()-> 89 GetContentsMimeType().c_str(); 90 91 EXPECT_STREQ("audio/wav", mime_type); 92 EXPECT_FALSE(chrome::CanViewSource(browser())); 93 } 94 95 // Make sure that when looking at the page source, we can't select "View Source" 96 // from the menu. 97 IN_PROC_BROWSER_TEST_F(ViewSourceTest, 98 ViewSourceInMenuDisabledWhileViewingSource) { 99 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); 100 101 GURL url_viewsource(content::kViewSourceScheme + std::string(":") + 102 embedded_test_server()->GetURL(kTestHtml).spec()); 103 ui_test_utils::NavigateToURL(browser(), url_viewsource); 104 105 EXPECT_FALSE(chrome::CanViewSource(browser())); 106 } 107 108 // Tests that reload initiated by the script on the view-source page leaves 109 // the page in view-source mode. 110 // Times out on Mac, Windows, ChromeOS Linux: crbug.com/162080 111 IN_PROC_BROWSER_TEST_F(ViewSourceTest, DISABLED_TestViewSourceReload) { 112 ASSERT_TRUE(embedded_test_server()->InitializeAndWaitUntilReady()); 113 114 GURL url_viewsource(content::kViewSourceScheme + std::string(":") + 115 embedded_test_server()->GetURL(kTestHtml).spec()); 116 117 content::WindowedNotificationObserver observer( 118 content::NOTIFICATION_LOAD_STOP, 119 content::NotificationService::AllSources()); 120 ui_test_utils::NavigateToURL(browser(), url_viewsource); 121 observer.Wait(); 122 123 ASSERT_TRUE( 124 content::ExecuteScript(browser()->tab_strip_model()->GetWebContentsAt(0), 125 "window.location.reload();")); 126 127 content::WindowedNotificationObserver observer2( 128 content::NOTIFICATION_LOAD_STOP, 129 content::NotificationService::AllSources()); 130 observer2.Wait(); 131 ASSERT_TRUE(browser()->tab_strip_model()->GetWebContentsAt(0)-> 132 GetController().GetActiveEntry()->IsViewSourceMode()); 133 } 134