Fix tests

getsentry · Sep 3, 2024 · b7f8c96 · b7f8c96
1 parent b3ec817
commit b7f8c96
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/automation/summarize/test_issue.py b/tests/automation/summarize/test_issue.py
@@ -110,6 +110,13 @@ def test_summarize_issue_event_details(self, mock_from_event, mock_gpt_client, s
 class TestRunSummarizeIssue:
  @patch("seer.automation.summarize.issue.summarize_issue")
  def test_run_summarize_issue_langfuse_metadata(self, mock_summarize_issue):
+ mock_summarize_issue.return_value = SummarizeIssueResponse(
+ group_id=1,
+ headline="headline",
+ summary="summary",
+ impact="impact"
+ )
+
  # Create a sample request
  request = SummarizeIssueRequest(
  group_id=123,
@@ -132,6 +139,13 @@ def test_run_summarize_issue_langfuse_metadata(self, mock_summarize_issue):
 
  @patch("seer.automation.summarize.issue.summarize_issue")
  def test_run_summarize_issue_langfuse_metadata_no_org_slug(self, mock_summarize_issue):
+ mock_summarize_issue.return_value = SummarizeIssueResponse(
+ group_id=1,
+ headline="headline",
+ summary="summary",
+ impact="impact"
+ )
+
  # Create a sample request without organization_slug
  request = SummarizeIssueRequest(
  group_id=123, issue=next(generate(IssueDetails)), organization_id=456, project_id=789