summaryrefslogtreecommitdiff
path: root/service
diff options
context:
space:
mode:
authorDuda Dornelles <ddornell@thoughtworks.com>2014-12-18 10:04:42 -0200
committerDuda Dornelles <ddornell@thoughtworks.com>2014-12-18 10:04:42 -0200
commit0a6f346fc2c1e3294d6216e3b1f662124ecf7de0 (patch)
tree277acb99df4e43e5b6b8cbb9e328c5ba9f34c5e6 /service
parentee1b4ebbed56cf8775e62ba9a3caf9c672b01e13 (diff)
Adding a tornado proxy to simulate the behavior for the user agent when running with the dispatcher
Diffstat (limited to 'service')
-rw-r--r--service/test/support/dispatcher/__init__.py122
-rw-r--r--service/test_requirements.txt2
2 files changed, 124 insertions, 0 deletions
diff --git a/service/test/support/dispatcher/__init__.py b/service/test/support/dispatcher/__init__.py
new file mode 100644
index 00000000..f1045042
--- /dev/null
+++ b/service/test/support/dispatcher/__init__.py
@@ -0,0 +1,122 @@
+#
+# Copyright (c) 2014 ThoughtWorks Deutschland GmbH
+#
+# Pixelated is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# Pixelated is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+# GNU Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with Pixelated. If not, see <http://www.gnu.org/licenses/>.
+from __future__ import print_function
+import traceback
+import sys
+
+from tornado.httpclient import AsyncHTTPClient
+from tornado.httpserver import HTTPServer
+import tornado.ioloop
+import tornado.web
+import tornado.escape
+from tornado import gen
+
+
+class MainHandler(tornado.web.RequestHandler):
+
+ @tornado.web.asynchronous
+ @gen.engine
+ def get(self):
+ self.forward('3333', '127.0.0.1')
+
+ @tornado.web.asynchronous
+ @gen.engine
+ def post(self):
+ self.get()
+
+ @tornado.web.asynchronous
+ @gen.engine
+ def put(self):
+ self.get()
+
+ @tornado.web.asynchronous
+ @gen.engine
+ def delete(self):
+ self.get()
+
+ def handle_response(self, response):
+ if response.error and not isinstance(response.error, tornado.httpclient.HTTPError):
+ print(self.request.uri)
+ print(response.error)
+ self.set_status(500)
+ self.write("Internal server error:\n" + str(response.error))
+ self.finish()
+ else:
+ self.set_status(response.code)
+ for header in ("Date", "Cache-Control", "Server", "Content-Type", "Location"):
+ v = response.headers.get(header)
+ if v:
+ self.set_header(header, v)
+ if response.body:
+ self.write(response.body)
+ self.finish()
+
+ def forward(self, port=None, host=None):
+ url = "%s://%s:%s%s" % (
+ 'http', host or "127.0.0.1", port or 80, self.request.uri)
+ try:
+ tornado.httpclient.AsyncHTTPClient().fetch(
+ tornado.httpclient.HTTPRequest(
+ url=url,
+ method=self.request.method,
+ body=None if not self.request.body else self.request.body,
+ headers=self.request.headers,
+ follow_redirects=False,
+ request_timeout=10),
+ self.handle_response)
+ except tornado.httpclient.HTTPError, x:
+ if hasattr(x, 'response') and x.response:
+ self.handle_response(x.response)
+ except Exception, e:
+ self.set_status(500)
+ self.write("Internal server error:\n" + ''.join(traceback.format_exception(*sys.exc_info())))
+ self.finish()
+
+
+def _create_app():
+ app = tornado.web.Application(
+ [
+ (r"/.*", MainHandler)
+ ],
+ xsrf_cookies=False,
+ debug=True)
+ return app
+
+
+class Proxy:
+
+ def __init__(self):
+ pass
+
+ def serve_forever(self):
+ app = _create_app()
+ self._server = HTTPServer(app)
+ self._server.listen(port='8888', address='127.0.0.1')
+ self._ioloop = tornado.ioloop.IOLoop.instance()
+ self._ioloop.start() # this is a blocking call, server has stopped on next line
+ self._ioloop = None
+
+ def shutdown(self):
+ if self._ioloop:
+ self._server.stop()
+ self._ioloop.stop()
+
+
+proxy = Proxy()
+try:
+ proxy.serve_forever()
+except Exception:
+ proxy.shutdown()
diff --git a/service/test_requirements.txt b/service/test_requirements.txt
index dad39d5d..aeba39d0 100644
--- a/service/test_requirements.txt
+++ b/service/test_requirements.txt
@@ -6,3 +6,5 @@ mock==1.0.1
httmock==1.2.2
mockito==0.5.2
funkload==1.16.1
+tornado==4.0.2
+