unittest设置断言

unittest设置断言

COURSE ARRANGEMENT

断言

unittest单元测试框架中也提供断言的方式，用来判断用例是否执行成功。

**unittest常用断言 **

unittest框架的TestCase类提供如下方法用于测试结果的判断：
方法****检查assertEqual(a,b)a == bassertNotEqual(a,b)a != bassertTrue(x)bool(x) is TrueassertFalse(x)bool(x) is FalseassertIs(a,b)a is bassertIsNot(a,b)a is not bassertIsNone(x)x is NoneassertIsNotNone(x)x is not NoneassertIn(a,b)a in bassertNotIn(a,b)a not in b
用法：

在用例执行之后，设置断言，检查用例是否执行通过，如：

import unittestclass Test(unittest.TestCase):    def test01(self):        '''        判断 a = b        '''        a = '111'        b = '1112'        self.assertEqual(a,b)if __name__ == '__main__':    unittest.main()

执行成功返回的结果，返回“.”：

若断言失败，页面直接返回对应的报错信息，用例执行不通过，状态为“F”：

也可以自定义报错信息：

import unittestclass Test(unittest.TestCase):    def test01(self):        '''        判断 a = b        '''        a = '111'        b = '1112'        self.assertEqual(b,a,msg='报错原因：{}没有包含{}'.format(a,b))if __name__ == '__main__':    unittest.main()

执行报错信息如下，可以按照自己设置的报错信息打印，更有利于问题定位：

结合百度的搜索页面，设置断言如下：

#导入Webdriver，UI自动化主要靠Webdriver实现from selenium import webdriver#导入时间模块import time#导入unittest单元测试模块import unittest#定义百度网页登录的测试用例集class BaiduLogin(unittest.TestCase):  #定义Setup，用于初始化测试固件。本次设置的含义为在执行每个测试用例之前，打开浏览器，设置url为https://www.baidu.com    def setUp(self):        self.url = 'https://www.baidu.com/'        self.browser = webdriver.Chrome()    #定义teardown，用于销毁测试固件。本次设置的含义为在执行每个测试用例之后，关闭浏览器    #所以当前脚本的执行顺序为：    ##setup-->test_query_001-->teardown-->setup-->test_login_002-->teardown    def tearDown(self):      self.browser.quit()    def test_001_query(self):      #创建一个名称为test_query_001的测试用例，在浏览器中打开百度首页           self.browser.get(self.url)      #将窗口最大化           self.browser.maximize_window()      #在搜索框中输入关键词“123”      self.browser.find_element_by_id('kw').send_keys('123')      #点击“百度一下”按钮      self.browser.find_element_by_id('su').click()      #print("查询成功")      #等待3S，查看页面返回结果      time.sleep(3)      #添加断言，搜索成功后页面会返回“百度为您找到相关结果约100,000,000个”      #定位后将该元素中包含的文字提取出来      a = self.browser.find_element_by_class_name('nums_text').text      #定义一个变量b，使它的内容包含部分页面搜索成功后返回的文字      b = "找到相关结果"      #断言，若a包含b，则用例执行成功，在结果中打印“搜索成功”      self.assertIn(b,a,msg='报错原因：{}没有包含{}'.format(a,b)')      print("搜索成功")if __name__ == '__main__':    unittest.main()

用例通过的执行结果：