New pyauto translate tests.

chrome/test/functional/translate.py

 #!/usr/bin/python
 # Copyright (c) 2010 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 glob
 import logging
 import os
 import shutil
 import tempfile
@@ skipping 23 matching lines @@
     """Return the file URL for the given file in the data directory."""
     return self.GetFileURLForPath(os.path.join(self.DataDir(), filename))
 
   def _GetDefaultSpanishURL(self):
     return self._GetURLForDataDirFile(
         os.path.join('translate', self.spanish, 'google.html'))
 
   def _GetDefaultEnglishURL(self):
     return self._GetURLForDataDirFile('title1.html')
 
-  def _NavigateAndWaitForBar(self, url):
-    self.NavigateToURL(url)
-    self.WaitForInfobarCount(1)
+  def _NavigateAndWaitForBar(self, url, window_index=0, tab_index=0):
+    self.NavigateToURL(url, window_index, tab_index)
+    self.WaitForInfobarCount(1, windex=window_index, tab_index=tab_index)
 
   def _ClickTranslateUntilSuccess(self, window_index=0, tab_index=0):
     """Since the translate can fail due to server error, continue trying until
        it is successful or until it has tried too many times."""
     max_tries = 10
     curr_try = 0
     while (curr_try < max_tries and
            not self.ClickTranslateBarTranslate(window_index=window_index,
                                                tab_index=tab_index)):
       curr_try = curr_try + 1
@@ skipping 132 matching lines @@
     self.assertFalse('translate_bar' in translate_info)
 
   def testTranslateDiffURLs(self):
     """Test that HTTP, HTTPS, and file urls all get translated."""
     http_url = 'http://www.google.com/webhp?hl=es'
     https_url = 'https://www.google.com/webhp?hl=es'
     file_url = self._GetDefaultSpanishURL()
     for url in (http_url, https_url, file_url):
       self._AssertTranslateWorks(url, self.spanish)
 
-  def testNeverTranslateLanguage(self):

[Alyssa, 2010/08/23 22:42:54]

A test with the same name and almost duplicate code exists. Removing this one.

-    """Verify no translate bar for blacklisted language."""
-    self._NavigateAndWaitForBar(self._GetDefaultSpanishURL())
-    self.SelectTranslateOption('never_translate_language')
-    self.assertFalse(self.GetBrowserInfo()['windows'][0]['tabs'][0]['infobars'])
-    # Navigate to a page that will show the translate bar, then navigate away.
-    self._NavigateAndWaitForBar(
-        self._GetURLForDataDirFile('french_page.html'))
-    self.NavigateToURL('http://es.wikipedia.org/wiki/Wikipedia:Portada')
-    self.WaitForInfobarCount(0)
-    translate_info = self.GetTranslateInfo()
-    self.assertFalse('translate_bar' in translate_info)
-    self.assertFalse(translate_info['page_translated'])
-    self.assertFalse(translate_info['can_translate_page'])
-
   def testNotranslateMetaTag(self):
     """Test "notranslate" meta tag."""
     self._NavigateAndWaitForBar(self._GetDefaultSpanishURL())
     self.NavigateToURL(self._GetURLForDataDirFile(
           os.path.join('translate', 'notranslate_meta_tag.html')))
     self.WaitForInfobarCount(0)
     translate_info = self.GetTranslateInfo()
     self.assertFalse('translate_bar' in translate_info)
 
   def testToggleTranslateOption(self):
@@ skipping 106 matching lines @@
     self.assertFalse(self.GetPrefsInfo().Prefs(pyauto.kEnableTranslate))
     self.NavigateToURL(self._GetDefaultSpanishURL())
     self.assertFalse('translate_bar' in self.GetTranslateInfo())
 
   def testAlwaysTranslateLanguageButton(self):
     """Test the always translate language button."""
     spanish_url = self._GetDefaultSpanishURL()
     self._NavigateAndWaitForBar(spanish_url)
 
     # The 'Always Translate' button doesn't show up until the user has clicked
-    # 'Translate' for a language several times.
-    max_tries = 10
-    curr_try = 0
-    while (curr_try < max_tries and
-           not self.GetTranslateInfo()['translate_bar']\
-                                      ['always_translate_lang_button_showing']):
+    # 'Translate' for a language 3 times.
+    for i in range(3):

