Index: chrome/browser/tab_contents/view_source_browsertest.cc |
=================================================================== |
--- chrome/browser/tab_contents/view_source_browsertest.cc (revision 144460) |
+++ chrome/browser/tab_contents/view_source_browsertest.cc (working copy) |
@@ -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 @@ |
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 @@ |
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())); |
} |