unittst与webdriver的关系(二)

在初始python单元测试框架unittest与webdriver的关系(一)中，介绍了unittest与自动化的结合，下面介绍测试套件TestSuite()

构建测试套件

一、addTest()

当有多个或者几百测试用例的时候，我们必须需要一个集合，来把测试用例放在该集合中，来进行执行，幸运的是，unittest模块中提供了TestSuite类来生成测试套件，使用该类的构造函数可以生成一个测试套件的实例，该类提供了addTest来把单元测试用例加入到测试套件中，测试代码使用TestSuite的addTest进行更改，具体见如下的测试脚本：

#coding:utf-8

from selenium import webdriver

import unittest,sys,time

class BaiduTest(unittest.TestCase):

defsetUp(self):

self.driver=webdriver.Firefox()

self.driver.maximize_window()

self.driver.get('http://www.baidu.com')

self.driver.implicitly_wait(30)

deftestTitle(self):

self.assertEqual(u'百度一下，你就知道',self.driver.title)

deftestUrl(self):

self.assertEqual('https://www.baidu.com/',self.driver.current_url)

deftearDown(self):

self.driver.quit()

if __name__=='__main__':

suite=unittest.TestSuite()

suite.addTest(BaiduTest('testTitle'))

suite.addTest(BaiduTest('testUrl'))

unittest.TextTestRunner(verbosity=2).run(suite)

字体加粗的部分就是构建测试容器(测试套件)后，把测试用例添加到测试容器，来进行执行，这样的一个好处就是测试用例是按顺序的来执行。我们可以把suite写成一个静态方法来进行调用，那么如上实现的测试脚本就为：

#coding:utf-8

from seleniumimport webdriver

importunittest,sys,time

classBaiduTest(unittest.TestCase):

def setUp(self):

self.driver=webdriver.Firefox()

self.driver.maximize_window()

self.driver.get('http://www.baidu.com')

self.driver.implicitly_wait(30)

def testTitle(self):

self.assertEqual(u'百度一下，你就知道',self.driver.title)

def testUrl(self):

self.assertEqual('https://www.baidu.com/',self.driver.current_url)

def tearDown(self):

self.driver.quit()

@staticmethod

def suite():

suite=unittest.TestSuite()

suite.addTest(BaiduTest('testTitle'))

suite.addTest(BaiduTest('testUrl'))

return suite

if __name__=='__main__':

unittest.TextTestRunner(verbosity=2).run(BaiduTest.suite())

二、makeSuite()

利用addTest()可以实现测试用例的顺序执行，但是在自动化的测试中，每个case在一个测试容器中，应该是随机执行的，更加容易让人接受，这样也就每个case之间不会有互相依赖和牵制，另外一方面，在一个测试容器中，测试的case的数量最小在十以内，最多可能在几百或者几千，如果采用addTest的方法，那么对每个case都要进行添加，从而导致很多垃圾代码，如suite.addTest(BaiduTest('case'))

...

很显然，这样的代码，并不是我们想要的最好的结果，后期维护测试代码，也会带来无谓的时间浪费和增加人力的成本，自动化的优点就是符合在敏捷的规格中，来达到测试的结果，从而达到保障质量。可性的是，unittest提供了makeSuite的方法，makeSuite可以实现测试用例类内所有的测试用例组成的测试容器TestSuite，我们可以把如上的代码使用makeSuite来调整下，调整后的代码见如下：

#coding:utf-8

from selenium import webdriver

import unittest,sys,time

class BaiduTest(unittest.TestCase):

defsetUp(self):

self.driver=webdriver.Firefox()

self.driver.maximize_window()

self.driver.get('http://www.baidu.com')

self.driver.implicitly_wait(30)

deftestTitle(self):

self.assertEqual(u'百度一下，你就知道',self.driver.title)

deftestUrl(self):

self.assertEqual('https://www.baidu.com/',self.driver.current_url)

deftearDown(self):

self.driver.quit()

if __name__=='__main__':

suite=unittest.TestSuite(unittest.makeSuite(BaiduTest))

