Change the way how the saved passwords are cleaned in the password manager python tests.

components/test/data/password_manager/environment.py

 # Copyright 2014 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.
 
 """The testing Environment class."""
 
 import logging
 import shutil
 import sys
 import time
@@ skipping 147 matching lines @@
             self.passwords_tree.find(
                 ".//*[@name='%s']/password" % websitetest.name))
         if password_tag.text:
           websitetest.password = password_tag.text
     self.websitetests.append(websitetest)
     if disabled:
       self.disabled_tests.append(websitetest.name)
     else:
       self.working_tests.append(websitetest.name)
 
-  def RemoveAllPasswords(self):
-    """Removes all the stored passwords."""
-    logging.info("\nRemoveAllPasswords\n")
-    self.driver.get("chrome://settings/passwords")
-    self.driver.switch_to_frame("settings")
-    while True:
-      try:
-        self.driver.execute_script(
-            "document.querySelector('#saved-passwords-list .row-delete-button')"
-            ".click()")
-        time.sleep(1)
-      except NoSuchElementException:
-        break
-      except WebDriverException:
-        break
+  def ClearCache(self, clear_passwords):
+    """Clear the browser cookies. If |clear_passwords| is true, clear all the
+    saved passwords too.
 
-  def ClearAllCookies(self):
-    """Removes all the cookies."""
-    logging.info("\nClearAllCookies\n")
+    Args:
+      clear_passwords : Clear all the passwords if the bool value is true.
+    """
+    logging.info("\nClearCache\n")
     self.driver.get("chrome://settings/clearBrowserData")
     self.driver.switch_to_frame("settings")
-    self.driver.execute_script(
-        "var checkboxes = document.querySelectorAll("
-        "    '#clear-data-checkboxes [type=\\\'checkbox\\\']');"
-        "for (var i in checkboxes)"
-        "    checkboxes[i].checked = false;"
-        "document.querySelector('#delete-cookies-checkbox').checked = true;"
-        "document.querySelector('#clear-browser-data-commit').click();")
+    script = (
+        "if (!document.querySelector('#delete-cookies-checkbox').checked)"
+        "  document.querySelector('#delete-cookies-checkbox').click();"
+        )
+    negation = ""
+    if clear_passwords:
+      negation = "!"
+    script += (
+        "if (%sdocument.querySelector('#delete-passwords-checkbox').checked)"
+        "  document.querySelector('#delete-passwords-checkbox').click();"
+        % negation)
+    script += "document.querySelector('#clear-browser-data-commit').click();"
+    self.driver.execute_script(script)
     time.sleep(2)
 
   def OpenTabAndGoToInternals(self, url):
     """If there is no |self.website_window|, opens a new tab and navigates to
     |url| in the new tab. Navigates to the passwords internals page in the
     first tab. Raises an exception otherwise.
 
     Args:
       url: Url to go to in the new tab.
 
@@ skipping 151 matching lines @@
 
   def TestList(self, websitetests):
     """Runs the tests on the websites in |websitetests|.
 
     Args:
       websitetests: A list of WebsiteTests that are going to be tested.
 
     Raises:
       Exception: An exception is raised if the tests fail.
     """
-    self.RemoveAllPasswords()
+    self.ClearCache(True)
 
     for websitetest in websitetests:
       successful = True
       error = ""
       try:
         websitetest.was_run = True
         websitetest.WrongLoginTest()
         websitetest.SuccessfulLoginTest()
+        self.ClearCache(False)
         websitetest.SuccessfulLoginWithAutofilledPasswordTest()
-        self.RemoveAllPasswords()
+        self.ClearCache(True)
         websitetest.SuccessfulLoginTest()
+        self.ClearCache(True)
       except Exception:
         successful = False
         error = traceback.format_exc()
       self.tests_results.append(TestResult(websitetest.name, "normal",
           successful, escape(error)))
 
+
   def PromptTestList(self, websitetests):
     """Runs the prompt tests on the websites in |websitetests|.
 
     Args:
       websitetests: A list of WebsiteTests that are going to be tested.
 
     Raises:
       Exception: An exception is raised if the tests fail.
     """
-    self.RemoveAllPasswords()
+    self.ClearCache(True)
 
     for websitetest in websitetests:
       successful = True
       error = ""
       try:
         websitetest.was_run = True
         websitetest.PromptTest()
       except Exception:
         successful = False
         error = traceback.format_exc()
       self.tests_results.append(TestResult(websitetest.name, "prompt",
           successful, escape(error)))
 
   def Quit(self):
     """Closes the tests."""
     # Close the webdriver.
     self.driver.quit()