unittest是python内置的一个单元测试框架，在学习怎么使用它之前，我们先来了解它的一些概念和原理。
- Test Case：测试用例，一个TestCase的实例就是一个测试用例。什么是测试用例呢？就是一个完整的测试流程，包括测试前准备环境的搭建(setUp)，执行测试代码(run)，以及测试后环境的还原(tearDown)。单元测试(unit test)的本质也就在这里，一个测试用例是一个完整的测试单元，通过运行这个测试单元，可以对某一个问题进行验证。
- Test Suite：测试套件，就是多个测试用例集合在一起
- Test Runner：测试执行，用来执行测试用例，其中的run(test)会执行TestSuite/TestCase
- Test Fixture：对一个测试用例环境的搭建和销毁，是一个fixture，通过覆盖TestCase的setUp()和tearDown()方法来实现。这个有什么用呢？比如说在这个测试用例中需要访问数据库，那么可以在setUp()中建立数据库连接以及进行一些初始化，在tearDown()中清除在数据库中产生的数据，然后关闭连接。注意tearDown的过程很重要，要为以后的TestCase留下一个干净的环境。
了解了概念后，我们来看一个登陆的测试脚本。
代码：
# coding = utf-8
import unittest
import warnings
from selenium import webdriver
from time import sleep
# 驱动文件路径
driverfile_path = r'D:\coship\Test_Framework\drivers\IEDriverServer.exe' class CmsLoginTest(unittest.TestCase):
def setUp(self):
# 这行代码的作用是忽略一些告警打印
warnings.simplefilter("ignore", ResourceWarning)
print("test start!")
self.driver = webdriver.Ie(executable_path=driverfile_path)
self.driver.get("http://172.21.13.83:28080/") def tearDown(self):
self.driver.quit()
print("test end!") def test_login1(self):
'''用户名、密码为空'''
self.driver.find_element_by_css_selector("#imageField").click()
error_message1 = self.driver.find_element_by_css_selector("[for='loginName']").text
error_message2 = self.driver.find_element_by_css_selector("[for='textfield']").text
self.assertEqual(error_message1, '用户名不能为空')
self.assertEqual(error_message2, '密码不能为空')
print("用例test_login1执行结束") def test_login3(self):
'''用户名、密码正确'''
self.driver.find_element_by_css_selector("[name='admin.loginName']").send_keys("autotest")
self.driver.find_element_by_css_selector("[name='admin.password']").send_keys("")
self.driver.find_element_by_css_selector("#imageField").click()
sleep(1)
self.driver.switch_to.frame("topFrame")
username = self.driver.find_element_by_css_selector("#nav_top>ul>li>a").text
self.assertEqual(username,"autotest")
print("用例test_login3执行结束") def test_login2(self):
'''用户名正确，密码错误'''
self.driver.find_element_by_css_selector("[name='admin.loginName']").send_keys("autotest")
self.driver.find_element_by_css_selector("[name='admin.password']").send_keys("")
self.driver.find_element_by_css_selector("#imageField").click()
error_message = self.driver.find_element_by_css_selector(".errorMessage").text
self.assertEqual(error_message, '密码错误,请重新输入!')
print("用例test_login2执行结束") def login4(self):
'''用户名不存在'''
self.driver.find_element_by_css_selector("[name='admin.loginName']").send_keys("test007")
self.driver.find_element_by_css_selector("[name='admin.password']").send_keys("")
self.driver.find_element_by_css_selector("#imageField").click()
error_message = self.driver.find_element_by_css_selector(".errorMessage").text
self.assertEqual(error_message, '用户不存在!')
print("用例login4执行结束") if __name__ == "__main__":
unittest.main()
执行结果：
"C:\Program Files\Python36\python.exe" D:/Git/Test_Framework/utils/1.py
test start!
用例test_login1执行结束
.test end!
test start!
用例test_login2执行结束
.test end!
test start!
用例test_login3执行结束
test end!
.
----------------------------------------------------------------------
Ran 3 tests in 34.044s OK Process finished with exit code 0
从脚本和运行结果我们可以得到以下结论：
- 要使用unittest这个测试框架时，需要先导入unittest
- 定义一个测试类时，要继承unittest.TestCase这个类
- setUp()是测试用例执行前的环境准备，tearDown()是测试用例执行结束后的环境恢复，每个测试用例都会执行一次setUp()和tearDown()
- 测试用例必须以test开头，否则不会被执行。
- 测试用例执行顺序是根据用例名称顺序来执行的