[Nirnimesh, 2010/08/24 01:23:19]

replace i with '_' or 'unused' to avoid pychecker warnings

       self._ClickTranslateUntilSuccess()
-      self._NavigateAndWaitForBar(spanish_url)
-      curr_try = curr_try + 1
-    if curr_try == max_tries:
-      self.fail('Clicked translate %d times and always translate button never '\
-                'showed up.' % max_tries)
+      self.GetBrowserWindow(0).GetTab(0).Reload()
 

[Nirnimesh, 2010/08/24 01:23:19]

Shouldn't you check that the button shows up?

     # Click the 'Always Translate' button.
     self.SelectTranslateOption('click_always_translate_lang_button')
     # Navigate to another Spanish page and verify it was translated.
     self._NavigateAndWaitForBar('http://www.google.com/webhp?hl=es')
     self.WaitUntilTranslateComplete()
     # Assert that a translation was attempted. We don't care if it was error
     # or success.
     self.assertNotEqual(self.before_translate,
                         self.GetTranslateInfo()['translate_bar']['bar_state'])
 
+  def testNeverTranslateLanguageButton(self):
+    """Test the never translate language button."""
+    spanish_url = self._GetDefaultSpanishURL()
+    self._NavigateAndWaitForBar(spanish_url)
+
+    # The 'Never Translate' button doesn't show up until the user has clicked
+    # 'Nope' for a language 3 times.
+    for i in range(3):

[Nirnimesh, 2010/08/24 01:23:19]

same comments as above

+      self.SelectTranslateOption('decline_translation')
+      self.GetBrowserWindow(0).GetTab(0).Reload()
+
+    # Click the 'Never Translate' button.
+    self.SelectTranslateOption('click_never_translate_lang_button')
+    # Navigate to another Spanish page and verify the page can't be translated.
+    # First navigate to a French page, wait for bar, navigate to Spanish page
+    # and wait for bar to go away.
+    self._NavigateAndWaitForBar('http://www.google.com/webhp?hl=fr')
+    self.NavigateToURL('http://www.google.com/webhp?hl=es')
+    self.WaitForInfobarCount(0)
+    self.assertFalse(self.GetTranslateInfo()['can_translate_page'])
+
+  def testChangeTargetLanguageAlwaysTranslate(self):
+    """Tests that the change target language option works with always translate
+       on reload.
+   
+    This test is motivated by crbug.com/37313.
+    """
+    self._NavigateAndWaitForBar(self._GetDefaultSpanishURL())
+    self._ClickTranslateUntilSuccess()
+    # Select a different target language on translate info bar (French).
+    self.ChangeTranslateToLanguage('French')
+    translate_info = self.GetTranslateInfo()
+    self.assertTrue('translate_bar' in translate_info)
+    self.assertEqual('fr', translate_info['translate_bar']['target_lang_code'])
+    # Select always translate Spanish to French.
+    self.SelectTranslateOption('toggle_always_translate')
+    # Reload the page and assert that the page has been translated to French.
+    self.GetBrowserWindow(0).GetTab(0).Reload()
+    self.WaitUntilTranslateComplete()
+    translate_info = self.GetTranslateInfo()
+    self.assertTrue(translate_info['page_translated'])
+    self.assertTrue(translate_info['can_translate_page'])
+    self.assertTrue('translate_bar' in translate_info)
+    self.assertEqual('fr', translate_info['translate_bar']['target_lang_code'])
+
   def testSeveralLanguages(self):
     """Verify translation for several languages.
 
     This assumes that there is a directory of directories in the data dir.
     The directory is called 'translate', and within that directory there are
     subdirectories for each language code. Ex. a subdirectory 'es' with Spanish
     html files.
     """
     corpora_path = os.path.join(self.DataDir(), 'translate')
     corp_dir = glob.glob(os.path.join(corpora_path, '??')) + \
@@ skipping 31 matching lines @@
     # Ref: http://boodebr.org/main/python/all-about-python-and-unicode
     try:
       for filename in crazy_filenames:  # filename is unicode.
         utf8_filename = filename.encode('utf-8')
         file_path = os.path.join(temp_dir, utf8_filename)
         _CreateFile(os.path.join(temp_dir, filename))  # unicode file.
         file_url = self.GetFileURLForPath(file_path)
         downloaded_file = os.path.join(download_dir, filename)
         os.path.exists(downloaded_file) and os.remove(downloaded_file)
         self.DownloadAndWaitForStart(file_url)
