Python中的单元测试框架：使用unittest进行有效测试_unittest利用数据库

最新推荐文章于 2024-10-18 00:00:00 发布

2301_82244006

最新推荐文章于 2024-10-18 00:00:00 发布

阅读量717

点赞数 8

分类专栏：程序员文章标签： python 学习面试

本文链接：https://blog.csdn.net/2301_82244006/article/details/138977918

版权

程序员专栏收录该内容

98 篇文章 0 订阅

订阅专栏

最后

不知道你们用的什么环境，我一般都是用的Python3.6环境和pycharm解释器，没有软件，或者没有资料，没人解答问题，都可以免费领取（包括今天的代码），过几天我还会做个视频教程出来，有需要也可以领取~

给大家准备的学习资料包括但不限于：

Python 环境、pycharm编辑器/永久激活/翻译插件

python 零基础视频教程

Python 界面开发实战教程

Python 爬虫实战教程

Python 数据分析实战教程

python 游戏开发实战教程

Python 电子书100本

Python 学习路线规划

网上学习资料一大堆，但如果学到的知识不成体系，遇到问题时只是浅尝辄止，不再深入研究，那么很难做到真正的技术提升。

需要这份系统化学习资料的朋友，可以戳这里获取

一个人可以走的很快，但一群人才能走的更远！不论你是正从事IT行业的老鸟或是对IT行业感兴趣的新人，都欢迎加入我们的的圈子（技术交流、学习资源、职场吐槽、大厂内推、面试辅导），让我们一起学习成长！

....
----------------------------------------------------------------------
Ran 4 tests in 0.001s

OK

四、使用测试加载器和测试运行器

测试加载器用于搜索和加载测试，而测试运行器则负责执行这些测试并报告结果。Python的unittest库为我们提供了默认的测试加载器和测试运行器，但是，你也可以自定义它们以满足特殊的需求。

下面是一个示例，演示了如何使用 unittest.TestLoader 来加载测试：

import unittest

class TestStringMethods(unittest.TestCase):

    # ... 前面的内容 ...

def suite():
    suite = unittest.TestSuite()
    loader = unittest.TestLoader()
    suite.addTest(loader.loadTestsFromTestCase(TestStringMethods))
    return suite

if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite())

在这个例子中，我们首先创建了一个 unittest.TestLoader 实例。然后，我们调用了 loadTestsFromTestCase 方法，这个方法会加载 TestStringMethods 类中定义的所有测试。然后，我们将这些测试添加到了测试套件中。

至于测试运行器，unittest 提供了 unittest.TextTestRunner 类作为默认的测试运行器。这个类的实例会在控制台上输出一个文本报告。如果你想自定义测试运行器，你可以通过继承 unittest.TestRunner 类来实现。

五、测试套件

测试套件是测试用例或测试套件的集合。它用于指定 unittest 所需执行的测试。通过创建自己的测试套件，你可以精确控制 unittest 执行哪些测试。以下是一个创建测试套件并添加测试用例的例子：

import unittest

class TestStringMethods(unittest.TestCase):

    # ... 与前文相同 ...

def suite():
    suite = unittest.TestSuite()
    suite.addTest(TestStringMethods('test_upper'))
    suite.addTest(TestStringMethods('test_isupper'))
    return suite

if __name__ == '__main__':
    runner = unittest.TextTestRunner()
    runner.run(suite())

在这个例子中，我们创建了一个 suite 函数，这个函数创建一个 unittest.TestSuite 实例，然后向这个实例添加测试用例。在 main 部分，我们创建了一个 unittest.TextTestRunner 实例，然后调用它的 run 方法来运行测试套件。

六、setUp 和 tearDown 方法

除了用于测试的方法外，unittest.TestCase 还提供了两个特殊的方法：setUp 和 tearDown。这两个方法在每个测试方法之前和之后运行，可以用于准备测试环境和清理资源。

下面是一个使用 setUp 和 tearDown 的例子：

import unittest

class TestDatabaseMethods(unittest.TestCase):

    def setUp(self):
        self.conn = create_database_connection()
        self.cur = self.conn.cursor()

    def tearDown(self):
        self.cur.close()
        self.conn.close()

    def test_insert(self):
        self.cur.execute("INSERT INTO employees VALUES (1, 'John')")
        self.cur.execute("SELECT * FROM employees")
        result = self.cur.fetchone()
        self.assertEqual(result, (1, 'John'))

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

在这个例子中，我们在 setUp 方法中创建了数据库连接和游标，在 tearDown 方法中关闭了它们。这样，我们就可以确保每个测试方法都在干净的数据库环境中运行。

七、unittest.mock：模拟对象和行为

在某些情况下，你可能想要替换测试中的一些对象，或者模拟特定的行为。unittest.mock 模块提供了一个 Mock 类和许多其他工具，可以帮助你实现这一点。

下面是一个使用 unittest.mock 的例子：

from unittest import TestCase, mock
from my_module import MyObject

class TestMyObject(TestCase):

    @mock.patch('my_module.MyObject')
    def test_my_method(self, MockMyObject):
        obj = MockMyObject()
        obj.my_method.return_value = 42

        assert obj.my_method() == 42
        obj.my_method.assert_called_once()

在这个例子中，我们使用了 unittest.mock.patch 装饰器来替换 MyObject 类。然后，我们可以控制这个替代对象的行为，例如设置它的方法返回什么值，或者检查它的方法是否被正确调用。

总的来说，Python的 unittest 框架为我们提供了强大而灵活的工具来进行单元测试。这只是 unittest 的冰山一角，它还有更多的功能等待你去发现和利用。

最后：可以在我的VX公众号：【自动化测试老司机】免费领取一份216页软件测试工程师面试宝典文档资料。以及相对应的视频学习教程免费分享！，其中包括了有基础知识、Linux必备、Shell、互联网程序原理、Mysql数据库、抓包工具专题、接口测试工具、测试进阶-Python编程、Web自动化测试、APP自动化测试、接口自动化测试、测试高级持续集成、测试架构开发测试框架、性能测试、安全测试等。

如果你也是看准了Python，想自学Python，在这里为大家准备了丰厚的免费学习大礼包，带大家一起学习，给大家剖析Python兼职、就业行情前景的这些事儿。