# -*- coding: utf-8 -*- # type: ignore from __future__ import annotations import datetime import json import boto3 import moto import pytest from pytest_mock import MockerFixture from keep_it_secret.ext import aws from keep_it_secret.secrets import Secrets def test_init(): # When result = aws.AWSSecretsManagerField('keep_it_secret/tests/spam') # Then assert result.secret_id == 'keep_it_secret/tests/spam' assert result.default is None assert result.decoder == json.loads def test_init_with_default(): # When result = aws.AWSSecretsManagerField( 'keep_it_secret/tests/spam', default='eggs', ) # Then assert result.default == 'eggs' def test_init_with_decoder(): # When result = aws.AWSSecretsManagerField( 'keep_it_secret/tests/spam', decoder=int, ) # Then assert result.decoder == int def test_init_with_field_options(): # When result = aws.AWSSecretsManagerField( 'keep_it_secret/tests/spam', as_type=int, required=False, description='spameggs', ) # Then assert result.as_type is None assert result.required is False assert result.description == 'spameggs' def test_new(mocker: MockerFixture): # Given mock_init = mocker.patch.object( aws.AWSSecretsManagerField, '__init__', return_value=None, ) # When _ = aws.AWSSecretsManagerField.new( 'keep_it_secret/tests/spam', default='eggs', decoder=json.dumps, as_type=int, required=False, description='spameggs', ) # Then mock_init.assert_called_once_with( 'keep_it_secret/tests/spam', default='eggs', decoder=json.dumps, as_type=int, required=False, description='spameggs', ) def test_get_value_aws_dependency_missing(mocker: MockerFixture, testing_secrets: Secrets): # Given field = aws.AWSSecretsManagerField('keep_it_secret/tests/spam') with pytest.raises(field.DependencyMissing) as exception_info: # When _ = field(testing_secrets) # Then assert exception_info.value.args[0] == 'aws' @moto.mock_secretsmanager def test_get_value_required_value_not_found(testing_aws_secrets: Secrets): # Given field = aws.AWSSecretsManagerField('keep_it_secret/tests/spam') with pytest.raises(field.RequiredValueMissing) as exception_info: # When _ = field(testing_aws_secrets) # Then assert exception_info.value.args[0] == 'keep_it_secret/tests/spam' @moto.mock_secretsmanager def test_get_value_not_found_not_required(testing_aws_secrets: Secrets): # Given field = aws.AWSSecretsManagerField( 'keep_it_secret/tests/spam', required=False, ) result = field(testing_aws_secrets) # Then assert result is None @moto.mock_secretsmanager def test_get_value(testing_aws_secrets: Secrets): # Given aws_secrets_manager_client = boto3.client('secretsmanager') aws_secrets_manager_client.create_secret( Name='keep_it_secret/tests/spam', SecretString='{"spam":true}', ) field = aws.AWSSecretsManagerField('keep_it_secret/tests/spam') # When result = field(testing_aws_secrets) # Then assert result == {"spam": True} @moto.mock_secretsmanager def test_get_value_with_custom_decoder(testing_aws_secrets: Secrets): # Given aws_secrets_manager_client = boto3.client('secretsmanager') aws_secrets_manager_client.create_secret( Name='keep_it_secret/tests/spam', SecretString='1987-10-03T08:00:00', ) field = aws.AWSSecretsManagerField( 'keep_it_secret/tests/spam', decoder=datetime.datetime.fromisoformat, ) # When result = field(testing_aws_secrets) # Then assert result == datetime.datetime(1987, 10, 3, 8, 0, 0)