unittest测试框架（unittest框架的作用）

更多技术文章

unittest 是python 的单元测试框架， unittest 单元测试提供了创建测试用例，测试套件以及批量执行的方案。作为单元测试的框架， unittest 也是可以对程序最小模块的一种敏捷化的测试。

unittest 和 Junit 都是单元测试？区别在于： unittest 是基于功能测试的单元测试，是基于 UI 界面的功能性测试，而 Junit 是白盒单元测试框架。

解决了单个脚本重复的操作（导包，获取浏览器驱动，关闭浏览器）。

1、测试固件

1.1 setUp（）

在执行测试用例脚本之前进行初始化环境的方法。

1.2 tearDown（）

测试用例执行后，进行清理环境工作。

2、unittest 基本使用

测试用例的命名： test_ 。

from selenium import webdriver

import time

import unittest

class TestUnit1(unittest.TestCase):

# 获取浏览器的驱动

def setUp(self):

# 1、self 就是类的引用/实例

# 2、全局变量的定义：self.变量名

self.driver = webdriver.Chrome()

self.driver.maximize_window()

self.url = "https://www.baidu.com/"

self.driver.get(self.url)

time.sleep(3)

# 在百度中搜索信息

# 测试用例的命名： test_

def test_search1(self):

self.driver.find_element_by_id("kw").send_keys("顾一野")

self.driver.find_element_by_id("su").click()

time.sleep(6)

def test_search2(self):

self.driver.find_element_by_id("kw").send_keys("account")

self.driver.find_element_by_id("su").click()

time.sleep(6)

# 关闭浏览器

def tearDowm(self):

self.driver.quit()

# 一个入口

if __name__ == "__main__":

unittest.main()

3、测试套件（suit）

把不同文件下的测试用例组织起来形成的测试用例组。

把需要一次性执行的测试用例，放在一个套件中，就可以一次性运行。

import unittest

from Test import testbaidu1

from Test import testbaidu2

def creatSuit():

# 1、要把不同测试脚本的类中的需要执行的方法放在一个测试套件中

# suit = unittest.TestSuite()

# suit.addTest(testbaidu1.Baidu1("test_hao"))

# suit.addTest(testbaidu2.Baidu2("test_hao"))

# suit.addTest(testbaidu2.Baidu2("test_baidusearch"))

# return suit

# 2、如果需要把一个测试脚本中所有的测试用例都添加到suit中，怎么做？

# 第一种※：makeSuit

# suit = unittest.TestSuite()

# suit.addTest(unittest.makeSuite(testbaidu1.Baidu1))

# suit.addTest(unittest.makeSuite(testbaidu2.Baidu2))

# return suit

# 第二种：TestLoader

# suit1 = unittest.TestLoader().loadTestsFromTestCase(testbaidu1.Baidu1)

# suit2 = unittest.TestLoader().loadTestsFromTestCase(testbaidu2.Baidu2)

# suit = unittest.TestSuite([suit1, suit2])

# return suit

# 3、可以把一个文件夹下面所有的测试脚本中的测试用例放入测试套件

discover = unittest.defaultTestLoader.discover("../Test", pattern="testbaidu*.py", top_level_dir=None)

return discover

if __name__ == "__main__":

suit = creatSuit()

# verbersity= 0 测试用例成功多少，失败多少, 1 会标注哪个成功/失败, 2 会标注

runner = unittest.TextTestRunner()

runner.run(suit)

更多技术文章