66from sentry .seer .autofix .constants import AutofixReferrer , AutofixStatus
77from sentry .seer .autofix .utils import AutofixState , AutofixStoppingPoint , CodebaseState
88from sentry .seer .explorer .client_models import SeerRunState
9+ from sentry .seer .models import SeerProjectPreference
910from sentry .testutils .cases import APITestCase , SnubaTestCase
1011from sentry .testutils .helpers .datetime import before_now
1112from sentry .testutils .helpers .features import with_feature
@@ -315,7 +316,7 @@ def __init__(self):
315316 @patch ("sentry.seer.autofix.autofix._call_autofix" )
316317 @patch ("sentry.seer.autofix.autofix._get_trace_tree_for_event" )
317318 @patch ("sentry.tasks.seer.autofix.check_autofix_status.apply_async" )
318- @patch ("sentry.seer.autofix.autofix._resolve_project_preference" , return_value = None )
319+ @patch ("sentry.seer.autofix.autofix._resolve_project_preference" )
319320 def test_ai_autofix_post_endpoint (
320321 self ,
321322 mock_resolve_pref ,
@@ -331,14 +332,18 @@ def test_ai_autofix_post_endpoint(
331332
332333 release = self .create_release (project = self .project , version = "1.0.0" )
333334
334- repo = self .create_repo (
335- project = self .project ,
336- name = "getsentry/sentry" ,
337- provider = "integrations:github" ,
338- external_id = "123" ,
339- integration_id = 234 ,
335+ mock_resolve_pref .return_value = SeerProjectPreference (
336+ organization_id = self .organization .id ,
337+ project_id = self .project .id ,
338+ repositories = [
339+ {
340+ "provider" : "integrations:github" ,
341+ "owner" : "getsentry" ,
342+ "name" : "sentry" ,
343+ "external_id" : "123" ,
344+ }
345+ ],
340346 )
341- self .create_code_mapping (project = self .project , repo = repo )
342347
343348 data = load_data ("python" , timestamp = before_now (minutes = 1 ))
344349 event = self .store_event (
@@ -475,7 +480,7 @@ def test_ai_autofix_post_without_code_mappings(
475480 @patch ("sentry.seer.autofix.autofix._call_autofix" )
476481 @patch ("sentry.seer.autofix.autofix._get_trace_tree_for_event" )
477482 @patch ("sentry.tasks.seer.autofix.check_autofix_status.apply_async" )
478- @patch ("sentry.seer.autofix.autofix._resolve_project_preference" , return_value = None )
483+ @patch ("sentry.seer.autofix.autofix._resolve_project_preference" )
479484 def test_ai_autofix_post_without_event_id (
480485 self ,
481486 mock_resolve_pref ,
@@ -491,14 +496,18 @@ def test_ai_autofix_post_without_event_id(
491496
492497 release = self .create_release (project = self .project , version = "1.0.0" )
493498
494- repo = self .create_repo (
495- project = self .project ,
496- name = "getsentry/sentry" ,
497- provider = "integrations:github" ,
498- external_id = "123" ,
499- integration_id = 234 ,
499+ mock_resolve_pref .return_value = SeerProjectPreference (
500+ organization_id = self .organization .id ,
501+ project_id = self .project .id ,
502+ repositories = [
503+ {
504+ "provider" : "integrations:github" ,
505+ "owner" : "getsentry" ,
506+ "name" : "sentry" ,
507+ "external_id" : "123" ,
508+ }
509+ ],
500510 )
501- self .create_code_mapping (project = self .project , repo = repo )
502511
503512 data = load_data ("python" , timestamp = before_now (minutes = 1 ))
504513 event = self .store_event (
@@ -562,7 +571,7 @@ def test_ai_autofix_post_without_event_id(
562571 @patch ("sentry.seer.autofix.autofix._call_autofix" )
563572 @patch ("sentry.seer.autofix.autofix._get_trace_tree_for_event" )
564573 @patch ("sentry.tasks.seer.autofix.check_autofix_status.apply_async" )
565- @patch ("sentry.seer.autofix.autofix._resolve_project_preference" , return_value = None )
574+ @patch ("sentry.seer.autofix.autofix._resolve_project_preference" )
566575 def test_ai_autofix_post_without_event_id_no_recommended_event (
567576 self ,
568577 mock_resolve_pref ,
@@ -578,14 +587,18 @@ def test_ai_autofix_post_without_event_id_no_recommended_event(
578587
579588 release = self .create_release (project = self .project , version = "1.0.0" )
580589
581- repo = self .create_repo (
582- project = self .project ,
583- name = "getsentry/sentry" ,
584- provider = "integrations:github" ,
585- external_id = "123" ,
586- integration_id = 234 ,
590+ mock_resolve_pref .return_value = SeerProjectPreference (
591+ organization_id = self .organization .id ,
592+ project_id = self .project .id ,
593+ repositories = [
594+ {
595+ "provider" : "integrations:github" ,
596+ "owner" : "getsentry" ,
597+ "name" : "sentry" ,
598+ "external_id" : "123" ,
599+ }
600+ ],
587601 )
588- self .create_code_mapping (project = self .project , repo = repo )
589602
590603 data = load_data ("python" , timestamp = before_now (minutes = 1 ))
591604 event = self .store_event (
0 commit comments