Skip to content

Commit 1154d5a

Browse files
authored
test: harden 'bucket_cmek' systests against EC (#514)
Closes #513.
1 parent ddd013c commit 1154d5a

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

tests/system/_helpers.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,8 +54,13 @@ def _no_event_based_hold(blob):
5454
return not blob.event_based_hold
5555

5656

57+
def _has_kms_key_name(blob):
58+
return blob.kms_key_name is not None
59+
60+
5761
retry_bad_copy = RetryErrors(exceptions.BadRequest, error_predicate=_bad_copy)
5862
retry_no_event_based_hold = RetryInstanceState(_no_event_based_hold)
63+
retry_has_kms_key_name = RetryInstanceState(_has_kms_key_name)
5964

6065

6166
def unique_name(prefix):

tests/system/test_kms_integration.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -229,7 +229,7 @@ def test_blob_upload_w_bucket_cmek_enabled(
229229
blob.upload_from_string(payload)
230230
blobs_to_delete.append(blob)
231231

232-
_helpers.retry_429_harder(blob.reload)()
232+
_helpers.retry_429_harder(_helpers.retry_has_kms_key_name(blob.reload))()
233233
# We don't know the current version of the key.
234234
assert blob.kms_key_name.startswith(kms_key_name)
235235

0 commit comments

Comments
 (0)