1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
# -*- coding: utf-8 -*-
"""
tests.exceptions
~~~~~~~~~~~~~~~~
The tests for the exception classes.
TODO:
- This is undertested. HTML is never checked
:copyright: 2007 Pallets
:license: BSD-3-Clause
"""
import pytest
from werkzeug import exceptions
from werkzeug._compat import text_type
from werkzeug.datastructures import WWWAuthenticate
from werkzeug.wrappers import Response
def test_proxy_exception():
orig_resp = Response("Hello World")
with pytest.raises(exceptions.HTTPException) as excinfo:
exceptions.abort(orig_resp)
resp = excinfo.value.get_response({})
assert resp is orig_resp
assert resp.get_data() == b"Hello World"
@pytest.mark.parametrize(
"test",
[
(exceptions.BadRequest, 400),
(exceptions.Unauthorized, 401, 'Basic "test realm"'),
(exceptions.Forbidden, 403),
(exceptions.NotFound, 404),
(exceptions.MethodNotAllowed, 405, ["GET", "HEAD"]),
(exceptions.NotAcceptable, 406),
(exceptions.RequestTimeout, 408),
(exceptions.Gone, 410),
(exceptions.LengthRequired, 411),
(exceptions.PreconditionFailed, 412),
(exceptions.RequestEntityTooLarge, 413),
(exceptions.RequestURITooLarge, 414),
(exceptions.UnsupportedMediaType, 415),
(exceptions.UnprocessableEntity, 422),
(exceptions.Locked, 423),
(exceptions.InternalServerError, 500),
(exceptions.NotImplemented, 501),
(exceptions.BadGateway, 502),
(exceptions.ServiceUnavailable, 503),
],
)
def test_aborter_general(test):
exc_type = test[0]
args = test[1:]
with pytest.raises(exc_type) as exc_info:
exceptions.abort(*args)
assert type(exc_info.value) is exc_type
def test_aborter_custom():
myabort = exceptions.Aborter({1: exceptions.NotFound})
pytest.raises(LookupError, myabort, 404)
pytest.raises(exceptions.NotFound, myabort, 1)
myabort = exceptions.Aborter(extra={1: exceptions.NotFound})
pytest.raises(exceptions.NotFound, myabort, 404)
pytest.raises(exceptions.NotFound, myabort, 1)
def test_exception_repr():
exc = exceptions.NotFound()
assert text_type(exc) == (
"404 Not Found: The requested URL was not found on the server."
" If you entered the URL manually please check your spelling"
" and try again."
)
assert repr(exc) == "<NotFound '404: Not Found'>"
exc = exceptions.NotFound("Not There")
assert text_type(exc) == "404 Not Found: Not There"
assert repr(exc) == "<NotFound '404: Not Found'>"
exc = exceptions.HTTPException("An error message")
assert text_type(exc) == "??? Unknown Error: An error message"
assert repr(exc) == "<HTTPException '???: Unknown Error'>"
def test_method_not_allowed_methods():
exc = exceptions.MethodNotAllowed(["GET", "HEAD", "POST"])
h = dict(exc.get_headers({}))
assert h["Allow"] == "GET, HEAD, POST"
assert "The method is not allowed" in exc.get_description()
def test_unauthorized_www_authenticate():
basic = WWWAuthenticate()
basic.set_basic("test")
digest = WWWAuthenticate()
digest.set_digest("test", "test")
exc = exceptions.Unauthorized(www_authenticate=basic)
h = dict(exc.get_headers({}))
assert h["WWW-Authenticate"] == str(basic)
exc = exceptions.Unauthorized(www_authenticate=[digest, basic])
h = dict(exc.get_headers({}))
assert h["WWW-Authenticate"] == ", ".join((str(digest), str(basic)))
|
