Chế nhạo một hàm để nâng một Ngoại lệ để kiểm tra một khối ngoại trừ

Tôi có một hàm ( foo) gọi một hàm khác ( bar). Nếu việc gọi bar()tăng lên một HttpError, tôi muốn xử lý đặc biệt nếu mã trạng thái là 404, nếu không thì tăng lại.

Tôi đang cố gắng viết một số bài kiểm tra đơn vị xung quanh foochức năng này , chế nhạo cuộc gọi tới bar(). Thật không may, tôi không thể nhận được cuộc gọi bị chế nhạo bar()để nêu ra một Ngoại lệ bị chặn bởi exceptkhối của tôi .

Đây là mã của tôi minh họa vấn đề của tôi:

import unittest
import mock
from apiclient.errors import HttpError


class FooTests(unittest.TestCase):
    @mock.patch('my_tests.bar')
    def test_foo_shouldReturnResultOfBar_whenBarSucceeds(self, barMock):
        barMock.return_value = True
        result = foo()
        self.assertTrue(result)  # passes

    @mock.patch('my_tests.bar')
    def test_foo_shouldReturnNone_whenBarRaiseHttpError404(self, barMock):
        barMock.side_effect = HttpError(mock.Mock(return_value={'status': 404}), 'not found')
        result = foo()
        self.assertIsNone(result)  # fails, test raises HttpError

    @mock.patch('my_tests.bar')
    def test_foo_shouldRaiseHttpError_whenBarRaiseHttpErrorNot404(self, barMock):
        barMock.side_effect = HttpError(mock.Mock(return_value={'status': 500}), 'error')
        with self.assertRaises(HttpError):  # passes
            foo()

def foo():
    try:
        result = bar()
        return result
    except HttpError as error:
        if error.resp.status == 404:
            print '404 - %s' % error.message
            return None
        raise

def bar():
    raise NotImplementedError()

Tôi đã làm theo các tài liệu Mock nói rằng bạn nên đặt phiên bản side_effectcủa một lớp Mockthành một Exceptionlớp để hàm bị chế nhạo có thể phát sinh lỗi.

Tôi cũng đã xem xét một số Q & As có liên quan khác của StackOverflow, và có vẻ như tôi đang làm điều tương tự mà họ đang làm để gây ra và Ngoại lệ được nêu ra bởi chế độ của họ.

• https://stackoverflow.com/a/10310532/346561
• Cách sử dụng Python Mock để nâng cao một ngoại lệ - nhưng với Errno được đặt thành một giá trị nhất định

Tại sao việc đặt giá side_effecttrị barMockkhông làm tăng dự kiến Exception? Nếu tôi đang làm điều gì đó kỳ lạ, tôi nên kiểm tra logic trong exceptkhối của mình như thế nào?

Mô hình của bạn đang nâng cao ngoại lệ tốt, nhưng error.resp.statusgiá trị bị thiếu. Thay vì sử dụng return_value, chỉ cần nói Mockrằng đó statuslà một thuộc tính:

barMock.side_effect = HttpError(mock.Mock(status=404), 'not found')

Các đối số từ khóa bổ sung Mock()được đặt làm thuộc tính trên đối tượng kết quả.

Tôi đặt định nghĩa foovà của bạn bartrong một my_testsmô-đun, được thêm vào trong HttpErrorlớp để tôi cũng có thể sử dụng nó và sau đó bài kiểm tra của bạn có thể chạy thành công:

>>> from my_tests import foo, HttpError
>>> import mock
>>> with mock.patch('my_tests.bar') as barMock:
...     barMock.side_effect = HttpError(mock.Mock(status=404), 'not found')
...     result = my_test.foo()
... 
404 - 
>>> result is None
True

Bạn thậm chí có thể thấy print '404 - %s' % error.messagedòng chạy, nhưng tôi nghĩ bạn muốn sử dụng error.contentở đó thay thế; đó là các HttpError()tập thuộc tính từ đối số thứ hai, ở bất kỳ tỷ lệ nào.