getsentry · untitaker · Jul 15, 2019 · Jul 12, 2019 · Jul 12, 2019 · Jul 12, 2019
@@ -0,0 +1,117 @@
+"""
+An ASGI middleware.
+
+Based on Tom Christie's `sentry-asgi <https://github.com/encode/sentry-asgi>`_.
+"""
+
+import functools
+import urllib
+
+from sentry_sdk._types import MYPY
+from sentry_sdk.hub import Hub, _should_send_default_pii
+from sentry_sdk.integrations._wsgi_common import _filter_headers
+from sentry_sdk.utils import transaction_from_function
+
+if MYPY:
+    from typing import Dict
+
+
+class SentryAsgiMiddleware:
+    __slots__ = ("app",)
+
+    def __init__(self, app):
+        self.app = app
+
+    def __call__(self, scope, receive=None, send=None):
+        if receive is None or send is None:
+
+            async def run_asgi2(receive, send):
+                return await self._run_app(
+                    scope, lambda: self.app(scope)(receive, send)
+                )
+
+            return run_asgi2
+        else:
+            return self._run_app(scope, lambda: self.app(scope, receive, send))
+
+    async def _run_app(self, scope, callback):
+        hub = Hub.current
+        with Hub(hub) as hub:
+            with hub.configure_scope() as sentry_scope:
+                sentry_scope._name = "asgi"
+                sentry_scope.transaction = scope.get("path") or "unknown asgi request"
+
+                processor = functools.partial(self.event_processor, asgi_scope=scope)
+                sentry_scope.add_event_processor(processor)
+
+            try:
+                await callback()
+            except Exception as exc:
+                hub.capture_exception(exc)
+                raise exc from None
+
+    def event_processor(self, event, hint, asgi_scope):
+        request_info = event.setdefault("request", {})
+
+        if asgi_scope["type"] in ("http", "websocket"):
+            request_info["url"] = self.get_url(asgi_scope)
+            request_info["method"] = asgi_scope["method"]
+            request_info["headers"] = _filter_headers(self.get_headers(asgi_scope))
+            request_info["query_string"] = self.get_query(asgi_scope)
+
+        if asgi_scope.get("client") and _should_send_default_pii():
+            request_info["env"] = {"REMOTE_ADDR": asgi_scope["client"][0]}
+
+        if asgi_scope.get("endpoint"):
+            # Webframeworks like Starlette mutate the ASGI env once routing is
+            # done, which is sometime after the request has started. If we have
+            # an endpoint, overwrite our path-based transaction name.
+            event["transaction"] = self.get_transaction(asgi_scope)
+        return event
+
+    def get_url(self, scope):
+        """
+        Extract URL from the ASGI scope, without also including the querystring.
+        """
+        scheme = scope.get("scheme", "http")
+        server = scope.get("server", None)
+        path = scope.get("root_path", "") + scope["path"]
+
+        for key, value in scope["headers"]:
+            if key == b"host":
+                host_header = value.decode("latin-1")
+                return "%s://%s%s" % (scheme, host_header, path)
+
+        if server is not None:
+            host, port = server
+            default_port = {"http": 80, "https": 443, "ws": 80, "wss": 443}[scheme]
+            if port != default_port:
+                return "%s://%s:%s%s" % (scheme, host, port, path)
+            return "%s://%s%s" % (scheme, host, path)
+        return path
+
+    def get_query(self, scope):
+        """
+        Extract querystring from the ASGI scope, in the format that the Sentry protocol expects.
+        """
+        return urllib.parse.unquote(scope["query_string"].decode("latin-1"))
+
+    def get_headers(self, scope):
+        """
+        Extract headers from the ASGI scope, in the format that the Sentry protocol expects.
+        """
+        headers = {}  # type: Dict[str, str]
+        for raw_key, raw_value in scope["headers"]:
+            key = raw_key.decode("latin-1")
+            value = raw_value.decode("latin-1")
+            if key in headers:
+                headers[key] = headers[key] + ", " + value
+            else:
+                headers[key] = value
+        return headers
+
+    def get_transaction(self, scope):
+        """
+        Return a transaction string to identify the routed endpoint.
+        """
+        return transaction_from_function(scope["endpoint"])
@@ -97,6 +97,9 @@ def _capture_internal_warnings():
         if "SessionAuthenticationMiddleware" in str(warning.message):
             continue
 
+        if "Something has already installed a non-asyncio" in str(warning.message):
+            continue
+
         raise AssertionError(warning)
 
 

