如何在测试Django普通视图和API视图时使用同一个模拟数据库

核心原因说明 #

你遇到的问题本质是测试上下文隔离失效：Django TestCase 运行时会创建临时测试数据库，所有测试过程中的数据库操作都包裹在事务中，测试结束后自动回滚。如果视图发起的是外部HTTP请求调用API，该请求会直接命中你本地/线上运行的服务，使用的是正式环境的数据库配置，自然读不到测试事务内的临时数据。

具体解决方法 #

• 方案1：替换HTTP调用为直接调用视图逻辑
  不通过HTTP请求调用API，直接导入API对应的视图函数/方法，构造符合要求的HttpRequest对象传入调用，整个过程都在测试事务内运行，会直接读取测试数据库数据。
• 方案2：Mock外部HTTP请求
  如果业务逻辑必须保留HTTP调用，测试时用unittest.mock补丁替换HTTP请求方法，直接返回预设的模拟结果，不需要真实发起请求。示例代码如下：

  from unittest.mock import patch, Mock
  from django.test import TestCase
  from yourapp.models import YourModel
  
  class TemplateViewTest(TestCase):
      def setUp(self):
          # 预创建测试数据库数据
          self.test_record = YourModel.objects.create(name="测试数据", status=1)
  
      @patch("yourapp.views.requests.get") # 替换你视图里实际调用的requests方法
      def test_view_render(self, mock_get):
          # 构造模拟API返回结果
          mock_response = Mock()
          mock_response.status_code = 200
          mock_response.json.return_value = {
              "id": self.test_record.id,
              "name": self.test_record.name
          }
          mock_get.return_value = mock_response
  
          # 发起视图测试请求
          resp = self.client.get("/your-view-path/")
          self.assertEqual(resp.status_code, 200)
          # 其他模板渲染、上下文断言逻辑
  

• 方案3：使用Django测试客户端发起内部请求
  如果你需要模拟API调用的完整流程，在视图内部用Django自带的测试Client发起请求，该客户端不会真实走网络请求，全程在测试上下文内运行，会读取测试数据库数据。

测试拆分和测试类选择建议 #

• 建议分开测试普通模板视图和API视图，做到职责分离：
  • API视图单独测试，覆盖参数校验、权限控制、数据返回格式、数据库操作逻辑，避免和模板视图的测试耦合
  • 模板视图测试时直接Mock依赖的API返回结果，不需要重复验证API本身的逻辑，只需要验证视图渲染、跳转、上下文处理的正确性即可
• 是否使用APITestCase取决于你的技术栈：如果你用Django REST Framework开发API，优先使用APITestCase，它自带JSON请求/响应处理、身份认证模拟等便捷方法，比普通TestCase写API测试效率更高；如果是测试普通模板视图，使用默认的TestCase就足够。

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