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
|
#!/usr/bin/env python
from circuits.web import Controller
from circuits.web.exceptions import Forbidden, NotFound, Redirect
from .helpers import urlopen, HTTPError
class Root(Controller):
def index(self):
return "Hello World!"
def test_redirect(self):
raise Redirect("/")
def test_forbidden(self):
raise Forbidden()
def test_notfound(self):
raise NotFound()
def test_contenttype(self):
self.response.headers["Content-Type"] = "application/json"
raise Exception()
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_contenttype(webapp):
try:
f = urlopen("%s/test_contenttype" % webapp.server.http.base)
except HTTPError as e:
assert e.code == 500
assert e.msg == "Internal Server Error"
assert e.headers.get("Content-Type") == "text/html"
else:
assert False
|