Test-driven development(TDD)开发模式在今天已经不是什么新奇的事了，它的开发思维是在开发一个产品功能的时候，先编写好该功能的测试代码，在编写开发比如，比如要写二个数相除的函数，那么它的测试代码应该为：
#!/usr/bin/env Python
#coding:utf-8
import unittest
class TestDiv(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
def test_001(self):
self.assertEqual(div(1,1),1)
def test_002(self):
self.assertRaises(ZeroDivisionError,div,1,0)
if __name__=='__main__':
unittest.main(verbosity=2)
执行如上的代码，会提示如下NameError: global name 'div' is not defined的错误信息，事实上我们自己也是非常明白，因为我们实际没有实现这样的一个函数，而是先写了改函数功能的测试代码，那么现在来写函数部分，见完善后的源码：
#!/usr/bin/env python
#coding:utf-8
def div(a,b):
return a/b
import unittest
class TestDiv(unittest.TestCase):
def setUp(self):
pass
def tearDown(self):
pass
def test_001(self):
self.assertEqual(div(1,1),1)
def test_002(self):
self.assertRaises(ZeroDivisionError,div,1,0)
if __name__=='__main__':
unittest.main(verbosity=2)
再次执行我们的测试代码，就会通过，见执行的结果结果：
不论对于开发还是测试，都离不开单元测试框架，对于开发而言使用单元测试框架，可以编写测试代码来验证验证自己编写的功能是否正确，对于测试而言，使用单元测试框架，可以编写自动化的测试用例，在python中单元测试框架是Pyunit，即unittest,unittest我一直认为是一个很优秀的单元测试框架，至少我是这样认为在，它是python的标准库，官方详细的地址是：https://docs.python.org/2/library/unittest.html。单元测试支持测试自动化、 共享的安装程序和关闭代码测试、 聚合成集合，测试和报告框架从测试的独立性。单元测试模块提供可以很容易地支持这些素质的一组测试的类。关于unittest测试框架建议可以到官方查看详细的说明以及演示的实例。unittest各个模块的关系为：
在一个完整的单元测试用例中，是包含了测试固件(setUp()和tearDown())，在测试执行的阶段，我们更加愿意使用测试套件(TestSuite())来组织每个测试用例来执行(TestRunner)并得到测试结果(TestReport)，什么 是测试固件，在unittest中，setUp()与tearDown()被成为测试固件，某些人称为钩子(仅仅只一个称呼而已)，它的主要目标初始化测试用例，执行测试用例后，对测试用例执行的结果做后期的处理，我们再看上面的测试用例，总共是二个测试用例，不管执行那个测试用例，都会执行setUp()和tearDown()，也就是说，在一个测试类中，如果有N个测试用例，在执行该测试类中的测试用例的时候，会执行N次setUp()和tearDown()，我们修改源码并执行来看结果，见源码：
#!/usr/bin/env python
#coding:utf-8
def div(a,b):
return a/b
import unittest
class TestDiv(unittest.TestCase):
def setUp(self):
print u'开始...'
def tearDown(self):
print u'结束...'
def test_001(self):
self.assertEqual(div(1,1),1)
def test_002(self):
self.assertRaises(ZeroDivisionError,div,1,0)
if __name__=='__main__':
unittest.main(verbosity=2)
见执行的结果截图：
依据结果可以看到，执行了二个测试用例，也执行了2次setUp()和tearDown()方法，如果这样你感觉不明显，可以结合selenium的测试框架来看更加直观，见源码：
#!/usr/bin/env python
#coding:utf-8
import unittest
from selenium import webdriver
class TestDiv(unittest.TestCase):
def setUp(self):
self.driver=webdriver.Firefox()
self.driver.get('http://www.baidu.com')
def tearDown(self):
self.driver.quit()
def test_001(self):
self.assertEqual(self.driver.title,u'百度一下，你就知道')
def test_002(self):
self.assertEqual(self.driver.current_url,'https://www.baidu.com/')
if __name__=='__main__':
unittest.main(verbosity=2)
执二后，会看到打开浏览器二次，当然关闭浏览器也是二次，这里不在进行截图了。那么可不可以让测试固件只执行一次了，也就是说在一个测试类中，有N个测试用例，执行这个测试类中的测试用例后，测试固件只执行一次。当然是可以的，unittest提供了这样的解决方案，在这里钩子方法使用的是类方法(关于实例方法，类方法，静态方法不熟悉的建议看下python的OOP部分)，我们重构下代码来实现这样的一个过程，见源码：
#!/usr/bin/env python
#coding:utf-8
import unittes
tfrom selenium import webdriver
class TestDiv(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.driver=webdriver.Firefox()
cls.driver.get('http://www.baidu.com')
@classmethod
def tearDownClass(cls):
cls.driver.quit()
def test_001(self):
self.assertEqual(self.driver.title,u'百度一下，你就知道')
def test_002(self):
self.assertEqual(self.driver.current_url,'https://www.baidu.com/')
if __name__=='__main__':
unittest.main(verbosity=2)
OK，带着疑问继续触发，在一个测试用例中，测试用例想有顺序的执行该如何实现了，可以使用addTest()方法来实现，也就是说，把需要执行的或者按顺序的添加到测试套件中，见修改后的源码：
#!/usr/bin/env python
#coding:utf-8
import unittest
from selenium import webdriver class TestDiv(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.driver=webdriver.Firefox()
cls.driver.get('http://www.baidu.com')
@classmethod
def tearDownClass(cls):
cls.driver.quit()
def test_001(self):
self.assertEqual(self.driver.title,u'百度一下，你就知道')
def test_002(self):
self.assertEqual(self.driver.current_url,'https://www.baidu.com/')
if __name__=='__main__':
suite=unittest.TestSuite()
suite.addTest(TestDiv('test_001'))
unittest.TextTestRunner(verbosity=2).run(suite)
或者我们可以把代码重构为：
#!/usr/bin/env python
#coding:utf-8
import unittest
from selenium import webdriver class TestDiv(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.driver=webdriver.Firefox()
cls.driver.get('http://www.baidu.com')
@classmethod
def tearDownClass(cls):
cls.driver.quit()
def test_001(self):
self.assertEqual(self.driver.title,u'百度一下，你就知道')
def test_002(self):
self.assertEqual(self.driver.current_url,'https://www.baidu.com/')
@staticmethod def suites():
tests=['test_001','test_002'] return unittest.TestSuite(map(TestDiv,tests))
if __name__=='__main__':
unittest.TextTestRunner(verbosity=2).run(TestDiv.suites())
事实上，我个人不赞成使用addTest()方法，来把测试用例依次添加到测试套件中，理由非常简单，因为在一个测试类中，测试用例是非常多的，这样添加或者删除实在是浪费时间，我们可以把源码修改下，来实现执行一个测试，某些用例不执行的可以忽略，使用的方法makeSuite()，见修改后的源码：
#!/usr/bin/env python
#coding:utf-8
import unittest
from selenium import webdriver class TestDiv(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.driver=webdriver.Firefox()
cls.driver.get('http://www.baidu.com')
@classmethod
def tearDownClass(cls):
cls.driver.quit()
def test_001(self):
self.assertEqual(self.driver.title,u'百度一下，你就知道')
@unittest.skip(u'忽略该测试用例,谢谢!') def test_002(self):
self.assertEqual(self.driver.current_url,'https://www.baidu.com/')
if __name__=='__main__':
suite=unittest.TestSuite(unittest.makeSuite(TestDiv))
unittest.TextTestRunner(verbosity=2).run(suite)
见执行后的截图：
在这里总结的，只是个人喜好，每个人可以根据自己的实际情况来进行，这只是提供了一种选择而已。还有另外一种方法是TestLoader()加载测试类来执行测试类中的所有测试用例，见源码：
#!/usr/bin/env python
#coding:utf-8
import unittest
from selenium import webdriver
class TestDiv(unittest.TestCase):
@classmethod def setUpClass(cls):
cls.driver=webdriver.Firefox()
cls.driver.get('http://www.baidu.com')
@classmethod def tearDownClass(cls):
cls.driver.quit()
def test_001(self):
self.assertEqual(self.driver.title,u'百度一下，你就知道')
def test_002(self):
self.assertEqual(self.driver.current_url,'https://www.baidu.com/')
if __name__=='__main__':
suite=unittest.TestLoader().loadTestsFromTestCase(TestDiv)
unittest.TextTestRunner(verbosity=2).run(suite)
在一个测试用例中，会有期望结果这个说法，来验证这个测试用例是通过还是失败，在unittest的测试框架中，也提供了assert，我们先来看python中的断言assert，来修改下源码，看看python实际代码的断言，见源码：
#!/usr/bin/env python
#coding:utf-8
import unittest
from selenium import webdriver
class TestDiv(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.driver=webdriver.Firefox()
cls.driver.get('http://www.baidu.com')
@classmethod
def tearDownClass(cls):
cls.driver.quit()
def test_001(self):
assert self.driver.title in u'百度一下，你就知道'
def test_002(self):
assert self.driver.current_url in 'https://www.baidu.com/'
if __name__=='__main__':
suite=unittest.TestLoader().loadTestsFromTestCase(TestDiv)
unittest.TextTestRunner(verbosity=2).run(suite)
见执行的结果的截图：
上面的仅仅是python语言自带的assert，在unittest中提供了非常丰富的断言，具体见如下图：
下面就演示几个断言的使用方法，见案例的源码：
#!/usr/bin/env python
#coding:utf-8
import unittes
tfrom selenium import webdriver
class TestDiv(unittest.TestCase):
@classmethod
def setUpClass(cls):
cls.driver=webdriver.Firefox()
cls.driver.get('http://www.baidu.com')
@classmethod
def tearDownClass(cls):
cls.driver.quit()
def test_001(self):
self.assertEqual(self.driver.title, u'百度一下，你就知道')
def test_002(self):
self.assertTrue(self.driver.find_element_by_id('kw').is_enabled())
def test_003(self):
self.assertIsNot(self.driver.current_url,'www.baidu.com')
if __name__=='__main__':
suite=unittest.TestLoader().loadTestsFromTestCase(TestDiv)
unittest.TextTestRunner(verbosity=2).run(suite)
见执行后的结果：
unittest的断言是非常丰富的，这里就不在演示了，遇到了不知道，可以到官方查看。
在python中，提供了HTMLTestRunner.py来生成测试报告，把该文件下载后，直接放到C:\Python27\Lib的目录下，就可以导入该模块使用了，见该实现的代码：
#!/usr/bin/env python
#coding:utf-8
import unittest from selenium import webdriver
import HTMLTestRunner
class TestDiv(unittest.TestCase):
@classmethod def setUpClass(cls):
cls.driver=webdriver.Firefox()
cls.driver.get('http://www.baidu.com')
@classmethod def tearDownClass(cls):
cls.driver.quit()
def test_001(self):
self.assertEqual(self.driver.title, u'百度一下，你就知道')
def test_002(self):
self.assertTrue(self.driver.find_element_by_id('kw').is_enabled())
def test_003(self):
self.assertIsNot(self.driver.current_url,'www.baidu.com')
if __name__=='__main__':
suite=unittest.TestLoader().loadTestsFromTestCase(TestDiv)
runner=HTMLTestRunner.HTMLTestRunner(
stream=file('testReport.html','wb'),
title=u'TestReport',
description=u'测试报告详细信息'
)
runner.run(suite)
执行后，会在当前目录下生成testReport.html的测试报告，见该报告的截图：
