类级别的setUp()方法和tearDown()方法
我们可以通过setUp()方法为每个测试方法都创建了一个Firefox or Chrome实例,并且在每个测试方法执行结束后都要关闭实例。能否让各个测试方法共用一个实例,而不要每次都创建一个新的实例呢?
通过使用setUpClass()方法和tearDownClass()方法及@classmethod标识来实现。这两个方法使我们可以在类级别来初始化数据,替代了方法级别的初始化,这样各个测试方法就可以共享这些初始化数据。在下面的例子中,代码修改为调用setUpClass()方法和tearDownClass()方法并且加上@classmethod标识。
import unittest
from selenium import webdriver
class SearchTests(unittest.TestCase):
@classmethod
def setUpClass(cls):
# create a new Firefox session
cls.driver = webdriver.Firefox()
cls.driver.implicitly_wait(30)
cls.driver.maximize_window()
# navigate to the application home pageself.driver.get('http://demo-store.seleniumacademy.com/')
def test_search_by_category(self):
...
...
def test_search_by_name(self):
...
...
@classmethod
def tearDownClass(cls):
# close the browser window
cls.driver.quit()
if __name__ == '__main__':
unittest.main(verbosity=2)
运行这个测试将看到仅创建一个Firefox实例,所有的测试都用同一个实例。
资料来源:Selenium自动化测试:基于Python语言