22
33from django .urls import reverse
44
5+ from sentry .issues .grouptype import FeedbackGroup
56from sentry .testutils .cases import APITestCase
67
78
@@ -27,14 +28,20 @@ def test_get_errors(self):
2728 self .create_group (project = project1 , status = 1 , first_seen = prev , first_release = one , type = 2 )
2829 self .create_group (project = project2 , status = 1 , first_seen = curr , first_release = two , type = 3 )
2930 self .create_group (project = project2 , status = 2 , first_seen = curr , first_release = two , type = 4 )
30- self .create_group (project = project2 , status = 2 , first_seen = curr , first_release = two , type = 6 )
31+ self .create_group (
32+ project = project2 ,
33+ status = 2 ,
34+ first_seen = curr ,
35+ first_release = two ,
36+ type = FeedbackGroup .type_id ,
37+ )
3138
3239 # Time based issues.
3340 self .create_group (project = project1 , status = 0 , first_seen = curr , type = 1 )
3441 self .create_group (project = project1 , status = 1 , first_seen = curr , resolved_at = curr , type = 2 )
3542 self .create_group (project = project2 , status = 1 , first_seen = prev , resolved_at = prev , type = 3 )
3643 self .create_group (project = project2 , status = 2 , first_seen = prev , type = 4 )
37- self .create_group (project = project2 , status = 2 , first_seen = prev , type = 6 )
44+ self .create_group (project = project2 , status = 2 , first_seen = prev , type = FeedbackGroup . type_id )
3845
3946 response = self .client .get (
4047 self .url + f"?start={ prev .isoformat ()[:- 6 ]} &end={ curr .isoformat ()[:- 6 ]} &category=error"
@@ -146,13 +153,13 @@ def test_get_feedback(self):
146153 self .create_group (project = project1 , status = 0 , first_seen = curr , type = 1 )
147154 self .create_group (project = project1 , status = 1 , first_seen = curr , type = 2 )
148155 self .create_group (project = project2 , status = 1 , first_seen = curr , type = 3 )
149- self .create_group (project = project2 , status = 2 , first_seen = prev , type = 6 )
150- self .create_group (project = project2 , status = 2 , first_seen = curr , type = 6 )
156+ self .create_group (project = project2 , status = 2 , first_seen = prev , type = FeedbackGroup . type_id )
157+ self .create_group (project = project2 , status = 2 , first_seen = curr , type = FeedbackGroup . type_id )
151158 # Resolved cohort
152159 self .create_group (project = project1 , status = 0 , resolved_at = curr , type = 2 )
153160 self .create_group (project = project1 , status = 1 , resolved_at = curr , type = 3 )
154- self .create_group (project = project2 , status = 1 , resolved_at = prev , type = 6 )
155- self .create_group (project = project2 , status = 1 , resolved_at = curr , type = 6 )
161+ self .create_group (project = project2 , status = 1 , resolved_at = prev , type = FeedbackGroup . type_id )
162+ self .create_group (project = project2 , status = 1 , resolved_at = curr , type = FeedbackGroup . type_id )
156163 self .create_group (project = project2 , status = 2 , resolved_at = curr , type = 5 )
157164
158165 response = self .client .get (
0 commit comments