Skip to content

Commit acfb22f

Browse files
srest2021claude
andcommitted
test(autofix): Fix endpoint tests for removed code mapping fallback
Mock _resolve_project_preference to return a SeerProjectPreference with repos instead of None, since trigger_autofix no longer falls back to code mapping repos. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent a76ae51 commit acfb22f

File tree

1 file changed

+37
-24
lines changed

1 file changed

+37
-24
lines changed

tests/sentry/seer/endpoints/test_group_ai_autofix.py

Lines changed: 37 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
from sentry.seer.autofix.constants import AutofixReferrer, AutofixStatus
77
from sentry.seer.autofix.utils import AutofixState, AutofixStoppingPoint, CodebaseState
88
from sentry.seer.explorer.client_models import SeerRunState
9+
from sentry.seer.models import SeerProjectPreference
910
from sentry.testutils.cases import APITestCase, SnubaTestCase
1011
from sentry.testutils.helpers.datetime import before_now
1112
from 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

Comments
 (0)