Skip to content

Commit c6baca8

Browse files
committed
chore: add positive-case tests for filtering client suppliers when the requested region is unknown and resolved by underlying client supplier
1 parent c1cd751 commit c6baca8

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

test/functional/keyrings/aws_kms/test_client_suppliers.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -136,6 +136,30 @@ def test_allow_deny_nested_supplier():
136136
excinfo.match("Unable to provide client for region 'us-west-2'")
137137

138138

139+
def test_deny_region_allowed_region_resolved_by_client_supplier():
140+
region = "us-west-2"
141+
supplier = DenyRegionsClientSupplier(
142+
denied_regions=["us-east-1"], client_supplier=AlwaysOneRegionClientSupplier(region_name=region)
143+
)
144+
145+
# Resolves to a region that is not blocked.
146+
test = supplier(None)
147+
148+
assert test.meta.region_name == region
149+
150+
151+
def test_allow_region_allowed_region_resolved_by_client_supplier():
152+
region = "us-west-2"
153+
supplier = AllowRegionsClientSupplier(
154+
allowed_regions=[None, region], client_supplier=AlwaysOneRegionClientSupplier(region_name=region)
155+
)
156+
157+
# Resolves to a region that is allowed
158+
test = supplier(None)
159+
160+
assert test.meta.region_name == region
161+
162+
139163
def test_deny_region_block_on_response():
140164
region = "us-west-2"
141165
test = DenyRegionsClientSupplier(

0 commit comments

Comments
 (0)