python - 如何模拟补丁类的每个方法

Question

我有一个包含多达 50 种不同方法的测试类。我想用模拟函数修补每个方法。

prod = {"foo": "bar"}

def TestClass:
  @patch(db.get_product, return_value=prod)
  def test_1:
    pass
  @patch(db.get_product, return_value=prod)
  def test_2:
    pass
  .
  .
  .
  @patch(db.get_product, return_value=prod)
  def test_50:
    pass

有没有什么简单的方法可以做到这一点而不是重复 @patch(db.get_product, return=prod) 50次？

Answer 1

您可以使用 patch作为类装饰器:

@patch(db.get_product, return_value=prod)
class TestClass:
  def test_1:
    pass
  def test_2:
    pass
  .
  .
  .
  def test_50:
    pass

摘自 documentation :

Patch can be used as a TestCase class decorator. It works by decorating each test method in the class. This reduces the boilerplate code when your test methods share a common patchings set.