fix tests

getsentry · Feb 20, 2025 · 8811fc1 · 8811fc1
1 parent a9f97f1
commit 8811fc1
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 2 deletions.
diff --git a/tests/sentry/uptime/consumers/test_results_consumer.py b/tests/sentry/uptime/consumers/test_results_consumer.py
@@ -42,6 +42,7 @@
     ProjectUptimeSubscriptionMode,
     UptimeStatus,
     UptimeSubscription,
+    UptimeSubscriptionRegion,
 )
 from sentry.utils import json
 from tests.sentry.uptime.subscriptions.test_tasks import ConfigPusherTestMixin
@@ -1024,7 +1025,14 @@ def run_check_and_update_region_test(
 
         with (
             override_settings(UPTIME_REGIONS=region_configs),
-            override_options({"uptime.disabled-checker-regions": disabled_regions}),
+            override_options(
+                {
+                    "uptime.checker-regions-mode-override": {
+                        region: UptimeSubscriptionRegion.RegionMode.INACTIVE.value
+                        for region in disabled_regions
+                    }
+                }
+            ),
             self.tasks(),
             freeze_time((datetime.now() - timedelta(hours=1)).replace(minute=current_minute)),
             mock.patch("random.random", return_value=0),

diff --git a/tests/sentry/uptime/subscriptions/test_subscriptions.py b/tests/sentry/uptime/subscriptions/test_subscriptions.py
@@ -16,6 +16,7 @@
     ProjectUptimeSubscription,
     ProjectUptimeSubscriptionMode,
     UptimeSubscription,
+    UptimeSubscriptionRegion,
 )
 from sentry.uptime.subscriptions.subscriptions import (
     UPTIME_SUBSCRIPTION_TYPE,
@@ -292,7 +293,13 @@ def test_auto_associates_active_regions(self):
         ]
         with (
             override_settings(UPTIME_REGIONS=regions),
-            override_options({"uptime.disabled-checker-regions": ["region3"]}),
+            override_options(
+                {
+                    "uptime.checker-regions-mode-override": {
+                        "region3": UptimeSubscriptionRegion.RegionMode.INACTIVE.value
+                    }
+                }
+            ),
         ):
             subscription = get_or_create_uptime_subscription(
                 url="https://example.com",