29 lines
675 B
Python
29 lines
675 B
Python
import os
|
|
from functools import wraps
|
|
|
|
from moto import mock_kms
|
|
|
|
|
|
def kms_aws_verified(func):
|
|
"""
|
|
Function that is verified to work against AWS.
|
|
Can be run against AWS at any time by setting:
|
|
MOTO_TEST_ALLOW_AWS_REQUEST=true
|
|
|
|
If this environment variable is not set, the function runs in a `mock_kms` context.
|
|
"""
|
|
|
|
@wraps(func)
|
|
def pagination_wrapper():
|
|
allow_aws_request = (
|
|
os.environ.get("MOTO_TEST_ALLOW_AWS_REQUEST", "false").lower() == "true"
|
|
)
|
|
|
|
if allow_aws_request:
|
|
return func()
|
|
else:
|
|
with mock_kms():
|
|
return func()
|
|
|
|
return pagination_wrapper
|