62 lines
1.5 KiB
Python
62 lines
1.5 KiB
Python
from pecan.jsonify import jsonify, encode
|
|
from pecan import Pecan, expose
|
|
from webtest import TestApp
|
|
from json import loads
|
|
|
|
|
|
def make_person():
|
|
class Person(object):
|
|
def __init__(self, first_name, last_name):
|
|
self.first_name = first_name
|
|
self.last_name = last_name
|
|
|
|
@property
|
|
def name(self):
|
|
return '%s %s' % (self.first_name, self.last_name)
|
|
return Person
|
|
|
|
|
|
def test_simple_rule():
|
|
Person = make_person()
|
|
|
|
# create a Person instance
|
|
p = Person('Jonathan', 'LaCour')
|
|
|
|
# register a generic JSON rule
|
|
@jsonify.when_type(Person)
|
|
def jsonify_person(obj):
|
|
return dict(
|
|
name=obj.name
|
|
)
|
|
|
|
# encode the object using our new rule
|
|
result = loads(encode(p))
|
|
assert result['name'] == 'Jonathan LaCour'
|
|
assert len(result) == 1
|
|
|
|
|
|
class TestJsonify(object):
|
|
|
|
def test_simple_jsonify(self):
|
|
Person = make_person()
|
|
|
|
# register a generic JSON rule
|
|
@jsonify.when_type(Person)
|
|
def jsonify_person(obj):
|
|
return dict(
|
|
name=obj.name
|
|
)
|
|
|
|
class RootController(object):
|
|
@expose('json')
|
|
def index(self):
|
|
# create a Person instance
|
|
p = Person('Jonathan', 'LaCour')
|
|
return p
|
|
|
|
app = TestApp(Pecan(RootController()))
|
|
|
|
r = app.get('/')
|
|
assert r.status_int == 200
|
|
assert loads(r.body) == {'name':'Jonathan LaCour'}
|