Added py3.7 functionality

Osmar Coronel · Osmar Coronel · commit a7813a4a3b1f · 2019-08-01T14:57:56.000-07:00
diff --git a/sentry_sdk/integrations/beam.py b/sentry_sdk/integrations/beam.py
@@ -108,7 +108,7 @@ def _inner(*args, **kwargs):
 
 def _capture_exception(exc_info, hub):
     """
-    Send Beam exception to Sentry
+    Send Beam exception to Sentry.
     """
     integration = hub.get_integration(BeamIntegration)
     if integration:
@@ -142,6 +142,6 @@ def _wrap_generator_call(gen, client):
         try:
             yield next(gen)
         except StopIteration:
-            raise
+            break
         except Exception:
             raiseException(client)
diff --git a/tests/integrations/beam/test_beam.py b/tests/integrations/beam/test_beam.py
@@ -153,8 +153,11 @@ def test_monkey_patch_signature(f, args, kwargs):
 class _OutputProcessor(OutputProcessor):
     def process_outputs(self, windowed_input_element, results):
         print(windowed_input_element)
-        for result in results:
-            assert result
+        try:
+            for result in results:
+                assert result
+        except StopIteration:
+            print("In here")
 
 
 @pytest.fixture
diff --git a/tox.ini b/tox.ini
@@ -33,7 +33,7 @@ envlist =
     {pypy,py2.7}-celery-3
 
     py2.7-beam-{12,13,master}
-    {py3.6,py3.7,py3.8}-beam-{12,13,master,dev}
+    {py3.6,py3.7}-beam-{12,13,master}
 
     # The aws_lambda tests deploy to the real AWS and have their own matrix of Python versions.
     py3.7-aws_lambda
