Add three fullscreen tests and three mouse lock tests.

chrome/test/functional/fullscreen_mouselock.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 logging
 import os
 import re
 import shutil
 import time
@@ skipping 125 matching lines @@
         'fullscreen_mouselock', 'fullscreen_mouselock.html'))
     # Should not be in fullscreen mode during initial launch.
     self.assertFalse(self.IsFullscreenForBrowser())
     self.assertFalse(self.IsFullscreenForTab())
     # Initiate browser fullscreen.
     self.ApplyAccelerator(pyauto.IDC_FULLSCREEN)
     self.assertTrue(self.WaitUntil(self.IsFullscreenForBrowser))
     self.assertTrue(self.WaitUntil(lambda: not self.IsFullscreenForTab()))
     self.assertTrue(self.WaitUntil(lambda: not self.IsMouseLocked()))
 
+  def _InitiateTabFullscreen(self):
+    """Helper function that initiates tab fullscreen."""
+    self.NavigateToURL(self.GetHttpURLForDataPath(
+        'fullscreen_mouselock', 'fullscreen_mouselock.html'))
+    # Initiate tab fullscreen.
+    self._driver.find_element_by_id('enterFullscreen').click()
+    self.assertTrue(self.WaitUntil(self.IsFullscreenForTab))
+
   def _AcceptFullscreenOrMouseLockRequest(self):
     """Helper function to accept fullscreen or mouse lock request."""
     self.AcceptCurrentFullscreenOrMouseLockRequest()
     self.assertTrue(self.WaitUntil(
         lambda: not self.IsFullscreenBubbleDisplayingButtons()))
 
   def _EnableFullscreenAndMouseLockMode(self):
     """Helper function to enable fullscreen and mouse lock mode."""
     self._LaunchFSAndExpectPrompt(button_action='enterFullscreenAndLockMouse1')
     # Allow fullscreen.
