python - Python 的魔术模拟没有按预期工作。实际函数被调用

Question

我有一个模型：

class Evaluation(models.Model):
   def abc(self):
      #some logic here
      return something

说，我有另一个使用它的功能：

def filter_graded(evaluations):
    return filter(lambda x: x.is_completely_graded(), evaluations)

现在，当我为filter_graded方法编写测试时，我模拟is_completely_graded某些实例返回真/假。测试不尊重模拟，而是调用实际is_completely_graded。我该如何解决这个问题？

我的测试看起来像：

def test_filter_graded(self):
    eval1 = G(Evaluation)
    eval2 = G(Evaluation)
    eval1.is_completely_graded = mock.MagicMock(return_value=True)
    eval2.is_completely_graded = mock.MagicMock(return_value=False)
    filter_graded(Evaluation.objects.all())
    self.assertEqual(len(result), 1)

Answer 1

您没有传入模拟对象。您正在传递真实数据：

filter_graded(Evaluation.objects.all())

你没有在任何地方嘲笑EvaluationorEvaluation.objects或Evaluation.objects.all()。

如果您的代码所做的只是使用is_completely_graded方法查找对象，只需将模拟作为序列传递，无需任何 ORM 查询：

def test_filter_graded(self):
    eval1 = mock.MagicMock()
    eval2 = mock.MagicMock
    eval1.is_completely_graded.return_value = True
    eval2.is_completely_graded.return_value = False
    filter_graded([eval1, eval2])
    self.assertEqual(len(result), 1)