diff options
Diffstat (limited to 'chrome/browser/tab_contents/view_source_browsertest.cc')
-rw-r--r-- | chrome/browser/tab_contents/view_source_browsertest.cc | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/chrome/browser/tab_contents/view_source_browsertest.cc b/chrome/browser/tab_contents/view_source_browsertest.cc index 85299d2..7084bdc 100644 --- a/chrome/browser/tab_contents/view_source_browsertest.cc +++ b/chrome/browser/tab_contents/view_source_browsertest.cc @@ -5,6 +5,7 @@ #include "base/utf_string_conversions.h" #include "chrome/app/chrome_command_ids.h" #include "chrome/browser/ui/browser.h" +#include "chrome/browser/ui/browser_commands.h" #include "chrome/common/url_constants.h" #include "chrome/test/base/in_process_browser_test.h" #include "chrome/test/base/ui_test_utils.h" @@ -65,7 +66,7 @@ IN_PROC_BROWSER_TEST_F(ViewSourceTest, ViewSourceInMenuEnabledOnANormalPage) { GURL url(test_server()->GetURL(kTestHtml)); ui_test_utils::NavigateToURL(browser(), url); - EXPECT_TRUE(browser()->command_updater()->IsCommandEnabled(IDC_VIEW_SOURCE)); + EXPECT_TRUE(chrome::CanViewSource(browser())); } // Make sure that when looking at the page source, we can't select "View Source" @@ -80,5 +81,5 @@ IN_PROC_BROWSER_TEST_F(ViewSourceTest, test_server()->GetURL(kTestHtml).spec()); ui_test_utils::NavigateToURL(browser(), url_viewsource); - EXPECT_FALSE(browser()->command_updater()->IsCommandEnabled(IDC_VIEW_SOURCE)); + EXPECT_FALSE(chrome::CanViewSource(browser())); } |