-      self.WaitForAllDownloadsToComplete()
-
+        # Wait for files and remove them as we go.
+        self.WaitForAllDownloadsToComplete()
+        os.path.exists(downloaded_file) and os.remove(downloaded_file)
     finally:
       shutil.rmtree(unicode(temp_dir))  # unicode so that win treats nicely.
 
   def testHistoryNotTranslated(self):
     """Tests navigating to History page with other languages."""
     # Build the history with non-English content.
     history_file = os.path.join(
         self.DataDir(), 'translate', 'crazy_history.txt')
     history_items = self.EvalDataFrom(history_file)
     for history_item in history_items:
        self.AddHistoryItem(history_item)
     # Navigate to a page that triggers the translate bar so we can verify that
     # it disappears on the history page.
     self._NavigateAndWaitForBar(self._GetDefaultSpanishURL())
     self.NavigateToURL('chrome://history/')
     self.WaitForInfobarCount(0)
     self.assertFalse('translate_bar' in self.GetTranslateInfo())
 
   def testDownloadsNotTranslated(self):
     """Tests navigating to the Downloads page with other languages."""
     # Build the download history with non-English content.
     self._CreateCrazyDownloads()
     # Navigate to a page that triggers the translate bar so we can verify that
     # it disappears on the downloads page.
     self._NavigateAndWaitForBar(self._GetDefaultSpanishURL())
     self.NavigateToURL('chrome://downloads/')
     self.WaitForInfobarCount(0)
     self.assertFalse('translate_bar' in self.GetTranslateInfo())
 
+  def testAlwaysTranslateInIncognito(self):
+    """Verify that pages aren't auto-translated in incognito windows."""
+    url = self._GetDefaultSpanishURL()
+    self._NavigateAndWaitForBar(url)
+    info_before_translate = self.GetTranslateInfo()
+    self.assertTrue('translate_bar' in info_before_translate)
+    self.SelectTranslateOption('toggle_always_translate')
+
+    # Navigate to a page in incognito and verify that it is not auto-translated.
+    self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
+    self._NavigateAndWaitForBar(url, window_index=1)
+    info_before_translate = self.GetTranslateInfo()
+    self.assertTrue('translate_bar' in info_before_translate)
+    self.assertFalse(info_before_translate['page_translated'])
+    self.assertTrue(info_before_translate['can_translate_page'])
+    self.assertEqual(self.before_translate,
+                     info_before_translate['translate_bar']['bar_state'])
+
+  def testMultipleTabsAndWindows(self):
+    """Verify that translate infobar shows up in multiple tabs and windows."""
+    url = self._GetDefaultSpanishURL()
+    def _AssertTranslateInfobarShowing(window_index=0, tab_index=0):
+      """Navigate to a Spanish page in the given window/tab and verify that the
+         translate bar shows up.
+      """
+      self.NavigateToURL(url, window_index, tab_index)
+      self.WaitForInfobarCount(1, windex=window_index, tab_index=tab_index)
+      info_before_translate = self.GetTranslateInfo(window_index=window_index,
+                                                    tab_index=tab_index)
+      self.assertTrue('translate_bar' in info_before_translate)
+
+    _AssertTranslateInfobarShowing()
+    self.AppendTab(pyauto.GURL('about:blank'))
+    _AssertTranslateInfobarShowing(tab_index=1)
+    self.OpenNewBrowserWindow(True)
+    _AssertTranslateInfobarShowing(window_index=1)
+    self.AppendTab(pyauto.GURL('about:blank'), 1)
+    _AssertTranslateInfobarShowing(window_index=1, tab_index=1)
+    self.RunCommand(pyauto.IDC_NEW_INCOGNITO_WINDOW)
+    _AssertTranslateInfobarShowing(window_index=2)
+    self.AppendTab(pyauto.GURL('about:blank'), 2)
+    _AssertTranslateInfobarShowing(window_index=2, tab_index=1)
+
 
 if __name__ == '__main__':
   pyauto_functional.Main()