如何区分正常调用与测试调用？视图及序列化器识别方法

以下是几种实用的实现方式：

1. 自定义请求头标记 #

在测试用例中给请求添加自定义HTTP头，视图或序列化器通过检查该头的存在来判断是否为测试请求。

测试用例示例：

from django.test import TestCase, Client

class MyViewTest(TestCase):
    def test_my_view(self):
        client = Client()
        # 添加自定义测试标记头
        response = client.get('/my-endpoint/', HTTP_X_TEST_REQUEST='true')
        # 后续断言逻辑

视图中的判断逻辑：

from django.http import HttpResponse

def my_view(request):
    is_test_request = request.META.get('HTTP_X_TEST_REQUEST') == 'true'
    if is_test_request:
        # 测试请求专属处理逻辑
        return HttpResponse('Test mode response')
    # 真实业务逻辑
    return HttpResponse('Production response')

序列化器中可通过上下文获取请求进行判断：

from rest_framework import serializers

class MySerializer(serializers.ModelSerializer):
    def create(self, validated_data):
        request = self.context.get('request')
        is_test_request = request.META.get('HTTP_X_TEST_REQUEST') == 'true'
        if is_test_request:
            # 测试环境下的创建逻辑
            pass
        # 真实业务的创建逻辑
        return super().create(validated_data)

2. 使用测试环境变量 #

在测试运行时设置专属环境变量，视图/序列化器读取该变量判断当前是否为测试场景。

测试用例示例：

import os
from django.test import TestCase

class MyViewTest(TestCase):
    def setUp(self):
        os.environ['TEST_MODE'] = '1'
    
    def tearDown(self):
        del os.environ['TEST_MODE']
    
    def test_my_view(self):
        client = Client()
        response = client.get('/my-endpoint/')

视图中的判断逻辑：

import os
from django.http import HttpResponse

def my_view(request):
    is_test_mode = os.environ.get('TEST_MODE') == '1'
    if is_test_mode:
        # 测试请求处理逻辑
        pass
    # 真实业务逻辑

3. 自定义测试客户端或请求对象 #

创建自定义测试客户端，给请求对象添加专属属性，视图直接检查该属性即可识别测试请求。

自定义测试客户端示例：

from django.test import Client

class TestClient(Client):
    def request(self, **request):
        # 给请求对象添加测试标记属性
        request['is_test'] = True
        return super().request(**request)

测试用例中使用自定义客户端：

class MyViewTest(TestCase):
    def test_my_view(self):
        client = TestClient()
        response = client.get('/my-endpoint/')

视图中的判断逻辑：

def my_view(request):
    is_test_request = getattr(request, 'is_test', False)
    if is_test_request:
        # 测试请求专属逻辑
        pass

注意事项 #

• 自定义请求头建议使用X-前缀，避免与标准HTTP头冲突。
• 使用环境变量时，务必在测试结束后清理变量，防止影响其他测试用例。
• 测试标记仅用于测试专属逻辑（比如跳过第三方接口调用），不要依赖其修改核心业务逻辑。

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