views.py

Question

有没有办法强制执行except块，以便我可以通过单元测试验证错误消息/处理？例如：

views.py

def get_books(request):
    ...
    try:
        book = Books.objects.get_or_create(user=request.user)
    except:
        messages.error(request, 'Unable to create new book!')
        return HttpResponseRedirect(reverse('my_books'))

    # more try/except cases

所以在这个例子中，我可以尝试模拟ORM失败，或者我可以尝试在这种情况下发送无效用户。但是是否存在抛出异常的一般方法（我愿意使用库）？例如Mock。或者是否有其他方法可以使用unittest强制例外？显然在整个代码中会有很多try / except块，所以我正在寻找任何帮助。

Answer 1

您可以使用unittest.mock来执行此操作。您应该Books.objects对patch.object对象进行修补，并使用side_effect来引发异常。对方法异常行为的完整测试应该是：

import unittest
from unittest.mock import patch, ANY
from django.test.client import RequestFactory

class TestDjango(unittest.TestCase):    
    @patch("your_module.messages")
    @patch("your_module.HttpResponseRedirect")
    def test_get_books_exception(self,mock_redirect,mock_messages)
        r = RequestFactory().get('/I_dont_konw_how_to_build_your_request/')
        objs = Books.objects
        with patch.object(objs,"get_or_create", side_effect=Exception()):
              self.assertIs(get_books(r),mock_redirect.return_value)
              mock_messages.error.assert_called_with(r, ANY)
              mock_redirect.assert_called_with(reverse('my_books'))

我用ANY删除字符串依赖项。我不确定你想在你的代码中测试什么（完整的行为或只是重定向......），无论如何我写了一个完整的例子。

如果您的项目不是传统工作，请考虑以TDD样式重写您的方法（以及新的django＆＃ORM调用），例如，您可以调用django ORM方法的_get_books_proxy()函数：

def _get_books_proxy(request):
    return Books.objects.get_or_create(user=request.user)

现在通过_get_books_proxy()装饰器直接修补patch：

@patch("your_module.messages")
@patch("your_module.HttpResponseRedirect")
@patch("your_module._get_books_proxy", side_effect=Exception())
def test_get_books_exception(self,mock_get_book, mock_redirect, mock_messages)
    r = RequestFactory().get('/I_dont_konw_how_to_build_your_request/')
    self.assertIs(get_books(r),mock_redirect.return_value)
    mock_messages.error.assert_called_with(r, ANY)
    mock_redirect.assert_called_with(reverse('my_books'))

此外，可以通过修补_get_books_proxy()并传递一个简单的Books来测试Mock()行为：

@patch("your_module.Books")
def test__get_books_proxy(self,mock_books):
     r = Mock()
     self.assertIs(mock_books.objects.get_or_create.return_value, your_module._get_books_proxy(r))
     mock_books.objects.get_or_create.assert_called_with(user=r.user)

如果您的关键点行为将被更改，现在您有一个可以屈服于您的哨兵。

如果您发现一些错误（我无法测试），我道歉......但我希望这个想法很明确。

Django / Python单元测试：如何强制尝试/接受块的异常

views.py

1 个答案: