• RESTful doctesting using app.request
  • !/usr/bin/env python

    RESTful doctesting using app.request

    !/usr/bin/env python

    1. """
    2. RESTful web.py testing
    4. usage: python webapp.py 8080 [--test]
    6. >>> req = app.request('/mathematicians', method='POST')
    7. >>> req.status
    8. '400 Bad Request'
    10. >>> name = {'first': 'Beno\xc3\xaet', 'last': 'Mandelbrot'}
    11. >>> data = urllib.urlencode(name)
    12. >>> req = app.request('/mathematicians', method='POST', data=data)
    13. >>> req.status
    14. '201 Created'
    15. >>> created_path = req.headers['Location']
    16. >>> created_path
    17. '/mathematicians/b-mandelbrot'
    18. >>> fn = '<h1 class=fn>{0} {1}</h1>'.format(name['first'], name['last'])
    19. >>> assert fn in app.request(created_path).data
    20. """
    22. import doctest
    23. import urllib
    24. import sys
    26. import web
    29. paths = (
    30.     '/mathematicians(/)?', 'Mathematicians',
    31.     '/mathematicians/([a-z])-([a-z]{2,})', 'Mathematician'
    32. )
    34. app = web.application(paths, globals())
    36. dbname = {True: 'test', False: 'production'}[sys.argv[-1] == '--test']
    37. db = {} # db = web.database(..., db='math_{0}'.format(dbname))
    40. class Mathematicians:
    42.   def GET(self, slash=False):
    43.     """list all mathematicians and form to create new one"""
    44.     if slash:
    45.         raise web.seeother('/mathematicians')
    46.     mathematicians = db.items() # db.select(...)
    47.     return web.template.Template("""$def with (mathematicians)
    48.       <!doctype html>
    49.       <html>
    50.       <head>
    51.         <meta charset=utf-8>
    52.         <title>Mathematicians</title>
    53.       </head>
    54.       <body>
    55.         <h1>Mathematicians</h1>
    56.         $if mathematicians:
    57.           <ul class=blogroll>
    58.             $for path, name in mathematicians:
    59.               <li class=vcard><a class="fn url"
    60.               href=/mathematicians/$path>$name.first $name.last</a></li>
    61.           </ul>
    62.         <form action=/mathematicians method=post>
    63.           <label>First <input name=first type=text></label>
    64.           <label>Last <input name=last type=text></label>
    65.           <input type=submit value=Add>
    66.         </form>
    67.       </body>
    68.       </html>""")(mathematicians)
    70.   def POST(self, _):
    71.     """create new mathematician"""
    72.     name = web.input('first', 'last')
    73.     key = '{0}-{1}'.format(name.first[0].lower(), name.last.lower())
    74.     name.first, name.last = name.first.capitalize(), name.last.capitalize()
    75.     db[key] = name # db.insert(...)
    76.     path = '/mathematicians/{0}'.format(key)
    77.     web.ctx.status = '201 Created'
    78.     web.header('Location', path)
    79.     return web.template.Template("""$def with (path, name)
    80.       <!doctype html>
    81.       <html>
    82.       <head>
    83.         <meta charset=utf-8>
    84.         <title>Profile Created</title>
    85.       </head>
    86.       <body>
    87.         <p>Profile created for <a href=$path>$name.first $name.last</a>.</p>
    88.       </body>
    89.       </html>""")(path, name)
    92. class Mathematician:
    94.   def GET(self, first_initial, last_name):
    95.     """display mathematician"""
    96.     key = '{0}-{1}'.format(first_initial, last_name)
    97.     try:
    98.         mathematician = db[key] # db.select(...)
    99.     except KeyError:
    100.         raise web.notfound()
    101.     return web.template.Template("""$def with (name)
    102.       <!doctype html>
    103.       <html>
    104.       <head>
    105.         <meta charset=utf-8>
    106.         <title>$name.first $name.last</title>
    107.       </head>
    108.       <body class=vcard>
    109.         <p><a href=/mathematicians rel=up>Mathematicians</a> &#x25B8;</p>
    110.         <h1 class=fn>$name.first $name.last</h1>
    111.       </body>
    112.       </html>""")(mathematician)
    115. if __name__ == "__main__":
    116.   if sys.argv[-1] == '--test':
    117.     doctest.testmod()
    118.   else:
    119.     app.run()