7

我真的在寻找如何修补 What.objects.get_or_create，但我无法得到任何建议或想法如何去做。

好吧，我的问题是我有这样的事情：

def create_and_get_object(
    name, request, save_url=True, data=None):

    try:
        (object_type_, created) = Object.objects.get_or_create(
            name=name)
        try:
        ...
    except Exception:
        obj = None
        msg = ("Failed to get or create object `Object` with"
                "name='%s'" % name)
        logger_.exception(msg, extra={'request': request})
    return obj

所以，我想测试 Object.objects.get_or_create 在某个时候抛出异常......

我正在尝试，Mock(side_effect=Exception())但我真的不知道如何使用它来获得我所期望的。

所有这些都是为了获得该功能的 100% 覆盖（不包括异常行），以及许多其他类似的异常。

测试这个的正确方法是什么，以及这样的代码？

我究竟做错了什么？

4

1 回答 1

14

我喜欢将 python 的with语句与 mock 的patch.object 一起使用。在你的情况下，它应该是这样的：

from mock import patch
from django.test import TestCase
from django.db.models.manager import Manager
from anothermodule import create_and_get_object


class TestCreateAndGetObject(TestCase):
    def test_exception_throwing(self):
        with patch.object(Manager, 'get_or_create') as mock_method:
            mock_method.side_effect = Exception("test error")

            result = create_and_get_object('test', 'test')
            self.assertIsNone(result)

理论上应该可行，希望有所帮助。

于 2013-04-16T17:31:35.517 回答