feat(issue-platform): Use assignee passed to the issue platform to assign to group

This hooks up the new assignee that we pass to the issue platform to actually perform the assignment

Author: wedamija

src/sentry/issues/ingest.py

@@ -24,6 +24,7 @@
 from sentry.eventstore.models import Event, GroupEvent, augment_message_with_occurrence
 from sentry.issues.grouptype import FeedbackGroup, should_create_group
 from sentry.issues.issue_occurrence import IssueOccurrence, IssueOccurrenceData
+from sentry.models.groupassignee import GroupAssignee
 from sentry.models.grouphash import GroupHash
 from sentry.models.release import Release
 from sentry.ratelimits.sliding_windows import RedisSlidingWindowRateLimiter, RequestedQuota
@@ -237,6 +238,15 @@ def save_issue_from_occurrence(
                         "sdk": normalized_sdk_tag_from_event(event),
                     },
                 )
+        if is_new and occurrence.assignee:
+            try:
+                # Since this calls hybrid cloud it has to be run outside the transaction
+                assignee = occurrence.assignee.resolve()
+            except Exception:
+                logger.exception("Failed to resolve assignee for occurrence")
+            else:
+                GroupAssignee.objects.assign(group, assignee, create_only=True)
+
     else:
         group = existing_grouphash.group
         if group.issue_category.value != occurrence.type.category:

tests/sentry/issues/test_ingest.py

@@ -26,6 +26,7 @@
 )
 from sentry.models.environment import Environment
 from sentry.models.group import Group
+from sentry.models.groupassignee import GroupAssignee
 from sentry.models.groupenvironment import GroupEnvironment
 from sentry.models.grouprelease import GroupRelease
 from sentry.models.release import Release
@@ -147,6 +148,23 @@ def test_new_group_with_priority(self) -> None:
         assert group_info is not None
         assert group_info.group.priority == PriorityLevel.HIGH
 
+    def test_new_group_with_user_assignee(self) -> None:
+        event = self.store_event(data={}, project_id=self.project.id)
+        occurrence = self.build_occurrence(event_id=event.event_id, assignee=f"user:{self.user.id}")
+        _, group_info = save_issue_occurrence(occurrence.to_dict(), event)
+        assert group_info is not None
+        assert group_info.group.priority == PriorityLevel.LOW
+        assignee = GroupAssignee.objects.get(group=group_info.group)
+        assert assignee.user_id == self.user.id
+
+    def test_new_group_with_team_assignee(self) -> None:
+        event = self.store_event(data={}, project_id=self.project.id)
+        occurrence = self.build_occurrence(event_id=event.event_id, assignee=f"team:{self.team.id}")
+        _, group_info = save_issue_occurrence(occurrence.to_dict(), event)
+        assert group_info is not None
+        assignee = GroupAssignee.objects.get(group=group_info.group)
+        assert assignee.team_id == self.team.id
+
 
 class ProcessOccurrenceDataTest(OccurrenceTestMixin, TestCase):
     def test(self) -> None:

tests/sentry/issues/test_occurrence_consumer.py

@@ -21,6 +21,7 @@
     _process_message,
 )
 from sentry.models.group import Group
+from sentry.models.groupassignee import GroupAssignee
 from sentry.receivers import create_default_projects
 from sentry.testutils.cases import SnubaTestCase, TestCase
 from sentry.testutils.helpers.datetime import before_now, iso_format
@@ -201,6 +202,40 @@ def test_issue_platform_override_priority(self, mock_get_severity_score):
         assert group.priority == PriorityLevel.HIGH
         assert "severity" not in group.data["metadata"]
 
+    def test_new_group_with_user_assignee(self) -> None:
+        message = get_test_message(self.project.id, assignee=f"user:{self.user.id}")
+        with self.feature("organizations:profile-file-io-main-thread-ingest"):
+            result = _process_message(message)
+        assert result is not None
+        occurrence = result[0]
+        assert occurrence is not None
+        group = Group.objects.filter(grouphash__hash=occurrence.fingerprint[0]).first()
+        assignee = GroupAssignee.objects.get(group=group)
+        assert assignee.user_id == self.user.id
+
+    def test_new_group_with_team_assignee(self) -> None:
+        message = get_test_message(self.project.id, assignee=f"team:{self.team.id}")
+        with self.feature("organizations:profile-file-io-main-thread-ingest"):
+            result = _process_message(message)
+        assert result is not None
+        occurrence = result[0]
+        assert occurrence is not None
+        group = Group.objects.filter(grouphash__hash=occurrence.fingerprint[0]).first()
+        assignee = GroupAssignee.objects.get(group=group)
+        assert assignee.team_id == self.team.id
+
+    def test_new_group_with_invalid_user_assignee(self) -> None:
+        other_user = self.create_user()
+        message = get_test_message(self.project.id, assignee=f"user:{other_user.id}")
+        with self.feature("organizations:profile-file-io-main-thread-ingest"):
+            result = _process_message(message)
+        assert result is not None
+        occurrence = result[0]
+        assert occurrence is not None
+        group = Group.objects.filter(grouphash__hash=occurrence.fingerprint[0]).first()
+        with pytest.raises(GroupAssignee.DoesNotExist):
+            GroupAssignee.objects.get(group=group)
+
 
 class IssueOccurrenceLookupEventIdTest(IssueOccurrenceTestBase):
     def test_lookup_event_doesnt_exist(self) -> None: