[Password manager tests automation] Adds an ability to specify test_case which should be run.

components/test/data/password_manager/automated_tests/websitetest.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.
 
 """WebsiteTest testing class."""
 
 import logging
 import time
 
 from selenium.webdriver.common.action_chains import ActionChains
@@ skipping 12 matching lines @@
   """
 
   # Possible values of self.autofill_expectation.
   AUTOFILLED = 1  # Expect password and username to be autofilled.
   NOT_AUTOFILLED = 2  # Expect password and username not to be autofilled.
 
   # The maximal accumulated time to spend in waiting for website UI
   # interaction.
   MAX_WAIT_TIME_IN_SECONDS = 200
 
-  # Types of test to be passed to self.RunTest().
-  TEST_TYPE_PROMPT_FAIL = 1
-  TEST_TYPE_PROMPT_SUCCESS = 2
-  TEST_TYPE_SAVE_AND_AUTOFILL = 3
-
   def __init__(self, name, username_not_auto=False):
     """Creates a new WebsiteTest.
 
     Args:
       name: The website name, identifying it in the test results.
       username_not_auto: Expect that the tested website fills username field
         on load, and Chrome cannot autofill in that case.
     """
     self.name = name
     self.username = None
@@ skipping 313 matching lines @@
         [environment.MESSAGE_SAVE],
         True,
         "Error: did not detect login success on website: %s" % self.name)
     self.Logout()
     self.LoginWhenAutofilled()
     self.environment.CheckForNewString(
         [environment.MESSAGE_SAVE],
         True,
         "Error: failed autofilled login on website: %s" % self.name)
 
-  def RunTest(self, test_type):
-    """Runs test according to the |test_type|.
-
-    Raises:
-      Exception: If |test_type| is not one of the TEST_TYPE_* constants.
-    """
-
-    if test_type == WebsiteTest.TEST_TYPE_PROMPT_FAIL:
-      self.PromptFailTest()
-    elif test_type == WebsiteTest.TEST_TYPE_PROMPT_SUCCESS:
-      self.PromptSuccessTest()
-    elif test_type == WebsiteTest.TEST_TYPE_SAVE_AND_AUTOFILL:
-      self.SaveAndAutofillTest()
-    else:
-      raise Exception("Unknown test type {}.".format(test_type))