Files
pecan/tests/test_generic.py
2011-03-06 11:28:15 -05:00

36 lines
966 B
Python

from pecan import Pecan, expose, request, response, redirect
from webtest import TestApp
try:
from simplejson import dumps
except:
from json import dumps
class TestGeneric(object):
def test_simple_generic(self):
class RootController(object):
@expose(generic=True)
def index(self):
pass
@index.when(method='POST', template='json')
def do_post(self):
return dict(result='POST')
@index.when(method='GET')
def do_get(self):
return 'GET'
app = TestApp(Pecan(RootController()))
r = app.get('/')
assert r.status_int == 200
assert r.body == 'GET'
r = app.post('/')
assert r.status_int == 200
assert r.body == dumps(dict(result='POST'))
r = app.get('/do_get', status=404)
assert r.status_int == 404