Add support for Tornado 6

.gitignore

@@ -1,8 +1,14 @@
+.coverage
+.python-version
+.tox
 *.pyc
+.vscode
 dist
 bin
 eggs
 lib
 *.egg-info
 build
 env/
+venv/
+.pytest_cache/*

.travis.yml

@@ -1,11 +1,17 @@
 language: python
 python:
   - "2.7"
+  - "3.4"
+  - "3.5"
+  - "3.6"
+  - "3.7"
+  - "3.8"
 
 sudo: required
 
 install:
+  - pip install tox tox-travis
   - make bootstrap
 
 script:
-  -  make test lint
+  - make test

Makefile

@@ -38,9 +38,12 @@ clean-test:
 lint:
 	flake8 $(project) tests
 
-test:
+test-local:
 	py.test -s --cov-report term-missing:skip-covered --cov=$(project)
 
+test:
+	tox
+
 build: 
 	python setup.py build
 

README.rst

@@ -25,8 +25,8 @@ In order to implement tracing in your system (for all the requests), add the fol
 
 .. code-block:: python
 
-    from opentracing.scope_managers.tornado import TornadoScopeManager
     import tornado_opentracing
+    from tornado_opentracing.scope_managers import TornadoScopeManager
 
     # Create your opentracing tracer using TornadoScopeManager for active Span handling.
     tracer = SomeOpenTracingTracer(scope_manager=TornadoScopeManager())
@@ -48,7 +48,7 @@ In order to implement tracing in your system (for all the requests), add the fol
         ''' Other parameters here '''
         opentracing_tracer_callable='opentracing.mocktracer.MockTracer',
         opentracing_tracer_parameters={
-            'scope_manager': opentracing.scope_managers.TornadoScopeManager(),
+            'scope_manager': tornado_opentracing.scope_managers.TornadoScopeManager(),
         },
     )
 
@@ -115,11 +115,11 @@ For applications tracing individual requests, or using only the http client (no
 Active Span handling
 ====================
 
-For active ``Span`` handling and propagation, your ``Tracer`` should use ``opentracing.scope_managers.tornado.TornadoScopeManager``. Tracing both all requests and individual requests will set up a proper stack context automatically, and the active ``Span`` will be propagated from parent coroutines to their children. In any other case, code needs to be run under ``tracer_stack_context()`` explicitly:
+For active ``Span`` handling and propagation, your ``Tracer`` should use ``tornado_opentracing.scope_managers.TornadoScopeManager``. Tracing both all requests and individual requests will set up a proper stack context automatically, and the active ``Span`` will be propagated from parent coroutines to their children. In any other case, code needs to be run under ``tracer_stack_context()`` explicitly:
 
 .. code-block:: python
 
-    from opentracing.scope_managers.tornado import tracer_stack_context
+    from tornado_opentracing.scope_managers import tracer_stack_context
 
     with tracer_stack_context():
         ioloop.IOLoop.current().run_sync(main_func)

VERSION

@@ -1 +1,2 @@
-1.0.1
+1.0.1
+

examples/client-server/main.py

@@ -4,8 +4,8 @@
 from tornado import gen
 
 import opentracing
-from opentracing.scope_managers.tornado import TornadoScopeManager
 import tornado_opentracing
+from tornado_opentracing.scope_managers import TornadoScopeManager
 
 
 def client_start_span_cb(span, request):

examples/simple/main.py

@@ -3,8 +3,8 @@
 from tornado import gen
 
 import opentracing
-from opentracing.scope_managers.tornado import TornadoScopeManager
 import tornado_opentracing
+from tornado_opentracing.scope_managers import TornadoScopeManager 
 
 
 tornado_opentracing.init_tracing()

setup.py

@@ -15,15 +15,17 @@
     platforms='any',
     install_requires=[
         'tornado',
-        'opentracing>=2.0,<2.1',
+        'opentracing>=2.1,<2.4',
         'wrapt',
     ],
     extras_require={
         'tests': [
-            'flake8<3',  # see https://github.com/zheller/flake8-quotes/issues/29
+            'flake8<4',
             'flake8-quotes',
-            'pytest>=2.7,<3',
+            'pytest>=4.6.9',
             'pytest-cov',
+            'mock',
+            'tox',
         ],
     },
     classifiers=[

tests/helpers/__init__.py

@@ -0,0 +1,14 @@
+import sys
+
+import tornado_opentracing
+from opentracing.mocktracer import MockTracer
+from tornado_opentracing.scope_managers import TornadoScopeManager
+
+
+if sys.version_info >= (3, 3):
+    from ._test_case_gen import AsyncHTTPTestCase  # noqa
+else:
+    from ._test_case import AsyncHTTPTestCase  # noqa
+
+
+tracing = tornado_opentracing.TornadoTracing(MockTracer(TornadoScopeManager()))

tests/helpers/_test_case.py

@@ -0,0 +1,8 @@
+import tornado.testing
+
+
+class AsyncHTTPTestCase(tornado.testing.AsyncHTTPTestCase):
+
+    def http_fetch(self, url, *args, **kwargs):
+        self.http_client.fetch(url, self.stop, *args, **kwargs)
+        return self.wait()

tests/helpers/_test_case_gen.py

@@ -0,0 +1,31 @@
+import tornado.testing
+from tornado.httpclient import HTTPError
+from tornado import version_info as tornado_version
+
+from ._test_case import AsyncHTTPTestCase as BaseTestCase
+
+
+use_wait_stop = tornado_version < (5, 0, 0)
+
+if use_wait_stop:
+    def gen_test(func):
+        return func
+else:
+    gen_test = tornado.testing.gen_test
+
+
+class AsyncHTTPTestCase(BaseTestCase):
+
+    @gen_test
+    def _http_fetch_gen(self, url, *args, **kwargs):
+        try:
+            response = yield self.http_client.fetch(url, *args, **kwargs)
+        except HTTPError as exc:
+            response = exc.response
+        return response
+
+    def http_fetch(self, url, *args, **kwargs):
+        fetch = self._http_fetch_gen
+        if use_wait_stop:
+            fetch = super().http_fetch
+        return fetch(url, *args, **kwargs)

tests/helpers/handlers.py

@@ -0,0 +1,22 @@
+import sys
+
+import tornado.web
+
+
+class noopHandler(tornado.web.RequestHandler):
+    def get(self):
+        pass
+
+
+if sys.version_info > (3, 5):
+    from .handlers_async_await import (
+        AsyncScopeHandler,
+        DecoratedAsyncHandler,
+        DecoratedAsyncScopeHandler,
+        DecoratedAsyncErrorHandler
+    )
+else:
+    AsyncScopeHandler = noopHandler
+    DecoratedAsyncHandler = noopHandler
+    DecoratedAsyncScopeHandler = noopHandler
+    DecoratedAsyncErrorHandler = noopHandler

tests/helpers/handlers_async_await.py

@@ -0,0 +1,60 @@
+import asyncio
+
+import tornado.web
+
+from . import tracing
+
+
+class AsyncScopeHandler(tornado.web.RequestHandler):
+    async def do_something(self):
+        tracing = self.settings.get('opentracing_tracing')
+        with tracing.tracer.start_active_span('Child'):
+            tracing.tracer.active_span.set_tag('start', 0)
+            await asyncio.sleep(0)
+            tracing.tracer.active_span.set_tag('end', 1)
+
+    async def get(self):
+        tracing = self.settings.get('opentracing_tracing')
+        span = tracing.get_span(self.request)
+        assert span is not None
+        assert tracing.tracer.active_span is span
+
+        await self.do_something()
+
+        assert tracing.tracer.active_span is span
+        self.write('{}')
+
+
+class DecoratedAsyncHandler(tornado.web.RequestHandler):
+    @tracing.trace('protocol', 'doesntexist')
+    async def get(self):
+        await asyncio.sleep(0)
+        self.set_status(201)
+        self.write('{}')
+
+
+class DecoratedAsyncErrorHandler(tornado.web.RequestHandler):
+    @tracing.trace()
+    async def get(self):
+        await asyncio.sleep(0)
+        raise ValueError('invalid value')
+
+
+class DecoratedAsyncScopeHandler(tornado.web.RequestHandler):
+    async def do_something(self):
+        with tracing.tracer.start_active_span('Child'):
+            tracing.tracer.active_span.set_tag('start', 0)
+            await asyncio.sleep(0)
+            tracing.tracer.active_span.set_tag('end', 1)
+
+    @tracing.trace()
+    async def get(self):
+        span = tracing.get_span(self.request)
+        assert span is not None
+        assert tracing.tracer.active_span is span
+
+        await self.do_something()
+
+        assert tracing.tracer.active_span is span
+        self.set_status(201)
+        self.write('{}')

tests/helpers/markers.py

@@ -0,0 +1,28 @@
+import sys
+
+import pytest
+from tornado import version_info as tornado_version
+
+
+skip_generator_contextvars_on_tornado6 = pytest.mark.skipif(
+    tornado_version >= (6, 0, 0),
+    reason=(
+        'tornado6 has a bug (#2716) that '
+        'prevents contextvars from working.'
+    )
+)
+
+
+skip_generator_contextvars_on_py34 = pytest.mark.skipif(
+    sys.version_info.major == 3 and sys.version_info.minor == 4,
+    reason=('does not work on 3.4 with tornado context stack currently.')
+)
+
+
+skip_no_async_await = pytest.mark.skipif(
+    sys.version_info < (3, 5) or tornado_version < (5, 0),
+    reason=(
+        'async/await is not supported on python older than 3.5 '
+        'and tornado older than 5.0.'
+    )
+)