Skip to content

Commit 7f5e7fd

Browse files
authored
test(hc): Clean up clear_global_regions uses (#61470)
1 parent eb4ee6a commit 7f5e7fd

File tree

2 files changed

+17
-54
lines changed

2 files changed

+17
-54
lines changed

tests/sentry/api/test_base.py

Lines changed: 6 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,8 @@
1515
from sentry.silo import SiloMode
1616
from sentry.testutils.cases import APITestCase
1717
from sentry.testutils.helpers.options import override_options
18-
from sentry.testutils.region import override_region_config
19-
from sentry.testutils.silo import all_silo_test, assume_test_silo_mode
20-
from sentry.types.region import RegionCategory, clear_global_regions, subdomain_is_region
18+
from sentry.testutils.silo import all_silo_test, assume_test_silo_mode, create_test_regions
19+
from sentry.types.region import subdomain_is_region
2120
from sentry.utils.cursors import Cursor
2221

2322

@@ -500,33 +499,17 @@ def test_non_json_content_type(self):
500499
assert not self.request.json_body
501500

502501

502+
@all_silo_test(regions=create_test_regions("us", "eu"))
503503
class CustomerDomainTest(APITestCase):
504504
def test_resolve_region(self):
505-
clear_global_regions()
506-
507505
def request_with_subdomain(subdomain):
508506
request = self.make_request(method="GET")
509507
request.subdomain = subdomain
510508
return subdomain_is_region(request)
511509

512-
region_config = [
513-
{
514-
"name": "us",
515-
"snowflake_id": 1,
516-
"address": "http://us.testserver",
517-
"category": RegionCategory.MULTI_TENANT.name,
518-
},
519-
{
520-
"name": "eu",
521-
"snowflake_id": 1,
522-
"address": "http://eu.testserver",
523-
"category": RegionCategory.MULTI_TENANT.name,
524-
},
525-
]
526-
with override_region_config(region_config):
527-
assert request_with_subdomain("us")
528-
assert request_with_subdomain("eu")
529-
assert not request_with_subdomain("sentry")
510+
assert request_with_subdomain("us")
511+
assert request_with_subdomain("eu")
512+
assert not request_with_subdomain("sentry")
530513

531514

532515
class EndpointSiloLimitTest(APITestCase):

tests/sentry/middleware/test_customer_domain.py

Lines changed: 11 additions & 31 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
from __future__ import annotations
22

3-
from typing import Any
43
from unittest import mock
54

65
from django.conf import settings
@@ -15,9 +14,7 @@
1514
from sentry.middleware.customer_domain import CustomerDomainMiddleware
1615
from sentry.testutils.cases import APITestCase, TestCase
1716
from sentry.testutils.helpers import with_feature
18-
from sentry.testutils.region import override_region_config
19-
from sentry.testutils.silo import no_silo_test
20-
from sentry.types.region import RegionCategory, clear_global_regions
17+
from sentry.testutils.silo import all_silo_test, create_test_regions, no_silo_test
2118
from sentry.web.frontend.auth_logout import AuthLogoutView
2219

2320

@@ -27,9 +24,8 @@ def _session(d: dict[str, str]) -> SessionBase:
2724
return ret
2825

2926

30-
@override_settings(
31-
SENTRY_USE_CUSTOMER_DOMAINS=True,
32-
)
27+
@all_silo_test(regions=create_test_regions("us", "eu"))
28+
@override_settings(SENTRY_USE_CUSTOMER_DOMAINS=True)
3329
class CustomerDomainMiddlewareTest(TestCase):
3430
def test_sets_active_organization_if_exists(self):
3531
self.create_organization(name="test")
@@ -128,30 +124,14 @@ def test_no_op(self):
128124
assert response == mock.sentinel.response
129125

130126
def test_ignores_region_subdomains(self):
131-
clear_global_regions()
132-
region_configs: list[dict[str, Any]] = [
133-
{
134-
"name": "us",
135-
"snowflake_id": 1,
136-
"address": "http://us.testserver",
137-
"category": RegionCategory.MULTI_TENANT.name,
138-
},
139-
{
140-
"name": "eu",
141-
"snowflake_id": 1,
142-
"address": "http://eu.testserver",
143-
"category": RegionCategory.MULTI_TENANT.name,
144-
},
145-
]
146-
with override_region_config(region_configs):
147-
for region in region_configs:
148-
request = RequestFactory().get("/")
149-
request.subdomain = region["name"]
150-
request.session = _session({"activeorg": "test"})
151-
response = CustomerDomainMiddleware(lambda request: mock.sentinel.response)(request)
152-
153-
assert dict(request.session) == {"activeorg": "test"}
154-
assert response == mock.sentinel.response
127+
for region_name in ("us", "eu"):
128+
request = RequestFactory().get("/")
129+
request.subdomain = region_name
130+
request.session = _session({"activeorg": "test"})
131+
response = CustomerDomainMiddleware(lambda request: mock.sentinel.response)(request)
132+
133+
assert dict(request.session) == {"activeorg": "test"}
134+
assert response == mock.sentinel.response
155135

156136
def test_handles_redirects(self):
157137
self.create_organization(name="sentry")

0 commit comments

Comments
 (0)