Question

我想在我的测试中模拟一个模型的方法。这是我想要模拟的方法的模型

class Customer(models.Model):
    # Fields ...

    def save(self, *args, **kwargs):

        update_collector = self.id is None

        super(Customer, self).save(*args, **kwargs)

        if update_collector:
            self.send_to_address_book()

    def send_to_address_book(self):  # This is the method I want mocked
        email = self.user.email
        first_name = self.user.first_name
        last_name = self.user.last_name
        print("This is not being mocked")

        investigate_customer.apply_async(
            args=[first_name, last_name, email]
        )

然后，我希望从CustomerTestCase继承的所有测试都模拟send_to_address_book

class CustomerTestCase(object):
    @mock.patch('accounts.models.Customer.send_to_address_book')
    def create_user_and_customer(self, name, mock_method):
        mock_method.return_value = None
        if not name:
            name = 'test'
        email = name + '@test.si'
        user = User.objects.create_user(name, email)
        customer = Customer.objects.create(user=user)
        return user, customer

但是，当我运行以下测试时，send_to_address_book不会被模拟。

class CustomerQueriesTest(CustomerTestCase, TestCase):
    def setUp(self):
        G(Language, code='en', name='English')
        self.user, self.customer = self.create_user_and_customer()

    def test_queries_running(self):
        profile = self.user.profile
        resp = self.user.profile.queries_running()
        self.assertEqual(resp, 0)
        G(Package)  # using Django dynamic fixtures to create a new Package

我错过了什么？

Answer 1

所以，我发现了问题所在。

部分地，这是DDF的默认设置问题 - 如果模型字段为空/空，它仍将默认填充它。因此，当使用DDF创建Package时，由于FK，它还创建了Customer。

问题的第二部分是，Package与Customer位于不同的模块中，因此@mock.patch('accounts.models.Customer.send_to_address_book')什么都不做（有关详细信息，请参阅Python docs）。我必须添加第二个补丁，该补丁在Customer创建Package时负责：

@mock.patch('billing.models.Customer.send_to_address_book')

不能在Django测试中模拟模型的方法

1 个答案: