Python3.x 如何模拟 Django 模型对象(及其方法)

发布于08月05日

我试图用下面的虚拟方法模拟一个链接的Django模型对象，

# utils.py
def foo_something_exist():
   if Foo.objects.get(a='something').exist():
      return 'exist'



# test.py
import unittest.mock import patch

import utils

.
.
.
@patch('utils.Foo')
def test_foo_something_exist_returns_none(self, mock_foo):
   mock_foo.objects.get.exists.return_value = False
   self.assertIsNone(utils.foo_something_exist()) 
.
.
.

test_foo_something_exist()次测试失败.我在utils中发现了Foo.objects.get(a='something').exist().py是MagicMock对象(<MagicMock name='Member.objects.get().exists()' id='xxxxxx'>)而不是False，这导致了该测试功能的失败.是的，我也试过mock_foo.objects.return_value.get.return_value.exists.return_value = False，这是在一篇帖子中提到的.正确模拟模型对象(及其链接方法)的指针/提示非常有用.

提前谢谢你.

推荐答案

不要修补模型本身，而是修补模型.对象，然后模拟链的每个部分:

@patch('utils.Foo.objects')
def test_foo_something_exist_returns_none(self, mock_foo):
    # just to show how to do it with longer chains
    # mock_foo.filter.return_value = mock_foo
    # mock_foo.exclude.return_value = mock_foo            
    mock_foo.get.return_value = mock_foo
    mock_foo.exists.return_value = False

    self.assertIsNone(utils.foo_something_exist())

这也适用于.filter()、.exclude()和其他模型操作.在测试Django模型时，我还发现Model Mommy个非常有用.