selenium_test_case.py 文件源码

def setUpClass(cls):
        super(SeleniumTestCase, cls).setUpClass()

        # load the webdriver setting as late as possible
        # this is needed when no web driver is specified and no functional tests should be run
        from common.settings.webdriver import WEBDRIVER

        if WEBDRIVER == "firefox":
            cls.selenium = webdriver.Firefox()
        elif WEBDRIVER == "chrome":
            cls.selenium = webdriver.Chrome()
        elif WEBDRIVER == "safari":
            cls.selenium = webdriver.Safari()
        else:
            raise Exception("Webdriver not configured probably!")
        cls.selenium.implicitly_wait(10)

        # wrap also the find_element(s)_by methods of the webdriver and webelement classes
        decorate(cls.selenium.__class__, wait_for, "find_element_by_")
        decorate(cls.selenium.__class__, wait_for, "find_elements_by_")
        decorate(WebElement, wait_for, "find_element_by_")
        decorate(WebElement, wait_for, "find_elements_by_")