Django信号处理函数单元测试策略：mock.patch的局限与环境判断的应用

# myapp/signals.py
from django.db.models.signals import pre_save
from myapp.models import MyEntity

def do_stuff(sender, instance, **kwargs):
    print("Executing actual do_stuff - potentially making external calls")
    # ... 实际的外部调用逻辑 ...

pre_save.connect(do_stuff, sender=MyEntity)

# myapp/tests.py (尝试的模拟方式)
from django.test import TestCase
from unittest import mock
from myapp.models import MyEntity

class MyEntitySignalTest(TestCase):
    @mock.patch("myapp.signals.do_stuff", autospec=True)
    def test_entity_creation_mocks_do_stuff(self, mock_do_stuff):
        MyEntity.objects.create(name="Test Entity")
        mock_do_stuff.assert_called_once()
        # 实际情况可能是：do_stuff仍然被执行，mock_do_stuff从未被调用

# myapp/signals.py
import os
from django.db.models.signals import pre_save
from myapp.models import MyEntity # 假设MyEntity定义在此处或可导入

def do_stuff(sender, instance, **kwargs):
    # 检查环境变量，判断是否为测试或开发环境
    # 如果设置了'DJANGO_TESTING'为'True'，则跳过外部调用
    if os.environ.get('DJANGO_TESTING') == 'True':
        print("Skipping external calls in test environment.")
        # 可以执行一些仅在测试中需要的轻量级逻辑，或直接返回
        return

    # 只有在非测试环境下才执行实际的外部调用
    print("Executing actual external calls in non-test environment.")
    # ... 实际的外部调用逻辑，例如API请求、消息发布等 ...

pre_save.connect(do_stuff, sender=MyEntity)

# myapp/tests.py
import os
from django.test import TestCase
from myapp.models import MyEntity
# 确保在测试前，信号处理器已加载（通常Django启动时完成）

class MyEntitySignalEnvTest(TestCase):
    def setUp(self):
        # 记录原始环境变量状态，以便在tearDown中恢复
        self._original_django_testing_env = os.environ.get('DJANGO_TESTING')
        # 在测试开始时设置环境变量，指示当前为测试环境
        os.environ['DJANGO_TESTING'] = 'True'

    def tearDown(self):
        # 清理环境变量，恢复到测试前的状态，避免影响其他测试
        if self._original_django_testing_env is not None:
            os.environ['DJANGO_TESTING'] = self._original_django_testing_env
        else:
            if 'DJANGO_TESTING' in os.environ:
                del os.environ['DJANGO_TESTING']

    def test_entity_creation_skips_external_calls(self):
        # 创建一个MyEntity实例，这将触发pre_save信号
        entity = MyEntity.objects.create(name="Test Entity")

        # 在这个测试中，由于DJANGO_TESTING='True'，do_stuff中的外部调用逻辑会被跳过
        # 你可以断言其他与外部调用无关的逻辑是否正确执行
        self.assertIsNotNone(entity.pk)
        # 如果do_stuff有其他副作用，可以在这里检查