|
5 | 5 | from django.test import override_settings |
6 | 6 | from django.utils import timezone |
7 | 7 |
|
| 8 | +from sentry.analytics.events.eventuser_endpoint_request import EventUserEndpointRequest |
8 | 9 | from sentry.testutils.cases import APITestCase, PerformanceIssueTestCase, SnubaTestCase |
| 10 | +from sentry.testutils.helpers.analytics import assert_last_analytics_event |
9 | 11 | from sentry.testutils.helpers.datetime import before_now, freeze_time |
10 | 12 |
|
11 | 13 |
|
@@ -33,10 +35,12 @@ def test_simple(self, mock_record): |
33 | 35 |
|
34 | 36 | assert response.data[0]["value"] == "bar" |
35 | 37 |
|
36 | | - mock_record.assert_called_with( |
37 | | - "eventuser_endpoint.request", |
38 | | - project_id=project.id, |
39 | | - endpoint="sentry.api.endpoints.group_tagkey_values.get", |
| 38 | + assert_last_analytics_event( |
| 39 | + mock_record, |
| 40 | + EventUserEndpointRequest( |
| 41 | + project_id=project.id, |
| 42 | + endpoint="sentry.api.endpoints.group_tagkey_values.get", |
| 43 | + ), |
40 | 44 | ) |
41 | 45 |
|
42 | 46 | def test_simple_perf(self): |
@@ -208,8 +212,10 @@ def test_ratelimit(self, mock_record) -> None: |
208 | 212 | response = self.client.get(url) |
209 | 213 | assert response.status_code == 429 |
210 | 214 |
|
211 | | - mock_record.assert_called_with( |
212 | | - "eventuser_endpoint.request", |
213 | | - project_id=project.id, |
214 | | - endpoint="sentry.api.endpoints.group_tagkey_values.get", |
| 215 | + assert_last_analytics_event( |
| 216 | + mock_record, |
| 217 | + EventUserEndpointRequest( |
| 218 | + project_id=project.id, |
| 219 | + endpoint="sentry.api.endpoints.group_tagkey_values.get", |
| 220 | + ), |
215 | 221 | ) |
0 commit comments