Web自动化测试：测试用例断言！

运行测试用例时，需要判断用例是否执行成功，此时需要有一个我们期望的结果来进行验证。这里unittest中，如果一个case执行的过程中报错，或者我们判断结果不符合期望，就会判定此条用例执行失败，判断的条件主要是根据断言来实现，这节主要学习下断言的使用。

一、断言的方法
1.1 TestCase类中的部分断言方法

判断条件是否成立

assert_(expr,msg)

expr:判断条件，返回Boolean值
msg:判断失败后显示的信息
判断两个数据的值是否相同

assertEqual(first,second,msg)

first:第一个数据
second:第二个数据
msg:判断失败后显示的信息
判断两个数据的值是否不相同

assertEqual(first,second,msg)

first:第一个数据
second:第二个数据
msg:判断失败后显示的信息
判断条件是否为真

assertTrue(expr,msg)

expr:判断条件
msg:判断失败后显示的信息
判断条件是否为假

assertFalse(expr,msg)

expr:判断条件
msg:判断失败后显示的信息
判断两个数据是否相同

assertIs(first,second,msg)

first:第一个数据
second:第二个数据
msg:判断失败后显示的信息
判断两个数据是否不相同

assertIsNot(first,second,msg)

first:第一个数据
second:第二个数据
msg:判断失败后显示的信息
判断数据是否存在

assertIsNone(obj,msg)

obj:所传的数据
msg:判断失败后显示的信息
判断数据是否不存在

assertIsNotNone(obj,msg)

obj:所传的数据
msg:判断失败后显示的信息
判断数据a是否包含在数据b中

assertIn(a,b,msg)

a:数据a
b:数据b
msg:判断失败后显示的信息
判断数据a是否不包含在数据b中

assertNotIn(a,b,msg)

a:数据a
b:数据b
msg:判断失败后显示的信息
1.2 直接使用assert断言

习惯直接使用python中的assert断言，替代unittest.TestCase中的断言方法：

具体可以直接使用：

assert a == b

assert a != b

assert a in b

assert a not in b

assert a in None

assert a in not None

二、在框架中使用断言

由于在使用unittest中，每个类都继承了unittest.TestCase，所以使用时，直接在方法里使用self来调用断言方法。

import unittest
 
class Demo(unittest.TestCase):
 
    def test_01(self):
        self.assertEqual("demo","demo","两者不相等")

如果直接使用assert断言，则可以直接调用。

import unittest
 
class Demo(unittest.TestCase):
 
    def test_01(self):
        assert "demo" == "demo"

三、实例演示

这里使用百度贴吧的例子，第一个case是打开新页面，然后判断新页面的title是否是我们期望的，第二个case则是我们故意判断一个错误的title，使得断言失败，这两个case也分别使用上述的两种断言方法。

# -*- coding:utf-8 -*-
from selenium import webdriver
import unittest
from time import sleep
 
 
class PcLogin(unittest.TestCase):
 
    @classmethod
    def setUpClass(cls):
        print("所有测试开始")
 
    def setUp(self):
        print("测试开始")
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(30)
        self.driver.set_page_load_timeout(30)
        self.driver.get("https://tieba.baidu.com/")
 
    def test_01_demo(self):
        """打开新页签"""
        self.driver.find_element_by_css_selector("a[title = '娱乐明星']").click()
        all = self.driver.window_handles
        self.driver.switch_to.window(all[1])
        sleep(5)
        assert self.driver.title == "娱乐明星_百度贴吧", "断言失败，当前页面title是%s " % self.driver.title
 
    def test_02_demo2(self):
        """判断首页标题"""
        # 这个例子会断言失败
        self.assertEqual(self.driver.title, "娱乐明星_百度贴吧")
 
    def tearDown(self):
        print("测试完成")
        self.driver.close()
 
    @classmethod
    def tearDownClass(cls):
        print("所有测试结束")
 
 
if __name__ == '__main__':
    unittest.main()

运行结果：（运行了两条用例，其中失败了一条；符合我们预期的结果）

所有测试开始
测试开始
测试完成
.测试开始
测试完成
F
======================================================================
FAIL: test_02_demo2 (__main__.PcLogin)
判断首页标题
所有测试结束
----------------------------------------------------------------------
Traceback (most recent call last):
  File "D:/1git/uitest/test_case/demo.py", line 30, in test_02_demo2
    self.assertEqual(self.driver.title, "娱乐明星_百度贴吧")
AssertionError: '百度贴吧——全球最大的中文社区' != '娱乐明星_百度贴吧'
- 百度贴吧——全球最大的中文社区
+ 娱乐明星_百度贴吧
----------------------------------------------------------------------
Ran 2 tests in 35.273s
 
FAILED (failures=1)
 
Process finished with exit code 1