{{cookiecutter.project_slug}}.tests.app.hooks.test_error

Complexity

Total Complexity

7

Size/Duplication

Total Lines	47
Duplicated Lines	0 %

Importance

Changes

0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	TestBroadExceptionHandler.test_500()	0	6	1
A	TestHTTPExceptionHandler.test()	0	11	4
A	TestError.setUp()	0	7	1
A	TestError.add_route_raises_exception()	0	6	1

from werkzeug.exceptions import HTTPException
from werkzeug.routing import RequestRedirect

from tests import BaseTestCase


class TestError(BaseTestCase):
    def setUp(self):
        super(TestError, self).setUp()

        self.app.debug = False
        # to prevent AssertionError: A setup function was called ...

        self.path = "/foo"

    def add_route_raises_exception(self, exception_cls):
        self.app.view_functions.pop("handler", None)

        @self.app.route(self.path)
        def handler():
            raise exception_cls()


class TestHTTPExceptionHandler(TestError):
    def test(self):
        for exception_cls in HTTPException.__subclasses__():
            if exception_cls is RequestRedirect or exception_cls().code == 412:
                continue

            self.add_route_raises_exception(exception_cls)

            resp = self.request()
            self.assertEqual(exception_cls().code, resp.status_code)
            self.assertTrue(resp.is_json)
            self.assertDictEqual({"message": exception_cls().description}, resp.json)


class TestBroadExceptionHandler(TestError):
    # TODO broad_exception_handler가 세분화될 때마다 테스트 케이스 추가

    def test_500(self):
        self.add_route_raises_exception(Exception)

        resp = self.request()
        self.assertEqual(500, resp.status_code)
        self.assertEqual("", resp.data.decode())