NTP: Hide "Make this my home page" after setting the home page....

chrome/browser/dom_ui/new_tab_ui_uitest.cc

 // 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/test/ui/ui_test.h"
 
 #include "base/file_path.h"
 #include "chrome/app/chrome_dll_resource.h"
 #include "chrome/browser/dom_ui/new_tab_ui.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/pref_service.h"
 #include "chrome/test/automation/browser_proxy.h"
 #include "chrome/test/automation/tab_proxy.h"
 #include "chrome/test/automation/window_proxy.h"
 
 class NewTabUITest : public UITest {
  public:
   NewTabUITest() {
     dom_automation_enabled_ = true;
+    // Set home page to the empty string so that we can set the home page using
+    // preferences.
+    homepage_ = L"";
 
     // Setup the DEFAULT_THEME profile (has fake history entries).
     set_template_user_data(UITest::ComputeTypicalUserDataSource(
         UITest::DEFAULT_THEME));
   }
 };
 
 TEST_F(NewTabUITest, NTPHasThumbnails) {
   // Switch to the "new tab" tab, which should be any new tab after the
   // first (the first is about:blank).
@@ skipping 66 matching lines @@
   ASSERT_EQ(0, prefs.GetInteger(prefs::kNTPPrefVersion));
 
   bool migrated = NewTabUI::UpdateUserPrefsVersion(&prefs);
   ASSERT_TRUE(migrated);
   ASSERT_EQ(NewTabUI::current_pref_version(),
             prefs.GetInteger(prefs::kNTPPrefVersion));
 
   migrated = NewTabUI::UpdateUserPrefsVersion(&prefs);
   ASSERT_FALSE(migrated);
 }
+
+TEST_F(NewTabUITest, HomePageLink) {
+  scoped_refptr<BrowserProxy> browser(automation()->GetBrowserWindow(0));
+  ASSERT_TRUE(browser.get());
+
+  ASSERT_TRUE(
+      browser->SetBooleanPreference(prefs::kHomePageIsNewTabPage, false));
+
+  int tab_count = -1;
+  ASSERT_TRUE(browser->GetTabCount(&tab_count));
+  ASSERT_EQ(1, tab_count);
+
+  // Bring up a new tab page.
+  browser->ApplyAccelerator(IDC_NEW_TAB);
+  WaitUntilTabCount(2);
+  int load_time;
+  ASSERT_TRUE(automation()->WaitForInitialNewTabUILoad(&load_time));
+
+  scoped_refptr<TabProxy> tab = browser->GetActiveTab();
+
+  // TODO(arv): Extract common patterns for doing js testing.
+
+  // Fire click
+  // TODO(arv): Find screen position of element and use a lower level click
+  // emulation.
+  bool result;
+  ASSERT_TRUE(tab->ExecuteAndExtractBool(L"",
+    L"window.domAutomationController.send("
+    L"(function() {"
+    L"  var e = document.createEvent('Event');"
+    L"  e.initEvent('click', true, true);"
+    L"  var el = document.querySelector('#set-as-home-page > *');"
+    L"  el.dispatchEvent(e);"
+    L"  return true;"
+    L"})()"
+    L")",
+    &result));
+  ASSERT_TRUE(result);
+
+  // Make sure set as home page element is hidden.
+  std::wstring style_display;
+  ASSERT_TRUE(tab->ExecuteAndExtractString(L"",
+    L"window.domAutomationController.send("
+    L"(function() {"
+    L"  var el = document.querySelector('#set-as-home-page');"
+    L"  return el.style.display;"
+    L"})()"
+    L")",
+    &style_display));
+  ASSERT_EQ(L"none", style_display);
+
+  // Make sure that the notification is visible
+  bool has_class;
+  ASSERT_TRUE(tab->ExecuteAndExtractBool(L"",
+    L"window.domAutomationController.send("
+    L"(function() {"
+    L"  var el = document.querySelector('#notification');"
+    L"  return hasClass(el, 'show');"
+    L"})()"
+    L")",
+    &has_class));
+  ASSERT_TRUE(has_class);
+
+  bool is_home_page;
+  ASSERT_TRUE(browser->GetBooleanPreference(prefs::kHomePageIsNewTabPage,
+              &is_home_page));
+  ASSERT_TRUE(is_home_page);
+}