[Password manager tests automation] Refactor test_runner.

components/test/data/password_manager/automated_tests/tests.py

 # -*- coding: utf-8 -*-
 # 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.
 
 """Automated tests for many websites"""
 
 import argparse
 
 from environment import Environment
 from websitetest import WebsiteTest
 
 
+TEST_CASES = ("PromptFailTest", "PromptSuccessTest", "SaveAndAutofillTest")
+
+
 class Alexa(WebsiteTest):
 
   def Login(self):
     self.GoTo("https://www.alexa.com/secure/login")
     self.FillUsernameInto("#email")
     self.FillPasswordInto("#pwd")
     self.Submit("#pwd")
 
 
 class Dropbox(WebsiteTest):
@@ skipping 473 matching lines @@
     xml = "<result>"
     for (name, test_type, success, failure_log) in environment_tests_results:
       if not (save_only_failures and success):
         xml += (
             "<test name='{0}' successful='{1}' type='{2}'>{3}</test>".format(
                 name, success, test_type, failure_log))
     xml += "</result>"
     with open(environment_save_path, "w") as save_file:
       save_file.write(xml)
 
+
 def RunTest(chrome_path, chromedriver_path, profile_path,
-            environment_passwords_path, website_test_name, test_case_name):
+            environment_passwords_path, website_test_name,
+            test_case_name):
   """Runs the test for the specified website.
 
   Args:
     chrome_path: The chrome binary file.
     chromedriver_path: The chromedriver binary file.
     profile_path: The chrome testing profile folder.
     environment_passwords_path: The usernames and passwords file.
     website_test_name: Name of the website to test (refer to keys in
         all_tests above).
 
   Returns:
     The results of the test as list of TestResults.
 
   Raises:
     Exception: An exception is raised if one of the tests for the website
         fails, or if the website name is not known.
   """
 
   enable_automatic_password_saving = (test_case_name == "SaveAndAutofillTest")
   environment = Environment(chrome_path, chromedriver_path, profile_path,
                             environment_passwords_path,
                             enable_automatic_password_saving)
+  try:
+    if website_test_name in all_tests:
+      environment.AddWebsiteTest(all_tests[website_test_name])
+    else:
+      raise Exception("Test name {} is unknown.".format(website_test_name))
 
-  if website_test_name in all_tests:
-    environment.AddWebsiteTest(all_tests[website_test_name])
-  else:
-    raise Exception("Test name {} is unknown.".format(website_test_name))
-
-  environment.RunTestsOnSites(test_case_name)
-  environment.Quit()
-  return environment.tests_results
+    environment.RunTestsOnSites(test_case_name)
+    return environment.tests_results
+  finally:
+    environment.Quit()
 
 def main():
   parser = argparse.ArgumentParser(
       description="Password Manager automated tests help.")
 
   parser.add_argument(
       "--chrome-path", action="store", dest="chrome_path",
       help="Set the chrome path (required).", required=True)
   parser.add_argument(
       "--chromedriver-path", action="store", dest="chromedriver_path",
@@ skipping 18 matching lines @@
   parser.add_argument("--test-cases-to-run", help="Names of test cases which"
                       "should be run. Currently supported test cases are:"
                       "PromptFailTest, PromptSuccessTest, SaveAndAutofillTest",
                       dest="test_cases_to_run", action="store", nargs="*")
   args = parser.parse_args()
 
   save_path = None
   if args.save_path:
     save_path = args.save_path
 
-  test_cases_to_run = args.test_cases_to_run or\
-      ("PromptFailTest", "PromptSuccessTest", "SaveAndAutofillTest")
-
+  test_cases_to_run = args.test_cases_to_run or TEST_CASES
   for test_case in test_cases_to_run:
     tests_results = RunTest(
         args.chrome_path, args.chromedriver_path, args.profile_path,
         args.passwords_path, args.website, test_case)
 
 
   SaveResults(tests_results, save_path,
               save_only_failures=args.save_only_failures)
 
 if __name__ == "__main__":
   main()