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
|
#!/usr/bin/env python
import pytest
from circuits.six import b, u
from circuits.web import Controller
from .helpers import urlencode, urlopen, HTTPError
class Root(Controller):
def index(self):
return "Hello World!"
def test_args(self, *args, **kwargs):
return "{0}\n{1}".format(repr(args), repr(kwargs))
def test_default_args(self, a=None, b=None):
return "a={0}\nb={1}".format(a, b)
def test_redirect(self):
return self.redirect("/")
def test_forbidden(self):
return self.forbidden()
def test_notfound(self):
return self.notfound()
def test_failure(self):
raise Exception()
def test_root(webapp):
f = urlopen(webapp.server.http.base)
s = f.read()
assert s == b"Hello World!"
def test_404(webapp):
try:
urlopen("%s/foo" % webapp.server.http.base)
except HTTPError as e:
assert e.code == 404
assert e.msg == "Not Found"
else:
assert False
def test_args(webapp):
args = ("1", "2", "3")
kwargs = {"1": "one", "2": "two", "3": "three"}
url = "%s/test_args/%s" % (webapp.server.http.base, "/".join(args))
data = urlencode(kwargs).encode('utf-8')
f = urlopen(url, data)
data = f.read().split(b"\n")
assert eval(data[0]) == args
assert eval(data[1]) == kwargs
@pytest.mark.parametrize("data,expected", [
((["1"], {}), b("a=1\nb=None")),
((["1", "2"], {}), b("a=1\nb=2")),
((["1"], {"b": "2"}), b("a=1\nb=2")),
])
def test_default_args(webapp, data, expected):
args, kwargs = data
url = u("{0:s}/test_default_args/{1:s}".format(
webapp.server.http.base,
u("/").join(args)
))
data = urlencode(kwargs).encode("utf-8")
f = urlopen(url, data)
assert f.read() == expected
def test_redirect(webapp):
f = urlopen("%s/test_redirect" % webapp.server.http.base)
s = f.read()
assert s == b"Hello World!"
def test_forbidden(webapp):
try:
urlopen("%s/test_forbidden" % webapp.server.http.base)
except HTTPError as e:
assert e.code == 403
assert e.msg == "Forbidden"
else:
assert False
def test_notfound(webapp):
try:
urlopen("%s/test_notfound" % webapp.server.http.base)
except HTTPError as e:
assert e.code == 404
assert e.msg == "Not Found"
else:
assert False
def test_failure(webapp):
try:
urlopen("%s/test_failure" % webapp.server.http.base)
except HTTPError as e:
assert e.code == 500
else:
assert False
|