unittest.TextTestRunner(verbosity=2).run(suite)

这样，我们就不需要考虑测试容器中有多少case，也就没有了利用addTest产生的烦恼。我们把suite写成一个静态方法，来重构代码：

#coding:utf-8

from selenium import webdriver

import unittest,sys,time

class BaiduTest(unittest.TestCase):

defsetUp(self):

self.driver=webdriver.Firefox()

self.driver.maximize_window()

self.driver.get('http://www.baidu.com')

self.driver.implicitly_wait(30)

deftestTitle(self):

self.assertEqual(u'百度一下，你就知道',self.driver.title)

deftestUrl(self):

self.assertEqual('https://www.baidu.com/',self.driver.current_url)

deftearDown(self):

self.driver.quit()

@staticmethod

def suite():

suite=unittest.TestSuite(unittest.makeSuite(BaiduTest))

return suite

if __name__=='__main__':

unittest.TextTestRunner(verbosity=2).run(BaiduTest.suite())

三、TestLoader()

TestLoader用于创建类和模块的测试套件，一般的情况下，使用TestLoader().loadTestsFromTestCase(TestClass)来加载测试类。见如下使用TestLoader调整的测试脚本：

#coding:utf-8

from selenium import webdriver

import unittest,sys,time

class BaiduTest(unittest.TestCase):

defsetUp(self):

self.driver=webdriver.Firefox()

self.driver.maximize_window()

self.driver.get('http://www.baidu.com')

self.driver.implicitly_wait(30)

deftestTitle(self):

self.assertEqual(u'百度一下，你就知道',self.driver.title)

deftestUrl(self):

self.assertEqual('https://www.baidu.com/',self.driver.current_url)

deftearDown(self):

self.driver.quit()

if __name__=='__main__':

suite=unittest.TestLoader().loadTestsFromTestCase(BaiduTest)

unittest.TextTestRunner(verbosity=2).run(suite)

我们把suite进行封装下，那么调整后的测试代码为：

#coding:utf-8

from selenium import webdriver

import unittest,sys,time

class BaiduTest(unittest.TestCase):

defsetUp(self):

self.driver=webdriver.Firefox()

self.driver.maximize_window()

self.driver.get('http://www.baidu.com')

self.driver.implicitly_wait(30)

deftestTitle(self):

self.assertEqual(u'百度一下，你就知道',self.driver.title)

deftestUrl(self):

self.assertEqual('https://www.baidu.com/',self.driver.current_url)

deftearDown(self):

self.driver.quit()

@staticmethod

def suite():

suite=unittest.TestLoader().loadTestsFromTestCase(BaiduTest)

return suite

if __name__=='__main__':

unittest.TextTestRunner(verbosity=2).run(BaiduTest.suite())

四、discover()

discover是通过递归的方式到其子目录中从指定的目录起始，找到所有测试模块并返回一个包含他们对象的TestSuite，然后进行加载与模式匹配唯一的测试文件。如下的实例就是在'D:/git/PyCharm/TestCase的目录下，存在test_share.py和test_share2模块的测试文件，我们利用正则表达式，程序会在'D:/git/PyCharm/TestCase的目录下，寻找到test_*.py的文件，然后进行加载并执行测试对象，见如下的测试代码：

#coding:utf-8

from selenium import webdriver

import unittest,sys,time

class BaiduTest(unittest.TestCase):

defsetUp(self):

self.driver=webdriver.Firefox()

self.driver.maximize_window()

self.driver.get('http://www.baidu.com')

self.driver.implicitly_wait(30)

deftestTitle(self):

self.assertEqual(u'百度一下，你就知道',self.driver.title)

deftestUrl(self):

self.assertEqual('https://www.baidu.com/',self.driver.current_url)

deftearDown(self):

self.driver.quit()

@staticmethod

def suite():

suite=unittest.defaultTestLoader.discover(

'D:/git/PyCharm/TestCase',

pattern='test_*.py',

top_level_dir=None

)

return suite

if __name__=='__main__':

unittest.TextTestRunner(verbosity=2).run(BaiduTest.suite())

见执行后的测试结果：