@@ skipping 10 matching lines @@
 
     Args:
       button_action: The button id to click to initiate an action. Default is to
           click lockMouse1.
     """
     self._driver.find_element_by_id(button_action).click()
     self.assertTrue(self.WaitUntil(self.IsMouseLockPermissionRequested))
     self.AcceptCurrentFullscreenOrMouseLockRequest()
     self.assertTrue(self.IsMouseLocked())
 
+  def _EnableAndReturnLockMouseResult(self):
+    """Helper function to enable and return mouse lock result."""
+    self.NavigateToURL(self.GetHttpURLForDataPath(
+        'fullscreen_mouselock', 'fullscreen_mouselock.html'))
+    self._driver.find_element_by_id('lockMouse2').click()
+    self.assertTrue(
+        self.WaitUntil(self.IsMouseLockPermissionRequested))
+    self.AcceptCurrentFullscreenOrMouseLockRequest()
+    # Waits until lock_result gets 'success' or 'failure'.
+    return self._driver.execute_script('return lock_result')
+
   def testPrefsForFullscreenAllowed(self):
     """Verify prefs when fullscreen is allowed."""
     self._LaunchFSAndExpectPrompt()
     self._AcceptFullscreenOrMouseLockRequest()
     content_settings = (
         self.GetPrefsInfo().Prefs()['profile']['content_settings'])
     self.assertEqual(
         {self._hostname_pattern + ',*': {'fullscreen': 1}},  # Allow hostname.
         content_settings['pattern_pairs'],
         msg='Saved hostname pattern does not match expected pattern.')
@@ skipping 182 matching lines @@
     """Verify mouse lock bubble goes away when tab loses focus."""
     self.NavigateToURL(self.GetHttpURLForDataPath(
         'fullscreen_mouselock', 'fullscreen_mouselock.html'))
     self._driver.find_element_by_id('lockMouse1').click()
     self.assertTrue(self.WaitUntil(self.IsMouseLockPermissionRequested))
     self.AppendTab(pyauto.GURL('chrome://newtab'))
     self.assertTrue(self.WaitUntil(
         lambda: not self.IsFullscreenBubbleDisplayingButtons()),
                     msg='Mouse lock bubble did not clear when tab lost focus.')
 
+  def testTabFSExitWhenNavBackToPrevPage(self):
+    """Verify tab fullscreen exit when navigating back to previous page.
+
+    This test navigates to a new page while in tab fullscreen mode by using
+    GoBack() to navigate to the previous google.html page.
+    """
+    self.NavigateToURL(self.GetHttpURLForDataPath('google', 'google.html'))
+    self._InitiateTabFullscreen()
+    self.GetBrowserWindow().GetTab().GoBack()
+    self.assertFalse(
+        self.IsFullscreenForTab(),
+        msg='Tab fullscreen did not exit when navigating to a new page.')
+
+  def testTabFSExitWhenNavToNewPage(self):
+    """Verify tab fullscreen exit when navigating to a new website.
+
+    This test navigates to a new website while in tab fullscreen.
+    """
+    self._InitiateTabFullscreen()
+    self.NavigateToURL(self.GetHttpURLForDataPath('google', 'google.html'))
+    self.assertFalse(
+        self.IsFullscreenForTab(),
+        msg='Tab fullscreen did not exit when navigating to a new website.')
+
+  def testTabFSDoesNotExitForAnchorLinks(self):
+    """Verify tab fullscreen does not exit for anchor links.
+
+    Tab fullscreen should not exit when following a link to the same page such
+    as example.html#anchor.
+    """
+    self._InitiateTabFullscreen()
+    self._driver.find_element_by_id('anchor').click()
+    self.assertTrue(
+        self.WaitUntil(self.IsFullscreenForTab),
+        msg='Tab fullscreen should not exit when clicking on an anchor link.')
+
+  def testMLExitWhenNavBackToPrevPage(self):
+    """Verify mouse lock exit when navigating back to previous page.
+
+    This test navigates to a new page while mouse lock is activated by using
+    GoBack() to navigate to the previous google.html page.
+    """
+    self.NavigateToURL(self.GetHttpURLForDataPath('google', 'google.html'))
+    lock_result = self._EnableAndReturnLockMouseResult()
+    self.assertEqual(
+        lock_result, 'success', msg='Mouse is not locked.')
+    self.GetBrowserWindow().GetTab().GoBack()
+    self.assertFalse(
+        self.IsMouseLocked(),
+        msg='Mouse lock did not exit when navigating to the prev page.')
+
+  def testMLExitWhenNavToNewPage(self):
+    """Verify mouse lock exit when navigating to a new website."""
+    lock_result = self._EnableAndReturnLockMouseResult()
+    self.assertEqual(
+        lock_result, 'success', msg='Mouse is not locked.')
+    self.NavigateToURL(self.GetHttpURLForDataPath('google', 'google.html'))
+    self.assertFalse(
+        self.IsMouseLocked(),
+        msg='Mouse lock did not exit when navigating to a new website.')
+
+  def testMLDoesNotExitForAnchorLinks(self):
+    """Verify mouse lock does not exit for anchor links.
+
+    Mouse lock should not exist when following a link to the same page such as
+    example.html#anchor.
+    """
+    lock_result = self._EnableAndReturnLockMouseResult()
+    self.assertEqual(
+        lock_result, 'success', msg='Mouse is not locked.')
+    self._driver.execute_script('document.getElementById("anchor").click()')
+    self.assertTrue(self.WaitUntil(self.IsMouseLocked),
+                    msg='Mouse lock broke when clicking on an anchor link.')
+
   def ExitTabFSToBrowserFS(self):
     """Verify exiting tab fullscreen leaves browser in browser fullscreen.
 
     This test is semi-automated.
 
     The browser initiates browser fullscreen, then initiates tab fullscreen. The
     test verifies that existing tab fullscreen by simulating ESC key press or
     clicking the js function to exitFullscreen() will exit the tab fullscreen
     leaving browser fullscreen intact.
     """
@@ skipping 88 matching lines @@
     self.NavigateToURL(url2, 0, 1)
     self._driver.switch_to_window(tab2)
     self._EnableMouseLockMode()  # Lock mouse in tab 2.
     raw_input('Manually move the mouse cursor on the page in tab 2. Shift+Tab \
               into tab 1, click on lockMouse1() button, and move the mouse \
               cursor on the page in tab 1. Verify mouse movement is smooth.')
 
 
 if __name__ == '__main__':
   pyauto_functional.Main()