[protector] Homepage UI.

chrome/test/functional/protector.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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.
 
 import pyauto_functional  # Must be imported first
 import pyauto
 import test_utils
 
 import json
@@ skipping 162 matching lines @@
 
     Args:
       pinned_tabs: list(str) with a list of pinned tabs URLs.
     """
     prefs = self._LoadPreferences()
     prefs['pinned_tabs'] = []
     for tab in pinned_tabs:
       prefs['pinned_tabs'].append({'url': tab})
     self._WritePreferences(prefs)
 
+  def _ChangeHomepage(self, homepage, homepage_is_ntp, show_homepage_button):
+    """Changes homepage settings.
+
+    Args:
+      homepage: new homepage URL (str),
+      homepage_is_ntp: whether homepage is NTP.
+      show_homepage_button: whether homepage button is visible.
+    """
+    prefs = self._LoadPreferences()
+    prefs['homepage'] = homepage
+    prefs['homepage_is_newtabpage'] = homepage_is_ntp
+    prefs['browser']['show_home_button'] = show_homepage_button
+    self._WritePreferences(prefs)
+    #input('Pause')

[Nirnimesh, 2012/04/16 21:10:29]

remove

[Ivan Korotkov, 2012/04/17 09:54:10]

Done.

+
   def _AssertSingleTabOpen(self, url):
     """Asserts that a single tab with given url is open.
 
     Args:
       url: URL of the single tab.
     """
     info = self.GetBrowserInfo()
     self.assertEqual(1, len(info['windows']))  # one window
     self.assertEqual(1, len(info['windows'][0]['tabs']))  # one tab
     self.assertEqual(url, info['windows'][0]['tabs'][0]['url'])
@@ skipping 199 matching lines @@
                      self.GetPrefsInfo().Prefs(pyauto.kRestoreOnStartup))
     # Verify that open tabs are consistent with restored prefs.
     self._AssertSingleTabOpen(previous_url)
     self.ApplyProtectorChange()
     # Now the new preference values are active.
     self.assertEqual(self._SESSION_STARTUP_URLS,
                      self.GetPrefsInfo().Prefs(pyauto.kRestoreOnStartup))
     # No longer showing the change.
     self.assertFalse(self.GetProtectorState()['showing_change'])
 
-  def testDetectSessionStartupChangeAndApply(self):
+  def testDetectSessionStartupChangeAndDiscard(self):
     """Test for detecting and discarding a session startup pref change."""
     # Set startup prefs to restoring last open tabs.
     self.SetPrefs(pyauto.kRestoreOnStartup, self._SESSION_STARTUP_LAST)
     # Restart browser with startup prefs set to open google.com.
     self.RestartBrowser(
         clear_profile=False,
         pre_launch_hook=lambda: self._ChangeSessionStartupPrefs(
             self._SESSION_STARTUP_URLS,
             startup_urls=['http://www.google.com']))
     # The change must be detected by Protector.
@@ skipping 108 matching lines @@
     self.DiscardProtectorChange()
     # No longer showing the change.
     self.assertFalse(self.GetProtectorState()['showing_change'])
     # Pinned tabs are not opened after another restart.
     self.RestartBrowser(clear_profile=False)
     self._AssertSingleTabOpen('chrome://newtab/')
     # Not showing the change after a restart.
     self.assertFalse(self.GetProtectorState()['showing_change'])
 
 
+class ProtectorHomepageTest(BaseProtectorTest):
+  """Test suite for homepage changes with Protector enabled."""
+
+  def testDetectHomepageChangeWhenButtonNotShown(self):
+    """Test that homepage change is detected and reverted when homepage button
+    is set to not be shown.
+    """
+    previous_homepage = 'http://example.com/'
+    self.SetPrefs(pyauto.kHomePage, previous_homepage)
+    self.SetPrefs(pyauto.kHomePageIsNewTabPage, False)
+    self.SetPrefs(pyauto.kShowHomeButton, False)
+    self.RestartBrowser(
+        clear_profile=False,
+        pre_launch_hook=lambda: self._ChangeHomepage(
+            'http://example.info/', False, False))
+    # Change has been automatically reverted, no bubble shown.
+    self.assertFalse(self.GetProtectorState()['showing_change'])
+    self.assertEquals(previous_homepage,
+                      self.GetPrefsInfo().Prefs(pyauto.kHomePage))
+
+  def testDetectHomepageChangeAndApply(self):
+    """Test that homepage change is detected and can be applied."""
+    previous_homepage = 'http://example.com/'
+    new_homepage = 'http://example.info/'
+    self.SetPrefs(pyauto.kHomePage, previous_homepage)
+    self.SetPrefs(pyauto.kHomePageIsNewTabPage, False)
+    self.SetPrefs(pyauto.kShowHomeButton, False)
+    self.RestartBrowser(
+        clear_profile=False,
+        pre_launch_hook=lambda: self._ChangeHomepage(new_homepage, False, True))
+    # The change must be detected by Protector.
+    self.assertTrue(self.GetProtectorState()['showing_change'])
+    # Protector must restore old preference values.
+    self.assertEquals(previous_homepage,
+                      self.GetPrefsInfo().Prefs(pyauto.kHomePage))
+    self.assertEquals(False, self.GetPrefsInfo().Prefs(pyauto.kShowHomeButton))
+    self.ApplyProtectorChange()
+    # Now new values are active.
+    self.assertEquals(new_homepage, self.GetPrefsInfo().Prefs(pyauto.kHomePage))
+    self.assertEquals(True, self.GetPrefsInfo().Prefs(pyauto.kShowHomeButton))
+    # No longer showing the change
+    self.assertFalse(self.GetProtectorState()['showing_change'])
+
+  def testDetectHomepageChangeAndDiscard(self):
+    """Test that homepage change is detected and can be discarded."""
+    previous_homepage = 'http://example.com/'
+    new_homepage = 'http://example.info/'
+    self.SetPrefs(pyauto.kHomePage, previous_homepage)
+    self.SetPrefs(pyauto.kHomePageIsNewTabPage, False)
+    self.SetPrefs(pyauto.kShowHomeButton, False)
+    self.RestartBrowser(
+        clear_profile=False,
+        pre_launch_hook=lambda: self._ChangeHomepage(new_homepage, False, True))
+    # The change must be detected by Protector.
+    self.assertTrue(self.GetProtectorState()['showing_change'])
+    # Protector must restore old preference values.
+    self.assertEquals(previous_homepage,
+                      self.GetPrefsInfo().Prefs(pyauto.kHomePage))
+    self.assertEquals(False, self.GetPrefsInfo().Prefs(pyauto.kShowHomeButton))
+    self.DiscardProtectorChange()
+    # Nothing changed
+    self.assertEquals(previous_homepage,
+                      self.GetPrefsInfo().Prefs(pyauto.kHomePage))
+    self.assertEquals(False, self.GetPrefsInfo().Prefs(pyauto.kShowHomeButton))
+    # No longer showing the change
+    self.assertFalse(self.GetProtectorState()['showing_change'])
+
+
 class ProtectorDisabledTest(BaseProtectorTest):
   """Test suite for Protector in disabled state."""
 
   def ExtraChromeFlags(self):
     """Ensures Protector is disabled.
 
     Returns:
       A list of extra flags to pass to Chrome when it is launched.
     """
     return super(ProtectorDisabledTest, self).ExtraChromeFlags() + [
@@ skipping 58 matching lines @@
     self.assertFalse(self.GetProtectorState()['showing_change'])
     # New preference values must be active.
     self.assertEqual(self._SESSION_STARTUP_URLS,
                      self.GetPrefsInfo().Prefs(pyauto.kRestoreOnStartup))
     # Verify that open tabs are consistent with new prefs.
     info = self.GetBrowserInfo()
     self.assertEqual(1, len(info['windows']))  # one window
     self.assertEqual(1, len(info['windows'][0]['tabs']))  # one tab
     self.assertEqual(new_url, info['windows'][0]['tabs'][0]['url'])
 
+  def testNoHomepageChangeReported(self):
+    """Test that homepage change is neither reported nor reverted."""
+    new_homepage = 'http://example.info/'
+    self.SetPrefs(pyauto.kHomePage, 'http://example.com/')
+    self.SetPrefs(pyauto.kHomePageIsNewTabPage, False)
+    self.SetPrefs(pyauto.kShowHomeButton, False)
+    self.RestartBrowser(
+        clear_profile=False,
+        pre_launch_hook=lambda: self._ChangeHomepage(new_homepage, False, True))
+    # Not showing the change.
+    self.assertFalse(self.GetProtectorState()['showing_change'])
+    # New values must be active.
+    self.assertEquals(new_homepage, self.GetPrefsInfo().Prefs(pyauto.kHomePage))
+    self.assertEquals(True, self.GetPrefsInfo().Prefs(pyauto.kShowHomeButton))
+
 
 if __name__ == '__main__':
   pyauto_functional.Main()