Python3.6/Django2.0下TestCase中单元测试相互影响的原因及规避方案

嘿，针对你在Django 2.0/Python 3.6项目里遇到的TestCase测试用例相互影响的问题，我来给你理清楚来龙去脉和解决方案～

Django的TestCase默认会给每个测试方法包裹一个数据库事务，测试结束后自动回滚，理论上测试用例是相互隔离的，但实际中还是可能出现影响，常见场景有这些：

• 数据库层面的意外污染：
  如果你的被测代码（比如Foo.method_x）里手动调用了transaction.commit()，或者用了原生SQL（比如直接用psycopg2执行语句），Django的自动回滚机制就会失效。比如test_a创建的Foo(a="bar")没有被回滚，那test_c再创建相同a值的对象时，如果a字段有唯一约束，就会抛出唯一性冲突的错误。

• 全局/静态变量的状态残留：
  要是Foo类里有静态变量，或者你的项目里有全局状态（比如全局配置、单例对象），某个测试修改了这些状态后没有重置，后面的测试就会沿用修改后的值。比如假设method_x依赖Foo.some_global_flag，test_a把这个flag改成了True，test_b运行时这个flag没重置，就可能导致断言不符合预期。

• 缓存、Session等共享存储的残留：
  如果你用了Redis这类外部缓存（而非Django默认的内存缓存），TestCase不会自动清理缓存数据。比如test_a往缓存里存了一个值，test_b读取时就会拿到这个旧值，导致逻辑出错。Session同理，如果测试里操作了session却没清理，也会影响后续测试。

• 第三方服务的状态残留：
  如果你的测试调用了外部API、消息队列等第三方服务，这些服务的数据不会被Django的事务回滚，前面的测试留下的数据会直接影响后面的测试。比如test_a往Redis里存了一条数据，test_b读取时就会拿到这条数据。

针对上面的场景，对应这些解决办法：

• 严格遵循Django的事务规则：
  不要在测试或被测代码里手动提交事务，如果必须使用事务逻辑，用Django的transaction.atomic()上下文管理器包裹，确保测试结束后能正常回滚。同时尽量避免使用原生SQL操作数据库，如果一定要用，记得在tearDown()里手动清理这些数据。

• 重置全局/静态变量：
  在测试类的setUp()方法里重置所有可能被修改的全局或静态变量到初始状态，或者在tearDown()里恢复。比如：

  class TestFoo(TestCase):
      def setUp(self):
          super().setUp()
          # 重置Foo类的静态变量到初始值
          Foo.some_static_var = "default_value"
  

• 清理共享存储：
  如果用了外部缓存，在tearDown()里手动清理缓存：

  from django.core.cache import cache
  
  def tearDown(self):
      cache.clear()
      super().tearDown()
  

  要是测试涉及Session，也可以在setUp()里重置Session对象。

• 隔离第三方服务：
  用unittest.mock来模拟第三方服务的调用，避免真实修改外部服务的状态。比如模拟API响应：

  from unittest.mock import patch
  
  def test_method_x_with_api_call(self):
      with patch('yourapp.utils.call_external_api') as mock_call:
          mock_call.return_value = {"status": "ok"}
          # 执行测试逻辑
          obj = Foo.objects.create(a="bar")
          self.assertEqual(obj.method_x(), {"a": "bar"})
  

  或者给测试环境单独配置一套第三方服务实例（比如测试专用的Redis、数据库），和生产环境彻底隔离。

• 保持测试用例的独立性：
  每个测试方法都应该自己创建所需的测试数据，不要依赖其他测试创建的数据——你现在的test_a、test_b、test_c就是这么做的，这是很好的习惯！确保每个测试都能独立运行，不依赖其他测试的执行顺序。

• 必要时使用TransactionTestCase：
  如果你的测试需要真正提交事务（比如测试Django信号、Celery异步任务），可以用django.test.TransactionTestCase替代TestCase。它会在每个测试后truncate所有数据库表，彻底清理数据，不过速度会比TestCase慢一些。

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