如何编写单元测试验证Flask POST接口向MongoDB写入数据的功能

你运行测试时卡住报错，是因为测试逻辑没有替换真实的MongoDB连接，业务代码会尝试连接环境变量中配置的真实MongoDB实例，大概率出现连接超时、无写入权限的问题，最终卡在insert操作上报错。

推荐使用mongomock库实现MongoDB的内存级Mock，无需修改原有业务代码，所有数据库操作都会在内存中执行，不会操作真实数据库，同时兼容MongoDB原生语法。

1. 安装依赖 #

执行以下命令安装需要的工具：

pip install mongomock freezegun

其中freezegun用于固定测试时间，方便验证日期格式是否符合预期，属于可选依赖，不需要可以不装。

2. 修改测试代码 #

修改后的测试代码如下：

import os
from unittest import TestCase
import mongomock
from freezegun import freeze_time
from app import app

class AppTest(TestCase):
    # 每个测试用例执行前运行
    def setUp(self):
        app.config['TESTING'] = True
        # 替换原有MongoDB连接为mock连接
        mock_client = mongomock.MongoClient()
        app.db = mock_client.securify
        self.app = app.test_client()

    def test_home(self):
        response = self.app.get('/')
        self.assertEqual(response.status_code, 200)
    
    # 用freeze_time固定测试时间，方便验证日期格式
    @freeze_time("2024-05-20 12:30:00")
    def test_post_data_insert_correctly(self):
        test_content = "this is a test"
        # 发送POST请求
        response = self.app.post('/', data={"content": test_content})
        self.assertEqual(response.status_code, 200)

        # 从mock数据库中读取插入的记录
        inserted_entry = app.db.entries.find_one()
        # 验证字段是否符合预期
        self.assertIsNotNone(inserted_entry)
        self.assertEqual(inserted_entry['content'], test_content)
        self.assertEqual(inserted_entry['date'], "2024-05-20/12:30")
        self.assertEqual(inserted_entry['IP'], "127.0.0.1")

• 测试启动时会直接将app.db替换为mongomock生成的虚拟数据库，所有insert、find操作都在内存中执行，不会发起真实网络请求，自然不会出现卡住超时的问题。
• POST请求执行完成后，直接调用find_one读取刚写入的记录，分别验证内容、日期格式、IP三个字段是否符合预期即可确认接口存储逻辑正常。

如果不想引入额外依赖，也可以用Python标准库的unittest.mock.patch直接Patchpymongo.MongoClient类，不过mongomock的语法和真实MongoDB完全一致，不需要手动写模拟返回值，开发效率更高。

内容的提问来源于stack exchange，提问作者John