88
99
1010@pytest .mark .django_db
11- class OrganizationCliBugPredictionTest (APITestCase ):
11+ class OrganizationCodeReviewLocalTest (APITestCase ):
1212 endpoint = "sentry-api-0-organization-code-review-local"
1313 method = "post"
1414
@@ -18,7 +18,7 @@ def setUp(self):
1818 self .project = self .create_project (organization = self .organization )
1919 self .repository = self .create_repo (
2020 project = self .project ,
21- name = "test-repo" ,
21+ name = "getsentry/ test-repo" ,
2222 provider = "github" ,
2323 external_id = "12345" ,
2424 )
@@ -36,8 +36,8 @@ def setUp(self):
3636 self .login_as (user = self .user )
3737
3838 @with_feature ("organizations:code-review-local" )
39- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.trigger_cli_bug_prediction " )
40- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.get_cli_bug_prediction_status " )
39+ @patch ("sentry.seer.code_review. endpoints.code_review_local.trigger_code_review_local " )
40+ @patch ("sentry.seer.code_review. endpoints.code_review_local.get_code_review_local_status " )
4141 def test_happy_path (self , mock_status , mock_trigger ):
4242 """Test successful prediction request"""
4343 mock_trigger .return_value = {"run_id" : 123 , "status" : "pending" }
@@ -176,7 +176,7 @@ def test_repository_not_found(self):
176176 assert "not found" in response .data ["detail" ]
177177
178178 @with_feature ("organizations:code-review-local" )
179- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.trigger_cli_bug_prediction " )
179+ @patch ("sentry.seer.code_review. endpoints.code_review_local.trigger_code_review_local " )
180180 def test_seer_trigger_timeout (self , mock_trigger ):
181181 """Test handling of Seer trigger timeout"""
182182 from urllib3 .exceptions import TimeoutError
@@ -192,7 +192,7 @@ def test_seer_trigger_timeout(self, mock_trigger):
192192 assert "unavailable" in response .data ["detail" ]
193193
194194 @with_feature ("organizations:code-review-local" )
195- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.trigger_cli_bug_prediction " )
195+ @patch ("sentry.seer.code_review. endpoints.code_review_local.trigger_code_review_local " )
196196 def test_seer_trigger_error (self , mock_trigger ):
197197 """Test handling of Seer trigger error"""
198198 mock_trigger .side_effect = ValueError ("Seer error" )
@@ -203,12 +203,12 @@ def test_seer_trigger_error(self, mock_trigger):
203203 status_code = 502 ,
204204 )
205205
206- assert "Failed " in response .data ["detail" ]
206+ assert "error " in response .data ["detail" ]. lower ()
207207
208208 @with_feature ("organizations:code-review-local" )
209- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.trigger_cli_bug_prediction " )
210- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.get_cli_bug_prediction_status " )
211- @patch ("sentry.api. endpoints.organization_cli_bug_prediction .time" )
209+ @patch ("sentry.seer.code_review. endpoints.code_review_local.trigger_code_review_local " )
210+ @patch ("sentry.seer.code_review. endpoints.code_review_local.get_code_review_local_status " )
211+ @patch ("sentry.seer.code_review. endpoints.code_review_local .time" )
212212 def test_seer_polling_timeout (self , mock_time_module , mock_status , mock_trigger ):
213213 """Test handling of polling timeout"""
214214 mock_trigger .return_value = {"run_id" : 123 , "status" : "pending" }
@@ -234,8 +234,8 @@ def fake_time():
234234 assert "exceeded maximum processing time" in response .data ["detail" ]
235235
236236 @with_feature ("organizations:code-review-local" )
237- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.trigger_cli_bug_prediction " )
238- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.get_cli_bug_prediction_status " )
237+ @patch ("sentry.seer.code_review. endpoints.code_review_local.trigger_code_review_local " )
238+ @patch ("sentry.seer.code_review. endpoints.code_review_local.get_code_review_local_status " )
239239 def test_seer_error_base_commit_not_found (self , mock_status , mock_trigger ):
240240 """Test mapping of base commit not found error"""
241241 mock_trigger .return_value = {"run_id" : 123 , "status" : "pending" }
@@ -253,8 +253,8 @@ def test_seer_error_base_commit_not_found(self, mock_status, mock_trigger):
253253 assert "pushed to the remote" in response .data ["detail" ]
254254
255255 @with_feature ("organizations:code-review-local" )
256- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.trigger_cli_bug_prediction " )
257- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.get_cli_bug_prediction_status " )
256+ @patch ("sentry.seer.code_review. endpoints.code_review_local.trigger_code_review_local " )
257+ @patch ("sentry.seer.code_review. endpoints.code_review_local.get_code_review_local_status " )
258258 def test_seer_error_diff_too_large (self , mock_status , mock_trigger ):
259259 """Test mapping of diff too large error"""
260260 mock_trigger .return_value = {"run_id" : 123 , "status" : "pending" }
@@ -272,8 +272,8 @@ def test_seer_error_diff_too_large(self, mock_status, mock_trigger):
272272 assert "500KB" in response .data ["detail" ]
273273
274274 @with_feature ("organizations:code-review-local" )
275- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.trigger_cli_bug_prediction " )
276- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.get_cli_bug_prediction_status " )
275+ @patch ("sentry.seer.code_review. endpoints.code_review_local.trigger_code_review_local " )
276+ @patch ("sentry.seer.code_review. endpoints.code_review_local.get_code_review_local_status " )
277277 def test_seer_error_too_many_files (self , mock_status , mock_trigger ):
278278 """Test mapping of too many files error"""
279279 mock_trigger .return_value = {"run_id" : 123 , "status" : "pending" }
@@ -291,8 +291,8 @@ def test_seer_error_too_many_files(self, mock_status, mock_trigger):
291291 assert "50 files" in response .data ["detail" ]
292292
293293 @with_feature ("organizations:code-review-local" )
294- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.trigger_cli_bug_prediction " )
295- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.get_cli_bug_prediction_status " )
294+ @patch ("sentry.seer.code_review. endpoints.code_review_local.trigger_code_review_local " )
295+ @patch ("sentry.seer.code_review. endpoints.code_review_local.get_code_review_local_status " )
296296 def test_seer_error_clone_failed (self , mock_status , mock_trigger ):
297297 """Test mapping of repository clone failed error"""
298298 mock_trigger .return_value = {"run_id" : 123 , "status" : "pending" }
@@ -310,8 +310,8 @@ def test_seer_error_clone_failed(self, mock_status, mock_trigger):
310310 assert "permissions" in response .data ["detail" ]
311311
312312 @with_feature ("organizations:code-review-local" )
313- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.trigger_cli_bug_prediction " )
314- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.get_cli_bug_prediction_status " )
313+ @patch ("sentry.seer.code_review. endpoints.code_review_local.trigger_code_review_local " )
314+ @patch ("sentry.seer.code_review. endpoints.code_review_local.get_code_review_local_status " )
315315 def test_seer_error_unknown (self , mock_status , mock_trigger ):
316316 """Test mapping of unknown Seer error"""
317317 mock_trigger .return_value = {"run_id" : 123 , "status" : "pending" }
@@ -358,8 +358,8 @@ def test_rate_limit_org(self, mock_is_limited):
358358 assert "Organization rate limit exceeded" in response .data ["detail" ]
359359
360360 @with_feature ("organizations:code-review-local" )
361- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.trigger_cli_bug_prediction " )
362- @patch ("sentry.api. endpoints.organization_cli_bug_prediction.get_cli_bug_prediction_status " )
361+ @patch ("sentry.seer.code_review. endpoints.code_review_local.trigger_code_review_local " )
362+ @patch ("sentry.seer.code_review. endpoints.code_review_local.get_code_review_local_status " )
363363 def test_optional_fields (self , mock_status , mock_trigger ):
364364 """Test that optional fields are not required"""
365365 mock_trigger .return_value = {"run_id" : 123 , "status" : "pending" }
0 commit comments