add support for tornado

Author: rande

ioc/extra/tornado/handler.py

@@ -0,0 +1,71 @@
+from werkzeug.routing import NotFound
+
+import tornado.web
+import tornado.httpclient
+
+class BaseHandler(tornado.web.RequestHandler):
+    def dispatch(self):
+        pass
+
+    def get(self):
+        return self.dispatch()
+
+    def post(self):
+        return self.dispatch()
+
+    def put(self):
+        return self.dispatch()
+
+    def delete(self):
+        return self.dispatch()
+
+    def head(self):
+        return self.dispatch()
+
+    def options(self):
+        return self.dispatch()
+
+class RouterHandler(BaseHandler):
+    def initialize(self, router, event_dispatcher):
+        self.router = router
+        self.event_dispatcher = event_dispatcher
+
+    def dispatch(self):
+        try:
+            self.event_dispatcher.dispatch('handler.request', {'handler': self})
+
+            name, parameters, callback = self.router.match(path_info=self.request.path, method=self.request.method)
+
+            event = self.event_dispatcher.dispatch('handler.callback', {
+                'handler': self,
+                'name': name,
+                'callback': callback,
+                'parameters': parameters
+            })
+
+            event.get('callback')(self, **event.get('parameters'))
+
+        except NotFound:
+            self.set_status(404)
+            self.write("Not Found")
+
+            self.event_dispatcher.dispatch('handler.not_found', {
+                'handler': self,
+            })
+        except Exception:
+            self.set_status(500)
+            self.write("An unexpected error occurred")
+
+            self.event_dispatcher.dispatch('handler.exception', {
+                'handler': self,
+            })
+
+        self.event_dispatcher.dispatch('handler.response', {
+            'handler': self,
+        })
+
+        self.finish()
+
+        self.event_dispatcher.dispatch('handler.terminate', {
+            'handler': self,
+        })

ioc/extra/tornado/router.py

@@ -0,0 +1,35 @@
+from werkzeug.routing import Map, Rule
+
+class Router(object):
+    def __init__(self, url_map=None):
+        self._url_map = url_map or Map([])
+        self._view_functions = {}
+        self._adapter = None
+
+
+    def add(self, name, pattern, view_func, defaults=None, subdomain=None, methods=None,
+                 build_only=False, strict_slashes=None,
+                 redirect_to=None, alias=False, host=None):
+
+        self._url_map.add(Rule(pattern, endpoint=name, defaults=defaults, subdomain=subdomain, methods=methods,
+                              build_only=build_only, strict_slashes=strict_slashes, redirect_to=redirect_to,
+                              alias=alias, host=host))
+
+        self._view_functions[name] = view_func
+
+        self._adapter = None
+
+    def adapter(self):
+        if not self._adapter:
+            self._adapter = self._url_map.bind("localhost")
+
+        return self._adapter
+
+    def match(self, path_info=None, method=None, return_rule=False, query_args=None):
+        name, parameters = self.adapter().match(path_info, method, return_rule, query_args)
+
+        return name, parameters, self._view_functions[name]
+
+    def generate(self, name, values=None, method=None, force_external=False, append_unknown=True):
+
+        return self.adapter().build(name, values, method, force_external, append_unknown)

tests/ioc/extra/__init__.py

@@ -0,0 +1 @@
+__author__ = 'rande'

tests/ioc/extra/tornado/__init__.py

@@ -0,0 +1 @@
+__author__ = 'rande'

tests/ioc/extra/tornado/test_handler.py

@@ -0,0 +1,36 @@
+from tornado.testing import AsyncHTTPTestCase
+from tornado.web import Application
+
+from ioc.event import Dispatcher
+
+from ioc.extra.tornado.router import Router
+from ioc.extra.tornado.handler import RouterHandler
+
+def view(handler, name=None):
+    handler.write("Hello %s" % name)
+
+def error(handler):
+    raise Exception()
+
+class MyHTTPTest(AsyncHTTPTestCase):
+    def get_app(self):
+        router = Router()
+        router.add("hello", "/hello/<string:name>", view, methods=['GET'])
+        router.add("exception", "/exception", error, methods=['GET'])
+
+        return Application([("/.*", RouterHandler, dict(router=router, event_dispatcher=Dispatcher()))])
+
+    def test_not_found(self):
+        response = self.fetch('/')
+        self.assertEquals("Not Found", response.body)
+        self.assertEquals(404, response.code)
+
+    def test_found(self):
+        response = self.fetch('/hello/Thomas')
+        self.assertEquals("Hello Thomas", response.body)
+        self.assertEquals(200, response.code)
+
+    def test_error(self):
+        response = self.fetch('/exception')
+        self.assertEquals("An unexpected error occurred", response.body)
+        self.assertEquals(500, response.code)

tests/ioc/extra/tornado/test_router.py

@@ -0,0 +1,30 @@
+import unittest
+
+from ioc.extra.tornado.router import Router
+
+def view():
+    return "hello"
+
+class RouterTest(unittest.TestCase):
+    def setUp(self):
+
+        self.router = Router()
+
+    def test_add_and_match_routes(self):
+        self.router.add("homepage", "/", view)
+
+        self.assertEquals(('homepage', {}, view), self.router.match("/"))
+
+        self.router.add("blog_post", "/blog/<string:slug>", view, methods=['GET'])
+
+        self.assertEquals(('blog_post', {'slug': 'hello'}, view), self.router.match("/blog/hello"))
+
+    def test_add_and_generate_routes(self):
+
+        self.router.add("homepage", "/", view)
+        self.router.add("blog_post", "/blog/<string:slug>", view)
+
+        self.assertEquals("/", self.router.generate("homepage"))
+        self.assertEquals("/blog/hello", self.router.generate("blog_post", {'slug': 'hello'}))
+
+        self.assertEquals("http://localhost/blog/hello", self.router.generate("blog_post", {'slug': 'hello'}, force_external=True))