Python unittest框架实战：登录模块单元测试详解

桃夭

发布时间：2026-10-14 | 4人浏览过

原创

本文详解Python中unittest框架的核心概念，包括TestCase、TestSuite、TextTestRunner及Fixture。通过登录模块实例，演示如何编写测试用例，涵盖用户名密码正确、不存在、长度限制等场景，并解析setUp与tearDown的执行逻辑，帮助开发者掌握自动化测试基础。

登录模块代码示例

本文以登录模块为例进行讲解，相关代码如下：


def login_check(username,password):

  ''' 登录校验的函数
  :param username:账号
  :param password: 密码
  :return: '''

  if 6<=len(password)<=18:
    if username=='admin' and password=='123456':
      return {'code':0,'msg':'登录成功'}
    else:
      return {'code':1,'msg':'账号密码不正确'}

  else:
    return {'code':1,'msg':'密码长度在6-18之间'}

单元测试基础概念

首先明确单元测试的定义与实施方法：

单元测试（unit testing）是对软件中最小可测试单元进行检查和验证的过程。

实施单元测试的第一步是编写测试用例，其核心要素包括：编号、标题、前置条件、操作步骤、预期结果以及实际结果。

unittest框架核心组件

接下来介绍unittest框架的四个核心概念：

1. TestCase：每个TestCase的实例代表一个具体的测试用例。

2. TestSuite：用于将多个测试用例集合在一起执行。

3. TextTestRunner：负责执行测试用例并输出结果。

4. Fixture：用于处理测试环境的搭建与销毁，包括测试前的环境准备（setUp）、执行测试代码（run）以及测试后环境的还原（tearDown）。

登录测试用例设计

针对登录功能设计的测试用例标题如下：

1. 用户名和密码正确。

2. 用户名不存在。

3. 密码长度小于6位。

4. 密码长度大于18位。

5. 用户名为空。随后我们将执行这些测试用例。


import unittest      #引用单元测试模块
from day8.登录测试代码 import login_check  #将登陆模块代码引入
class Login_test(unittest.TestCase):
　　def setUp(self): 
  　　print("执行用例前都会执行这个方法，类似前置条件")
　　def tearDown(self):
  　　print("用例执行完后 都会执行该方法")
　　def test_login_success(self):  #登录成功用例
  　　result=login_check('admin','123456')  #这个是实际结果
  　　 expect='登录成功'          #这个是预期结果
　　　　 
  　　self.assertEqual(result['msg'],expect)  #将实际结果和预期结果做比较，输出完全相同则通过 输出结果是OK


　　def test_username_nonentity(self):   #用户名不存在用例
  　　result=login_check('admininin','111222333')
  　　expect='用户名不存在'
  　　self.assertEqual(expect,result['msg'])  #反之，如果不通过输出结果如下图