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

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

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

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

    self.assertEqual('foo'.upper(), 'FOO')

def test_isupper(self):
    self.assertTrue('FOO'.isupper())
    self.assertFalse('Foo'.isupper())

def test_split(self):
    s = 'hello world'
    self.assertEqual(s.split(), ['hello', 'world'])
    # check that s.split fails when the separator is not a string
    with self.assertRaises(TypeError):
        s.split(2)

if name == ‘main’:
unittest.main()

在上述代码中，我们定义了一个 `TestStringMethods` 类，这个类继承自 `unittest.TestCase`。在这个类中，我们定义了三个方法：`test_upper`、`test_isupper` 和 `test_split`。这三个方法就是我们的三个测试用例。


`unittest.TestCase` 类提供了许多断言方法，例如 `assertEqual(a, b)`、`assertTrue(x)` 和 `assertFalse(x)`。这些断言方法用来检查我们的代码是否满足预期的行为。


### 


### 三、运行测试和查看测试结果


当我们定义好测试用例后，就可以运行这些测试用例，并查看测试结果了。你可以通过执行 `unittest.main()` 来运行所有的测试用例。


在上述代码的最后，我们调用了 `unittest.main()`。这个函数会搜索当前模块中所有 `unittest.TestCase` 的子类，然后运行这些子类中的所有以 `test` 开头的方法。


**当我们运行这段代码时，`unittest` 将会输出一个测试报告，显示每个测试用例的运行结果。例如，如果所有测试用例都通过了，你会看到如下的输出：**

…

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：模拟对象和行为



![img](https://img-blog.csdnimg.cn/img_convert/f12a14eaf09061b63cf02e77e05d9c1f.png)
![img](https://img-blog.csdnimg.cn/img_convert/d3af9aa5d3564c1c31a168d8d701a3bc.png)
![img](https://img-blog.csdnimg.cn/img_convert/fa1a224818098e51d5bfc8c9458862e8.png)

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

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

**[需要这份系统化的资料的朋友，可以戳这里获取](https://bbs.csdn.net/topics/618631832)**

上经验的小伙伴深入学习提升的进阶课程，涵盖了95%以上软件测试知识点，真正体系化！**

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

**[需要这份系统化的资料的朋友，可以戳这里获取](https://bbs.csdn.net/topics/618631832)**