unittest 3 实例：测试弱密码，如何测试多组数据

Posted 2021-03-11 candyyang

目的

考虑这样一个测试弱密码的实例。

我们需要判断用户的密码中包含简单密码，规则是这样的，密码必须至少6位，满足6位的话判断用户的密码不是password123或者password之类的弱密码。

对于如下的测试数据，我们要如何使用unittest来进行相关测试呢？

[
  {"name":"jack","password":"Iloverose"},
  {"name":"rose","password":"Ilovejack"},
  {"name":"tom","password":"password123"}
]

Test fixture

前文说过，text fixture的主要功能是初始化测试数据或环境以及清理测试数据或环境。

考虑上面的例子，对我们而言，在用例执行之前初始化上面的测试数据是有必要的，我们可以把上面的数据用python的数据结构来表示，比较合适的数据结构是python的字典。这样做有如下的好处

• 统一初始化一些需要在多个用例之间共享的数据
• 可以在初始化的时候做一些数据的处理工作，比如过滤一些无效数据等

Test fixture最简单的实现方式是通过自定义下面的2个方法:

• TestCase.setUp方法在每个测试方法运行之前都会运行一次，适合为每个用例都初始化一遍数据
• TestCase.tearDown方法在每个测试方法运行之后都会运行一次，适合为每个用例都清理一遍数据

代码

新建名为test_password_1.py的文本文件，输入如下内容

import unittest

class PasswordTestCase(unittest.TestCase):

    def setUp(self):
        print(‘set up‘)
        self.test_data = [
            dict(name=‘jack‘, password=‘Iloverose‘),
            dict(name=‘rose‘, password=‘Ilovejack‘),
            dict(name=‘tom‘, password=‘password123‘)
        ]

    def test_weak_password(self):
        for data in self.test_data:
            passwd = data[‘password‘]

            self.assertTrue(len(passwd) >= 6)

            msg = "user %s has a weak password" %(data[‘name‘])
            self.assertTrue(passwd != ‘password‘, msg)
            self.assertTrue(passwd != ‘password123‘, msg)

    def test_dummy(self):
        pass

if __name__ == ‘__main__‘:
    unittest.main()

运行

在命令行里输入 python test_password_1.py来运行用例，结果如下

$ python test_password_1.py
set up
.set up
F
======================================================================
FAIL: test_week_password (__main__.PasswordTestCase)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "test_password_1.py", line 21, in test_week_password
    self.assertTrue(passwd != ‘password123‘, msg)
AssertionError: False is not true : user tom has a weak password

----------------------------------
Ran 2 tests in 0.001s

FAILED (failures=1)

解释一下

• setUp方法运行了2次，所以打印出了2次‘set up‘，这是因为上面的用例中有2个测试方法(2个方法名以test开头的方法)，setUp会在每个测试方法执行之前执行1次

• 由于用户tom的密码是弱密码password123，所以上面的用例运行失败了，打印出1个"F"，有几个F就代表有几个测试用例失败

• 测试方法运行失败时测试结果里会打印出失败方法的方法名，因此好的测试方法名可以方便我们快速找出失败用例

• 为了让错误信息更加容易理解，我们经常会自定义断言出错提示消息，比如msg = "user %s has a weak password" %(data[‘name‘])。一旦断言失败，我们一眼就能看出是哪个用户的密码强度不够

亲自动手试一试

假设我们增加1条测试数据，如下所示

[
  {"name":"jack","password":"Iloverose"},
  {"name":"rose","password":"Ilovejack"},
  {"name":"tom","password":"password123"},
  {"name":"jerry","password":"password"}
]

再运行上面的用例，观察一下测试结果是否会有不同？如果没有不同，那是为什么？

 

注解：这里使用在一个测试函数中利用循环实现提取数据来实现对多个数据的测试，虽然是基本可行的，但多个数据混合在一起，不利于对测试点的细分，同时也存在发生异常后导致后面的数据不能被测试的问题。

解决的方法：1.为每个测试数据写一个测试方法（笨方法）2.写一个通用测试方法，将测试数据作为参数传递给测试方法。（数据驱动方法）

 

 

参考

http://www.testclass.net/pyunit/test_example_1