python如何使用unittest测试接口

测试是保障代码质量的重要手段之一，而 unittest 是 Python 中的一个用于编写单元测试的模块。下面将详细讲解如何使用 unittest 测试接口的完整攻略。

1. 创建测试用例

在使用 unittest 前，我们需要先创建一个测试用例。测试用例需要继承 unittest.TestCase 类，并通过方法重写的方式编写测试用例。下面是示例代码：

import unittest

class TestDemo(unittest.TestCase):

    def test_add(self):
        self.assertEqual(1+2, 3)

    def test_subtract(self):
        self.assertEqual(5-2, 3)

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

在上面的代码中，我们定义了一个 TestDemo 类，它继承了 unittest.TestCase 类。我们在 TestDemo 类中编写了两个测试用例：test_add() 和 test_subtract()，分别测试加法和减法的正确性。在测试用例中，我们使用了 self.assertEqual() 方法对结果进行断言，判断是否满足预期结果。

2. 运行测试用例

编写好测试用例后，我们需要运行测试用例。使用 unittest 提供的 main() 方法可以自动运行所有测试用例。下面是示例代码：

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

在上面的示例代码中，我们使用 if name == 'main' 来判断当前文件是否被其他文件调用，如果没有被调用，则执行 unittest.main() 方法，开始执行测试用例。

3. 使用 setUp() 和 tearDown() 方法

在编写测试用例时，我们可能需要先进行一些准备工作，或者在测试结束后需要进行清理工作。unittest 提供了 setUp() 和 tearDown() 方法来处理这些工作。setUp() 在每个测试用例执行前执行，tearDown() 在每个测试用例执行后执行。下面是示例代码：

class TestDemo(unittest.TestCase):

    def setUp(self):
        print("开始测试")

    def tearDown(self):
        print("结束测试")

    def test_add(self):
        self.assertEqual(1+2, 3)

    def test_subtract(self):
        self.assertEqual(5-2, 3)

在上面的示例代码中，我们重写了 setUp() 和 tearDown() 方法，并分别输出了开始测试和结束测试的提示信息。在每个测试用例执行前和执行后，都会执行这两个方法。

4. 使用参数化测试

在实际应用中，某些测试用例需要多次执行，只是测试数据不同。unittest 提供了 @parameterized.expand 装饰器来实现参数化测试。下面是示例代码：

import unittest
from parameterized import parameterized

class TestDemo(unittest.TestCase):

    @parameterized.expand([
        (1, 2, 3),
        (4, 5, 9),
        (7, 8, 15)
    ])
    def test_add(self, a, b, c):
        self.assertEqual(a+b, c)

在上面的示例代码中，我们使用 @parameterized.expand 装饰器来实现参数化测试。我们将测试数据定义在元组中，并传入测试用例中。在测试用例中，我们可以使用传入的参数进行测试。

5. 使用 mock 对象

在测试接口时，有时候需要测试一些依赖其他模块的逻辑，这就需要使用 mock 对象。mock 对象可以模拟其他模块的行为，从而让测试逻辑更加独立和稳定。下面是示例代码：

from unittest.mock import patch
import unittest

def is_admin(user):
    if user == 'admin':
        return True
    else:
        return False

class TestDemo(unittest.TestCase):

    def test_is_admin(self):
        with patch('__main__.is_admin', return_value=True):
            self.assertTrue(is_admin('admin'))

在上面的示例代码中，我们对 is_admin() 方法进行测试，但是它依赖了 main.is_admin 的返回结果。我们使用 patch() 来模拟 main.is_admin 的返回结果，以使接下来的测试更加独立和稳定。

总结

以上就是关于如何使用 unittest 测试接口的完整攻略，通过使用 setUp() 和 tearDown() 方法，参数化测试和 mock 对象等技巧，可以让测试逻辑更加独立和稳定。