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()