from app.models.post.enums import PostStatus, PostType from app.models.post.exceptions import PostException from . import xray S3_UPLOADS_BUCKET = os.environ.get('S3_UPLOADS_BUCKET') logger = logging.getLogger() xray.patch_all() secrets_manager_client = clients.SecretsManagerClient() clients = { 'appsync': clients.AppSyncClient(), 'cloudfront': clients.CloudFrontClient(secrets_manager_client.get_cloudfront_key_pair), 'dynamo': clients.DynamoClient(), 'mediaconvert': clients.MediaConvertClient(), 'post_verification': clients.PostVerificationClient( secrets_manager_client.get_post_verification_api_creds), 's3_uploads': clients.S3Client(S3_UPLOADS_BUCKET), } managers = {} post_manager = managers.get('post') or models.PostManager(clients, managers=managers)
def cloudfront_client(): yield mock.Mock(clients.CloudFrontClient(None, 'my-domain'))