Django unittest 过滤功能

Question

我有以下问题：

class Foo(models.Model):
    bars = models.ManyToMany(Bar)

# in a function somewhere else
def bla(bars):
    return Foo.objects.filter(bars__in=bars)

我想测试当我调用bla() 时是否使用给定参数调用.filter()。

我这样写测试：

@patch('my_module.models.Foo')
def test_bla(self, FooMock):
    Foo.objects.filter = Mock()
    foo([1, 2])
    Foo.objects.filter.assert_called_with(bars__in=[1, 2])

它以Not called 失败。不确定我是否做得正确。有人可以帮我以正确的方式做吗？

谢谢！

Answer 1

我相信问题出在你正在修补的地方。我认为这可以很好地引导您解决问题：

http://alexmarandon.com/articles/python_mock_gotchas/