|
1 |
| -from unittest.mock import MagicMock, patch |
| 1 | +from unittest.mock import Mock |
2 | 2 |
|
3 |
| -from django.http import HttpResponse |
| 3 | +from django.http import HttpResponse, HttpResponseBase |
4 | 4 | from django.test import RequestFactory
|
5 | 5 |
|
6 | 6 | from sentry.middleware.reporting_endpoint import ReportingEndpointMiddleware
|
7 | 7 | from sentry.testutils.cases import TestCase
|
8 |
| -from sentry.testutils.helpers.options import override_options |
9 | 8 |
|
10 | 9 |
|
11 | 10 | class ReportingEndpointMiddlewareTestCase(TestCase):
|
12 | 11 | def setUp(self) -> None:
|
13 | 12 | self.middleware = ReportingEndpointMiddleware(lambda request: HttpResponse())
|
14 | 13 | self.factory = RequestFactory()
|
15 | 14 |
|
16 |
| - def test_obeys_option(self) -> None: |
17 |
| - with override_options( |
18 |
| - {"issues.browser_reporting.reporting_endpoints_header_enabled": True} |
19 |
| - ): |
20 |
| - request = self.factory.get("/") |
21 |
| - response = self.middleware.process_response(request, HttpResponse()) |
22 |
| - |
23 |
| - assert ( |
24 |
| - response.get("Reporting-Endpoints") |
25 |
| - == "default=https://sentry.my.sentry.io/api/0/reporting-api-experiment/" |
26 |
| - ) |
27 |
| - |
28 |
| - with override_options( |
29 |
| - {"issues.browser_reporting.reporting_endpoints_header_enabled": False} |
30 |
| - ): |
31 |
| - request = self.factory.get("/") |
32 |
| - response = self.middleware.process_response(request, HttpResponse()) |
33 |
| - |
34 |
| - assert response.get("Reporting-Endpoints") is None |
35 |
| - |
36 |
| - @patch("src.sentry.middleware.reporting_endpoint.options.get") |
37 |
| - def test_no_options_check_in_relay_endpoints(self, mock_options_get: MagicMock) -> None: |
38 |
| - with override_options( |
39 |
| - {"issues.browser_reporting.reporting_endpoints_header_enabled": True} |
40 |
| - ): |
41 |
| - request = self.factory.get("/api/0/relays/register/challenge/") |
42 |
| - response = self.middleware.process_response(request, HttpResponse()) |
43 |
| - |
44 |
| - mock_options_get.assert_not_called() |
45 |
| - assert response.get("Reporting-Endpoints") is None |
| 15 | + def _no_header_set(self, result: HttpResponseBase) -> None: |
| 16 | + assert "Reporting-Endpoints" not in result |
| 17 | + |
| 18 | + def test_adds_header_for_staff_user(self) -> None: |
| 19 | + """Test that the ReportingEndpoint header is added when user is Sentry staff.""" |
| 20 | + request = self.factory.get("/") |
| 21 | + |
| 22 | + # Mock staff object with is_active = True |
| 23 | + staff_mock = Mock() |
| 24 | + staff_mock.is_active = True |
| 25 | + setattr(request, "staff", staff_mock) |
| 26 | + |
| 27 | + response = HttpResponse() |
| 28 | + result = self.middleware.process_response(request, response) |
| 29 | + |
| 30 | + assert "Reporting-Endpoints" in result |
| 31 | + assert ( |
| 32 | + result["Reporting-Endpoints"] |
| 33 | + == "default=https://sentry.my.sentry.io/api/0/reporting-api-experiment/" |
| 34 | + ) |
| 35 | + |
| 36 | + def test_no_header_for_non_staff_user(self) -> None: |
| 37 | + """Test that the ReportingEndpoint header is not added when user is not Sentry staff.""" |
| 38 | + request = self.factory.get("/") |
| 39 | + |
| 40 | + # Mock staff object with is_active = False |
| 41 | + staff_mock = Mock() |
| 42 | + staff_mock.is_active = False |
| 43 | + setattr(request, "staff", staff_mock) |
| 44 | + |
| 45 | + response = HttpResponse() |
| 46 | + result = self.middleware.process_response(request, response) |
| 47 | + |
| 48 | + self._no_header_set(result) |
| 49 | + |
| 50 | + def test_no_header_when_no_staff_attribute(self) -> None: |
| 51 | + """Test that the ReportingEndpoint header is not added when request has no staff attribute.""" |
| 52 | + request = self.factory.get("/") |
| 53 | + |
| 54 | + # No staff attribute on request |
| 55 | + response = HttpResponse() |
| 56 | + result = self.middleware.process_response(request, response) |
| 57 | + |
| 58 | + self._no_header_set(result) |
| 59 | + |
| 60 | + def test_no_header_when_staff_is_none(self) -> None: |
| 61 | + """Test that the ReportingEndpoint header is not added when staff is None.""" |
| 62 | + request = self.factory.get("/") |
| 63 | + setattr(request, "staff", None) |
| 64 | + |
| 65 | + response = HttpResponse() |
| 66 | + result = self.middleware.process_response(request, response) |
| 67 | + |
| 68 | + self._no_header_set(result) |
0 commit comments