unittest是Python语言自带的单元测试框架，原名PyUnit.
认识unittest
在unittest框架中有4个重要概念：test fixture、test case、test suite和test runner。详情：
test fixture:测试用例环境的搭建和销毁，通过覆盖TestCase的setUp()和tearDown()方法实现，非unittest属性。
unittest.TestCase：TestCase类，所有测试用例类继承的父类。
unittest.TestSuite()：unittest框架的TestSuite()类是用来创建测试套件的。
unittest.TextTestRunner():unittest框架的TextTextRunner()类，通过该类的run()方法来运行suite所组装的测试用例，入参为suite测试套件。
unittest.defaultTestLoader(): unittest框架的defaultTestLoader()类，通过该类的discover()方法找到指定目录下的所以测试模块，匹配到的文件均可以被加载。
unittest.main():可以方便的将一个单元测试模块变为可直接运行的测试脚本，main() 方法使用TestLoader类来搜索所有包含在该模块中以“test”命名开头的测试方法，并自动执行他们。
unittest.skip():装饰器，当运行用例时，有些用例可能不想执行等，可用装饰器暂时屏蔽该条测试用例。一种常见的用法就是比如说想调试某一个测试用例，想先屏蔽其他用例就可以用装饰器屏蔽。
@unittest.skip(reason): skip(reason)装饰器：无条件跳过装饰的测试，并说明跳过测试的原因。
@unittest.skipIf(reason): skipIf(condition,reason)装饰器：条件为真时，跳过装饰的测试，并说明跳过测试的原因。
@unittest.skipUnless(reason): skipUnless(condition,reason)装饰器：条件为假时，跳过装饰的测试，并说明跳过测试的原因。
@unittest.expectedFailure(): expectedFailure()测试标记为失败。
#widget.py 将要被测试的类
class Widget:
def __init__(self, size = (40, 40)):
self._size = size
def getSize(self):
return self._size
def resize(self, width, height):
if width 0 or height < 0:
raise ValueError, "illegal size"
self._size = (width, height)
from widget import Widget
import unittest
# 执行测试的类
class WidgetTestCase(unittest.TestCase):
def setUp(self):
self.widget = Widget()
def tearDown(self):
self.widget = None
def testSize(self):
self.assertEqual(self.widget.getSize(), (40, 40))
def testResize(self):
self.widget.resize(100, 100)
self.assertEqual(self.widget.getSize(), (100, 100)) if __name__ == "__main__":
# 构造测试集
suite = unittest.TestSuite()
suite.addTest(WidgetTestCase("testSize"))
suite.addTest(WidgetTestCase("testResize")) # 执行测试
runner = unittest.TextTestRunner()
runner.run(suite)
unittest的web应用
python没有专门的自动化测试框架，故应用unittest单元测试框架进行web自动化。
import unittest
from selenium import webdriver
from selenium.webdriver.common.keys import Keys class PythonOrgSearch(unittest.TestCase): def setUp(self):
self.driver = webdriver.Firefox() def test_search_in_python_org(self):
driver = self.driver
driver.get("http://www.python.org")
self.assertIn("Python", driver.title)
elem = driver.find_element_by_name("q")
elem.send_keys("python")
elem.send_keys(Keys.RETURN) #类似键盘enter操作
assert "No results found." not in driver.page_source def tearDown(self):
self.driver.close() if __name__ == "__main__":
unittest.main()
总结unittest框架好处：
1.提供用例组织和执行，简单说就是规定了测试用例的执行顺序
2.提供丰富的比较方法，将实际结果和预期结果进行比较断言
3.提供丰富的日志，用例执行结果都有一个很好的记录