test_f_aws_encryption_sdk_client.py 文件源码

def fake_kms_client(keysize=32):
    mock_kms_client = MagicMock(__class__=botocore.client.BaseClient)
    mock_kms_client.generate_data_key.return_value = {
        'Plaintext': VALUES['data_keys'][keysize]['plaintext'],
        'CiphertextBlob': VALUES['data_keys'][keysize]['encrypted'],
        'KeyId': VALUES['arn']
    }
    mock_kms_client.encrypt.return_value = {
        'CiphertextBlob': VALUES['data_keys'][keysize]['encrypted'],
        'KeyId': VALUES['arn']
    }
    mock_kms_client.decrypt.return_value = {
        'Plaintext': VALUES['data_keys'][keysize]['plaintext'],
        'KeyId': VALUES['arn']
    }
    return mock_kms_client