@@ -0,0 +1,3 @@
+import pytest
+
+pytest.importorskip("starlette")
@@ -0,0 +1,120 @@
+import sys
+
+import pytest
+from sentry_sdk import capture_message
+from sentry_sdk.integrations.asgi import SentryAsgiMiddleware
+from starlette.applications import Starlette
+from starlette.responses import PlainTextResponse
+from starlette.testclient import TestClient
+
+
+@pytest.fixture
+def app():
+    app = Starlette()
+
+    @app.route("/sync-message")
+    def hi(request):
+        capture_message("hi", level="error")
+        return PlainTextResponse("ok")
+
+    @app.route("/async-message")
+    async def hi2(request):
+        capture_message("hi", level="error")
+        return PlainTextResponse("ok")
+
+    app.add_middleware(SentryAsgiMiddleware)
+
+    return app
+
+
+@pytest.mark.skipif(sys.version_info < (3, 7), reason="requires python3.7 or higher")
+def test_sync_request_data(sentry_init, app, capture_events):
+    sentry_init(send_default_pii=True)
+    events = capture_events()
+
+    client = TestClient(app)
+    response = client.get("/sync-message?foo=bar")
+
+    assert response.status_code == 200
+
+    event, = events
+    assert event["transaction"] == "tests.integrations.asgi.test_asgi.app.<locals>.hi"
+    assert event["request"]["env"] == {"REMOTE_ADDR": "testclient"}
+    assert set(event["request"]["headers"]) == {
+        "accept",
+        "accept-encoding",
+        "connection",
+        "host",
+        "user-agent",
+    }
+    assert event["request"]["query_string"] == "foo=bar"
+    assert event["request"]["url"].endswith("/sync-message")
+    assert event["request"]["method"] == "GET"
+
+    # Assert that state is not leaked
+    events.clear()
+    capture_message("foo")
+    event, = events
+
+    assert "request" not in event
+    assert "transaction" not in event
+
+
+def test_async_request_data(sentry_init, app, capture_events):
+    sentry_init(send_default_pii=True)
+    events = capture_events()
+
+    client = TestClient(app)
+    response = client.get("/async-message?foo=bar")
+
+    assert response.status_code == 200
+
+    event, = events
+    assert event["transaction"] == "tests.integrations.asgi.test_asgi.app.<locals>.hi2"
+    assert event["request"]["env"] == {"REMOTE_ADDR": "testclient"}
+    assert set(event["request"]["headers"]) == {
+        "accept",
+        "accept-encoding",
+        "connection",
+        "host",
+        "user-agent",
+    }
+    assert event["request"]["query_string"] == "foo=bar"
+    assert event["request"]["url"].endswith("/async-message")
+    assert event["request"]["method"] == "GET"
+
+    # Assert that state is not leaked
+    events.clear()
+    capture_message("foo")
+    event, = events
+
+    assert "request" not in event
+    assert "transaction" not in event
+
+
+def test_errors(sentry_init, app, capture_events):
+    sentry_init(send_default_pii=True)
+    events = capture_events()
+
+    @app.route("/error")
+    def myerror(request):
+        raise ValueError("oh no")
+
+    client = TestClient(app, raise_server_exceptions=False)
+    response = client.get("/error")
+
+    assert response.status_code == 500
+
+    event, = events
+    assert (
+        event["transaction"]
+        == "tests.integrations.asgi.test_asgi.test_errors.<locals>.myerror"
+    )
+    exception, = event["exception"]["values"]
+
+    assert exception["type"] == "ValueError"
+    assert exception["value"] == "oh no"
+    assert any(
+        frame["filename"].endswith("tests/integrations/asgi/test_asgi.py")
+        for frame in exception["stacktrace"]["frames"]
+    )
@@ -0,0 +1,3 @@
+import pytest
+
+pytest.importorskip("channels")
@@ -0,0 +1,34 @@
+import pytest
+
+
+from channels.testing import HttpCommunicator
+
+from sentry_sdk.integrations.django import DjangoIntegration
+
+from tests.integrations.django.myapp.asgi import application
+
+
+@pytest.mark.asyncio
+async def test_basic(sentry_init, capture_events):
+    sentry_init(integrations=[DjangoIntegration()], send_default_pii=True)
+    events = capture_events()
+
+    comm = HttpCommunicator(application, "GET", "/view-exc?test=query")
+    response = await comm.get_response()
+    assert response["status"] == 500
+
+    event, = events
+
+    exception, = event["exception"]["values"]
+    assert exception["type"] == "ZeroDivisionError"
+
+    # Test that the ASGI middleware got set up correctly. Right now this needs
+    # to be installed manually (see myapp/asgi.py)
+    assert event["transaction"] == "/view-exc"
+    assert event["request"] == {
+        "cookies": {},
+        "headers": {},
+        "method": "GET",
+        "query_string": "test=query",
+        "url": "/view-exc",
+    }
@@ -13,7 +13,7 @@
 
 django.setup()
 
-from sentry_asgi import SentryMiddleware
+from sentry_sdk.integrations.asgi import SentryAsgiMiddleware
 
 application = get_default_application()
-application = SentryMiddleware(application)
+application = SentryAsgiMiddleware(application)
@@ -48,11 +48,15 @@ envlist =
 
     {py2.7,py3.7}-redis
 
+    py3.7-asgi
+
 [testenv]
 deps =
     -r test-requirements.txt
 
     django-{1.11,2.0,2.1,2.2}: djangorestframework>=3.0.0,<4.0.0
+    py3.7-django-{1.11,2.0,2.1,2.2}: channels>2
+    py3.7-django-{1.11,2.0,2.1,2.2}: pytest-asyncio
 
     django-{1.6,1.7,1.8}: pytest-django<3.0
     django-{1.9,1.10,1.11,2.0,2.1,2.2,dev}: pytest-django>=3.0
@@ -127,6 +131,9 @@ deps =
 
     redis: fakeredis
 
+    asgi: starlette
+    asgi: requests
+
     linters: black
     linters: flake8
     linters: flake8-import-order
@@ -150,6 +157,7 @@ setenv =
     aiohttp: TESTPATH=tests/integrations/aiohttp
     tornado: TESTPATH=tests/integrations/tornado
     redis: TESTPATH=tests/integrations/redis
+    asgi: TESTPATH=tests/integrations/asgi
 
     COVERAGE_FILE=.coverage-{envname}
 passenv =
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		import pytest

		pytest.importorskip("starlette")