diff --git a/tests/unit/samples/schemasample/container-boto3-session.yml b/tests/unit/samples/schemasample/container-boto3-session.yml index 7f530a4e..1c10c085 100644 --- a/tests/unit/samples/schemasample/container-boto3-session.yml +++ b/tests/unit/samples/schemasample/container-boto3-session.yml @@ -12,6 +12,7 @@ container: aws_access_key_id: container.config.aws_access_key_id aws_secret_access_key: container.config.aws_secret_access_key aws_session_token: container.config.aws_session_token + region_name: container.config.aws_region_name s3_client: provider: Factory diff --git a/tests/unit/schema/test_integration_py36.py b/tests/unit/schema/test_integration_py36.py index 38a48a04..2cc77e1a 100644 --- a/tests/unit/schema/test_integration_py36.py +++ b/tests/unit/schema/test_integration_py36.py @@ -280,6 +280,14 @@ class TestSchemaBoto3Session(unittest.TestCase): def test(self): container = containers.DynamicContainer() container.from_yaml_schema(f'{_SAMPLES_DIR}/schemasample/container-boto3-session.yml') + container.config.from_dict( + { + 'aws_access_key_id': 'key', + 'aws_secret_access_key': 'secret', + 'aws_session_token': 'token', + 'aws_region_name': 'us-east-1', + }, + ) self.assertEqual(container.s3_client().__class__.__name__, 'S3') self.assertEqual(container.sqs_client().__class__.__name__, 'SQS')