From 4c385574856ce96bd32f5e254837afe40ccbfe99 Mon Sep 17 00:00:00 2001
From: Katie Byers <lobsterkatie@gmail.com>
Date: Tue, 30 Jan 2024 15:08:03 -0800
Subject: [PATCH] fix tests

---
 .../test_ingest_consumer_processing.py            | 15 +++++++++++++--
 1 file changed, 13 insertions(+), 2 deletions(-)

diff --git a/tests/sentry/ingest/ingest_consumer/test_ingest_consumer_processing.py b/tests/sentry/ingest/ingest_consumer/test_ingest_consumer_processing.py
index 7e2f936898906f..95c56902cb9b76 100644
--- a/tests/sentry/ingest/ingest_consumer/test_ingest_consumer_processing.py
+++ b/tests/sentry/ingest/ingest_consumer/test_ingest_consumer_processing.py
@@ -17,6 +17,7 @@
 
 from sentry import eventstore
 from sentry.event_manager import EventManager
+from sentry.features import has
 from sentry.ingest.consumer.processors import (
     process_attachment_chunk,
     process_event,
@@ -260,7 +261,10 @@ def test_feedbacks_spawn_save_event_feedback(
 @django_db_all
 @pytest.mark.parametrize("missing_chunks", (True, False))
 def test_with_attachments(default_project, task_runner, missing_chunks, monkeypatch, django_cache):
-    monkeypatch.setattr("sentry.features.has", lambda *a, **kw: True)
+    def mock_has(feature_name, *args, **kwargs):
+        return True if "event-attachments" in feature_name else has(feature_name, *args, **kwargs)
+
+    monkeypatch.setattr("sentry.features.has", mock_has)
 
     payload = get_normalized_event({"message": "hello world"}, default_project)
     event_id = payload["event_id"]
@@ -427,7 +431,14 @@ def test_deobfuscate_view_hierarchy(default_project, task_runner):
 def test_individual_attachments(
     default_project, factories, monkeypatch, event_attachments, chunks, with_group, django_cache
 ):
-    monkeypatch.setattr("sentry.features.has", lambda *a, **kw: event_attachments)
+    def mock_has(feature_name, *args, **kwargs):
+        return (
+            event_attachments
+            if "event-attachments" in feature_name
+            else has(feature_name, *args, **kwargs)
+        )
+
+    monkeypatch.setattr("sentry.features.has", mock_has)
 
     event_id = uuid.uuid4().hex
     attachment_id = "ca90fb45-6dd9-40a0-a18f-8693aa621abb"