Mengolok-olok fungsi untuk memunculkan Exception guna menguji blok pengecualian

Saya memiliki fungsi ( foo) yang memanggil fungsi lain ( bar). Jika pemanggilan bar()memunculkan sebuah HttpError, saya ingin menanganinya secara khusus jika kode statusnya adalah 404, jika tidak, naikkan kembali.

Saya mencoba untuk menulis beberapa pengujian unit seputar foofungsi ini , mengejek panggilan ke bar(). Sayangnya, saya tidak bisa mendapatkan panggilan yang diejek bar()untuk mengajukan Pengecualian yang tertangkap oleh exceptblokir saya .

Inilah kode saya yang menggambarkan masalah saya:

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()

Aku mengikuti docs Mock yang mengatakan bahwa Anda harus mengatur side_effectdari Mockcontoh ke Exceptionkelas untuk memiliki fungsi mengejek kenaikan gaji kesalahan.

Saya juga melihat beberapa Tanya Jawab StackOverflow terkait lainnya, dan sepertinya saya melakukan hal yang sama yang mereka lakukan untuk menyebabkan dan Pengecualian dimunculkan oleh tiruan mereka.

• https://stackoverflow.com/a/10310532/346561
• Cara menggunakan Python Mock untuk memunculkan pengecualian - tetapi dengan Errno disetel ke nilai tertentu

Mengapa pengaturan side_effectdari barMocktidak menyebabkan diharapkan Exceptionakan dibangkitkan? Jika saya melakukan sesuatu yang aneh, bagaimana saya harus menguji logika di exceptblok saya ?

Tiruan Anda meningkatkan pengecualian dengan baik, tetapi error.resp.statusnilainya hilang. Daripada menggunakan return_value, cukup beri tahu Mockbahwa itu statusadalah atribut:

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

Argumen kata kunci tambahan untuk Mock()ditetapkan sebagai atribut pada objek yang dihasilkan.

Saya meletakkan definisi foodan Anda bardalam sebuah my_testsmodul, ditambahkan di HttpErrorkelas sehingga saya dapat menggunakannya juga, dan pengujian Anda kemudian dapat dijalankan hingga berhasil:

>>> 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

Anda bahkan dapat melihat print '404 - %s' % error.messagebaris tersebut dijalankan, tetapi saya pikir Anda ingin menggunakannya di error.contentsana; itulah HttpError()set atribut dari argumen kedua, bagaimanapun juga.