1.按照顺序执行
代码如下:
import unittest from selenium import webdriver class BaiduTest(unittest.TestCase): def setUp(self): self.driver=webdriver.Chrome() self.driver.maximize_window() self.driver.implicitly_wait(30) self.driver.get("http://www.baidu.com") def tearDown(self): self.driver.quit() def test_baidu_news(self): "验证百度首点击新闻后的跳转" self.driver.find_element_by_link_text("新闻").click() url = self.driver.current_url self.assertEqual(url,'http://news.baidu.com/') def test_baidu_map(self): "验证百度首页点击地图后的跳转" self.driver.find_element_by_link_text("地图").click() self.driver.get("http://www.baidu.com") if __name__ == '__main__': suite= unittest.TestSuite() suite.addTest(BaiduTest("test_baidu_news")) suite.addTest(BaiduTest("test_baidu_map")) unittest.TextTestRunner().run(suite)
注解:首先对TestSuite类进行实例化,使之成为一个对象suite,然后调用TestSuite类中addTest方法,把测试用例添加到测试套件中,最后执行测试套件,从而执行测试套件中的测试用例。运行以上代码后,测试用例会按照添加到测试套件的顺序执行,也就是说先添加进去的先执行,后添加进去的后执行。
2.按测试类执行
import unittest from selenium import webdriver class BaiduTest(unittest.TestCase): def setUp(self): self.driver=webdriver.Chrome() self.driver.maximize_window() self.driver.implicitly_wait(30) self.driver.get("http://www.baidu.com") def tearDown(self): self.driver.quit() def test_baidu_news(self): "验证百度首点击新闻后的跳转" self.driver.find_element_by_link_text("新闻").click() url = self.driver.current_url self.assertEqual(url,'http://news.baidu.com/') def test_baidu_map(self): "验证百度首页点击地图后的跳转" self.driver.find_element_by_link_text("地图").click() self.driver.get("http://www.baidu.com") if __name__ == '__main__': suite = unittest.TestSuite(unittest.makeSuite(BaiduTest)) unittest.TextTestRunner().run(suite)
注解:以上代码可以看出,在测试套件TestSuite类中,unittest模块调用了makeSuite的方法,makeSuite的参数是testCaseClass,也就是测试类。
3.加载测试类
在unittest模块中也可以使用TestLoader类来加载测试类,也就是说TestLoader加载测试类并将它们返回到TestSuite中,TestLoader类的应用代码如下:
import unittest from selenium import webdriver class BaiduTest(unittest.TestCase): def setUp(self): self.driver=webdriver.Chrome() self.driver.maximize_window() self.driver.implicitly_wait(30) self.driver.get("http://www.baidu.com") def tearDown(self): self.driver.quit() def test_baidu_news(self): "验证百度首点击新闻后的跳转" self.driver.find_element_by_link_text("新闻").click() url = self.driver.current_url self.assertEqual(url,'http://news.baidu.com/') def test_baidu_map(self): "验证百度首页点击地图后的跳转" self.driver.find_element_by_link_text("地图").click() self.driver.get("http://www.baidu.com") if __name__ == '__main__': suite = unittest.TestLoader().loadTestsFromTestCase(BaiduTest) unittest.TextTestRunner().run(suite)
4.按照测试模块执行
在TestLoader类中也可以按模块来执行测试。在python中,一个python文件就是一个模块,一个模块中可以有N个测试类,一个测试类中可以有N个测试用例。按照模块执行的代码如下:
import unittest from selenium import webdriver class BaiduTest(unittest.TestCase): def setUp(self): self.driver=webdriver.Chrome() self.driver.maximize_window() self.driver.implicitly_wait(30) self.driver.get("http://www.baidu.com") def tearDown(self): self.driver.quit() def test_baidu_news(self): "验证百度首点击新闻后的跳转" self.driver.find_element_by_link_text("新闻").click() url = self.driver.current_url self.assertEqual(url,'http://news.baidu.com/') def test_baidu_map(self): "验证百度首页点击地图后的跳转" self.driver.find_element_by_link_text("地图").click() self.driver.get("http://www.baidu.com") class BaiduMap(unittest.TestCase): def setUp(self): self.driver = webdriver.Chrome() self.driver.maximize_window() self.driver.implicitly_wait(30) self.driver.get("http://www.baidu.com") def tearDown(self): self.driver.quit() def test_baidu_map(self): "验证:测试百度首页点击地图后的跳转" self.driver.find_element_by_link_text("地图").click() self.driver.get("http://baidu.com") if __name__ == '__main__': suite = unittest.TestLoader().loadTestsFromModule("test.py") unittest.TextTestRunner(verbosity=2).run(suite) if __name__ == '__main__': suite = unittest.TestLoader().loadTestsFromTestCase(BaiduTest) unittest.TextTestRunner().run(suite)
注解:在以上代码中可以看到,测试类分别是BaiduMap和BaiduTest,模块名称为test.py,TestLoader类直接调用loadTestFromModule方法返回给指定模块中包含的所有测试用例套件。
5.优化测试套件
以上实例是把测试套件写在了main主函数中,也可以单独把测试套件写成一个方法来调用这里以加载测试类为例,把测试套件写成一个单独的方法,代码如下:
from selenium import webdriver import unittest class BaiduTest(unittest.TestCase): def setUp(self): self.driver = webdriver.Chrome() self.driver.maximize_window() self.driver.get('http://www.baidu.com') self.driver.implicitly_wait(30) def tearDown(self): self.driver.quit() def test_baidu_news(self): self.driver.find_element_by_link_text('新').click() def test_baid_taps(self): self.driver.find_element_by_link_text('地图').click() @staticmethod def suite(testcaseClass): suite = unittest.TestLoader().loadTestsFromTestCase(testcaseClass) return suite if __name__ == '__main__': unittest.TextTestRunner(verbosity=2).run(BaiduTest.suite(BaiduTest))