Improve SentryWorker middleware

1 files changed, 3 insertions, 4 deletions

diff --git a/raven/contrib/async.py b/raven/contrib/async.py
index 0b343ae..4f83c57 100644
--- a/raven/contrib/async.py
+++ b/raven/contrib/async.py
@@ -97,11 +97,10 @@ class SentryWorker(object):
     >>> from raven.base import Client
     >>> application = SentryWorker(application)
     """
-    def __init__(self, application):
+    def __init__(self, application, worker=None):
         self.application = application
-        self.worker = AsyncWorker()
+        self.worker = worker or AsyncWorker()
 
     def __call__(self, environ, start_response):
         environ['raven.worker'] = self.worker
-        for event in self.application(environ, start_response):
-            yield event
+        return iter(self.application(environ, start_response))