Django 单元测试中处理信号函数的策略：环境隔离法

# application/package/signals.py
import os
from django.db.models.signals import pre_save
from .models import MyEntity # 假设 MyEntity 在 .models 中定义

def do_stuff(sender, instance, **kwargs):
    """
    处理 MyEntity 实例的 pre_save 信号。
    此函数的核心逻辑仅在特定部署环境中执行。
    """
    # 检查环境变量，判断是否在需要执行核心逻辑的环境中
    # 例如，只有当 'RUN_SIGNAL_LOGIC' 环境变量被设置为 'true' 时才执行
    if os.environ.get('RUN_SIGNAL_LOGIC') == 'true':
        print(f"Executing do_stuff for {instance.pk} in deployed environment...")
        # ... 这里是原来包含外部调用或其他复杂逻辑的部分 ...
        # 例如：
        # external_api_call(instance.data)
        # update_related_records(instance)
    else:
        print(f"Skipping do_stuff for {instance.pk} in non-deployed/test environment.")
        # 在非部署环境（如本地开发或单元测试）中，这部分逻辑将被跳过

pre_save.connect(do_stuff, sender=MyEntity)

export RUN_SIGNAL_LOGIC='true'
# 然后启动您的Django应用
gunicorn myproject.wsgi

version: '3.8'
services:
  web:
    build: .
    environment:
      - RUN_SIGNAL_LOGIC=true
    # ... 其他配置

# 在运行测试前，确保没有设置 RUN_SIGNAL_LOGIC
unset RUN_SIGNAL_LOGIC 
python manage.py test

import os
from django.test import TestCase

class MyEntityTest(TestCase):
    def test_entity_creation_without_signal_logic(self):
        # 确保在测试执行期间，信号逻辑被跳过
        self.assertIsNone(os.environ.get('RUN_SIGNAL_LOGIC'))
        # ... 创建 MyEntity 实例 ...
        # 此时 do_stuff 的核心逻辑不会执行

    def test_entity_creation_with_signal_logic_mocked_dependency(self):
        # 如果需要测试信号逻辑本身，但要模拟其内部依赖
        # 可以临时设置环境变量，并mock do_stuff 内部的外部调用
        with self.settings(RUN_SIGNAL_LOGIC='true'): # 注意：settings 不直接修改 os.environ
            # 更好的做法是在测试 setup/teardown 中管理 os.environ
            original_env = os.environ.get('RUN_SIGNAL_LOGIC')
            os.environ['RUN_SIGNAL_LOGIC'] = 'true'
            try:
                # 此时 do_stuff 的核心逻辑会执行
                # 您可以在这里 mock do_stuff 内部的外部调用
                # @mock.patch("application.package.external_service.call_api")
                pass
            finally:
                if original_env is None:
                    del os.environ['RUN_SIGNAL_LOGIC']
                else:
                    os.environ['RUN_SIGNAL_LOGIC'] = original_env