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

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()

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数软件测试工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年软件测试全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上软件测试开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录大纲截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且后续会持续更新

如果你觉得这些内容对你有帮助，可以添加V获取：vip1024b （备注软件测试）

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

b （备注软件测试）**
[外链图片转存中…(img-NubPZ1nE-1712983442143)]

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