Как обрабатывать abort () во время модульного тестирования облачных функций Google с помощью Mock - PullRequest
Новая
       22

Как обрабатывать abort () во время модульного тестирования облачных функций Google с помощью Mock

0 голосов
/ 25 апреля 2019

Облачная функция Я тестирую вызовы abort(410) и хочу проверить, получен ли правильный код ошибки HTTP в моих модульных тестах.

При простом выполнении запроса я получаю следующую ошибку: 

/usr/lib/python3.7/site-packages/werkzeug/exceptions.py:707: in abort
    return _aborter(status, *args, **kwargs)
_ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _ _

self = <werkzeug.exceptions.Aborter object at 0x7fd474096b00>, code = 410, args = (), kwargs = {}

    def __call__(self, code, *args, **kwargs):
        if not args and not kwargs and not isinstance(code, integer_types):
            raise HTTPException(response=code)
        if code not in self.mapping:
            raise LookupError('no exception for %r' % code)
>       raise self.mapping[code](*args, **kwargs)
E       werkzeug.exceptions.Gone: 410 Gone: The requested URL is no longer available on this server and there is no forwarding address. If you followed a link from a foreign page, please contact the author of this page.

/usr/lib/python3.7/site-packages/werkzeug/exceptions.py:687: Gone

Вот мой код для создания пробного запроса: 

from unittest.mock import Mock
from cloud_functions import main

data = { ... }
headers = { ... }

req = Mock(get_json=Mock(return_value=data), args=data, headers=headers)
resp = main.my_function(req)

1 Ответ

0 голосов
/ 25 апреля 2019

Использование модуля HTTPException из werkzeug.exceptions для отлова кода ошибки отлично работает для меня.

Изменение моего кода на это заставляет его работать сейчас. 

from unittest.mock import Mock
from cloud_functions import main
from werkzeug.exceptions import HTTPException

data = { ... }
headers = { ... }

req = Mock(get_json=Mock(return_value=data), args=data, headers=headers)
try:
    resp = main.get_url_full(req)
    assert False
except HTTPException as e:
    assert e.code == 410
...