fix(test): Update IP extraction for ASGI tests (#1200)

Author: kamilogorek

sentry_sdk/integrations/asgi.py

@@ -227,13 +227,17 @@ def _get_query(self, scope):
 
     def _get_ip(self, scope):
         # type: (Any) -> str
+        """
+        Extract IP Address from the ASGI scope based on request headers with fallback to scope client.
+        """
+        headers = self._get_headers(scope)
         try:
-            return scope["headers"]["x_forwarded_for"].split(",")[0].strip()
+            return headers["x-forwarded-for"].split(",")[0].strip()
         except (KeyError, IndexError):
             pass
 
         try:
-            return scope["headers"]["x_real_ip"]
+            return headers["x-real-ip"]
         except KeyError:
             pass
 

tests/integrations/asgi/test_asgi.py

@@ -258,7 +258,7 @@ def test_x_forwarded_for(sentry_init, app, capture_events):
     events = capture_events()
 
     client = TestClient(app)
-    response = client.get("/", headers={"X-Forwarded-For": "testproxy"})
+    response = client.get("/sync-message", headers={"X-Forwarded-For": "testproxy"})
 
     assert response.status_code == 200
 
@@ -272,7 +272,7 @@ def test_x_forwarded_for_multiple_entries(sentry_init, app, capture_events):
 
     client = TestClient(app)
     response = client.get(
-        "/", headers={"X-Forwarded-For": "testproxy1,testproxy2,testproxy3"}
+        "/sync-message", headers={"X-Forwarded-For": "testproxy1,testproxy2,testproxy3"}
     )
 
     assert response.status_code == 200
@@ -286,7 +286,7 @@ def test_x_real_ip(sentry_init, app, capture_events):
     events = capture_events()
 
     client = TestClient(app)
-    response = client.get("/", headers={"X-Real-IP": "1.2.3.4"})
+    response = client.get("/sync-message", headers={"X-Real-IP": "1.2.3.4"})
 
     assert response.status_code == 200