From 4d9ba7b5cae358902747f925c79c2ad488924115 Mon Sep 17 00:00:00 2001 From: Roman Mogylatov Date: Thu, 14 Oct 2021 15:54:16 -0400 Subject: [PATCH] Migrate container.from_schema() integration tests --- tests/unit/samples/__init__.py | 1 + .../container-multiple-inline.yml | 14 +- .../container-multiple-reordered.yml | 6 +- .../schemasample/container-multiple.yml | 6 +- .../samples/schemasample/container-single.yml | 6 +- tests/unit/schema/conftest.py | 9 + tests/unit/schema/test_container_api_py36.py | 7 +- tests/unit/schema/test_integration_py36.py | 463 +++++++++--------- 8 files changed, 248 insertions(+), 264 deletions(-) create mode 100644 tests/unit/samples/__init__.py create mode 100644 tests/unit/schema/conftest.py diff --git a/tests/unit/samples/__init__.py b/tests/unit/samples/__init__.py new file mode 100644 index 00000000..b7aba378 --- /dev/null +++ b/tests/unit/samples/__init__.py @@ -0,0 +1 @@ +"""Sample code for testing.""" diff --git a/tests/unit/samples/schemasample/container-multiple-inline.yml b/tests/unit/samples/schemasample/container-multiple-inline.yml index a089508f..c9955337 100644 --- a/tests/unit/samples/schemasample/container-multiple-inline.yml +++ b/tests/unit/samples/schemasample/container-multiple-inline.yml @@ -12,7 +12,7 @@ container: provides: sqlite3.connect args: - provider: Callable - provides: schemasample.utils.return_ + provides: samples.schemasample.utils.return_ args: - container.core.config.database.dsn @@ -27,32 +27,32 @@ container: services: user: provider: Factory - provides: schemasample.services.UserService + provides: samples.schemasample.services.UserService kwargs: db: provider: Callable - provides: schemasample.utils.return_ + provides: samples.schemasample.utils.return_ args: - container.gateways.database_client auth: provider: Factory - provides: schemasample.services.AuthService + provides: samples.schemasample.services.AuthService kwargs: db: provider: Callable - provides: schemasample.utils.return_ + provides: samples.schemasample.utils.return_ args: - container.gateways.database_client token_ttl: container.core.config.auth.token_ttl.as_int() photo: provider: Factory - provides: schemasample.services.PhotoService + provides: samples.schemasample.services.PhotoService kwargs: db: provider: Callable - provides: schemasample.utils.return_ + provides: samples.schemasample.utils.return_ args: - container.gateways.database_client s3: container.gateways.s3_client diff --git a/tests/unit/samples/schemasample/container-multiple-reordered.yml b/tests/unit/samples/schemasample/container-multiple-reordered.yml index 245c4f37..cf2ec7f9 100644 --- a/tests/unit/samples/schemasample/container-multiple-reordered.yml +++ b/tests/unit/samples/schemasample/container-multiple-reordered.yml @@ -5,20 +5,20 @@ container: services: user: provider: Factory - provides: schemasample.services.UserService + provides: samples.schemasample.services.UserService kwargs: db: container.gateways.database_client auth: provider: Factory - provides: schemasample.services.AuthService + provides: samples.schemasample.services.AuthService kwargs: db: container.gateways.database_client token_ttl: container.core.config.auth.token_ttl.as_int() photo: provider: Factory - provides: schemasample.services.PhotoService + provides: samples.schemasample.services.PhotoService kwargs: db: container.gateways.database_client s3: container.gateways.s3_client diff --git a/tests/unit/samples/schemasample/container-multiple.yml b/tests/unit/samples/schemasample/container-multiple.yml index 03a5221a..661aa64e 100644 --- a/tests/unit/samples/schemasample/container-multiple.yml +++ b/tests/unit/samples/schemasample/container-multiple.yml @@ -24,20 +24,20 @@ container: services: user: provider: Factory - provides: schemasample.services.UserService + provides: samples.schemasample.services.UserService kwargs: db: container.gateways.database_client auth: provider: Factory - provides: schemasample.services.AuthService + provides: samples.schemasample.services.AuthService kwargs: db: container.gateways.database_client token_ttl: container.core.config.auth.token_ttl.as_int() photo: provider: Factory - provides: schemasample.services.PhotoService + provides: samples.schemasample.services.PhotoService kwargs: db: container.gateways.database_client s3: container.gateways.s3_client diff --git a/tests/unit/samples/schemasample/container-single.yml b/tests/unit/samples/schemasample/container-single.yml index ad732992..057eecca 100644 --- a/tests/unit/samples/schemasample/container-single.yml +++ b/tests/unit/samples/schemasample/container-single.yml @@ -20,20 +20,20 @@ container: user_service: provider: Factory - provides: schemasample.services.UserService + provides: samples.schemasample.services.UserService kwargs: db: container.database_client auth_service: provider: Factory - provides: schemasample.services.AuthService + provides: samples.schemasample.services.AuthService kwargs: db: container.database_client token_ttl: container.config.auth.token_ttl.as_int() photo_service: provider: Factory - provides: schemasample.services.PhotoService + provides: samples.schemasample.services.PhotoService kwargs: db: container.database_client s3: container.s3_client diff --git a/tests/unit/schema/conftest.py b/tests/unit/schema/conftest.py new file mode 100644 index 00000000..fe0a8a30 --- /dev/null +++ b/tests/unit/schema/conftest.py @@ -0,0 +1,9 @@ +"""Container schema fixtures.""" + +from dependency_injector import containers +from pytest import fixture + + +@fixture +def container(): + return containers.DynamicContainer() diff --git a/tests/unit/schema/test_container_api_py36.py b/tests/unit/schema/test_container_api_py36.py index f1ce38c5..b8bc3a0a 100644 --- a/tests/unit/schema/test_container_api_py36.py +++ b/tests/unit/schema/test_container_api_py36.py @@ -7,12 +7,7 @@ import re import yaml from dependency_injector import containers, providers, errors -from pytest import fixture, raises - - -@fixture -def container(): - return containers.DynamicContainer() +from pytest import raises def test_from_schema(container: containers.DynamicContainer): diff --git a/tests/unit/schema/test_integration_py36.py b/tests/unit/schema/test_integration_py36.py index f4eb615c..97cab1d2 100644 --- a/tests/unit/schema/test_integration_py36.py +++ b/tests/unit/schema/test_integration_py36.py @@ -1,294 +1,273 @@ +"""Container tests for building containers from configuration files.""" + +import os import sqlite3 -import unittest from dependency_injector import containers +from pytest import mark -# Runtime import -import os -_TOP_DIR = os.path.abspath( - os.path.sep.join(( - os.path.dirname(__file__), - "../", - )), -) -_SAMPLES_DIR = os.path.abspath( +from samples.schemasample.services import UserService, AuthService, PhotoService + + +SAMPLES_DIR = os.path.abspath( os.path.sep.join(( os.path.dirname(__file__), "../samples/", )), ) -import sys -sys.path.append(_SAMPLES_DIR) - -from schemasample.services import UserService, AuthService, PhotoService -class TestSchemaSingleContainer(unittest.TestCase): +def test_single_container_schema(container: containers.DynamicContainer): + container.from_yaml_schema(f"{SAMPLES_DIR}/schemasample/container-single.yml") + container.config.from_dict({ + "database": { + "dsn": ":memory:", + }, + "aws": { + "access_key_id": "KEY", + "secret_access_key": "SECRET", + }, + "auth": { + "token_ttl": 3600, + }, + }) - def test(self): - container = containers.DynamicContainer() - container.from_yaml_schema(f"{_SAMPLES_DIR}/schemasample/container-single.yml") - container.config.from_dict({ - "database": { - "dsn": ":memory:", - }, - "aws": { - "access_key_id": "KEY", - "secret_access_key": "SECRET", - }, - "auth": { - "token_ttl": 3600, - }, - }) + # User service + user_service1 = container.user_service() + user_service2 = container.user_service() + assert isinstance(user_service1, UserService) + assert isinstance(user_service2, UserService) + assert user_service1 is not user_service2 - # User service - user_service1 = container.user_service() - user_service2 = container.user_service() - self.assertIsInstance(user_service1, UserService) - self.assertIsInstance(user_service2, UserService) - self.assertIsNot(user_service1, user_service2) + assert isinstance(user_service1.db, sqlite3.Connection) + assert isinstance(user_service2.db, sqlite3.Connection) + assert user_service1.db is user_service2.db - self.assertIsInstance(user_service1.db, sqlite3.Connection) - self.assertIsInstance(user_service2.db, sqlite3.Connection) - self.assertIs(user_service1.db, user_service2.db) + # Auth service + auth_service1 = container.auth_service() + auth_service2 = container.auth_service() + assert isinstance(auth_service1, AuthService) + assert isinstance(auth_service2, AuthService) + assert auth_service1 is not auth_service2 - # Auth service - auth_service1 = container.auth_service() - auth_service2 = container.auth_service() - self.assertIsInstance(auth_service1, AuthService) - self.assertIsInstance(auth_service2, AuthService) - self.assertIsNot(auth_service1, auth_service2) + assert isinstance(auth_service1.db, sqlite3.Connection) + assert isinstance(auth_service2.db, sqlite3.Connection) + assert auth_service1.db is auth_service2.db + assert auth_service1.db is container.database_client() + assert auth_service2.db is container.database_client() - self.assertIsInstance(auth_service1.db, sqlite3.Connection) - self.assertIsInstance(auth_service2.db, sqlite3.Connection) - self.assertIs(auth_service1.db, auth_service2.db) - self.assertIs(auth_service1.db, container.database_client()) - self.assertIs(auth_service2.db, container.database_client()) + assert auth_service1.token_ttl == 3600 + assert auth_service2.token_ttl == 3600 - self.assertEqual(auth_service1.token_ttl, 3600) - self.assertEqual(auth_service2.token_ttl, 3600) + # Photo service + photo_service1 = container.photo_service() + photo_service2 = container.photo_service() + assert isinstance(photo_service1, PhotoService) + assert isinstance(photo_service2, PhotoService) + assert photo_service1 is not photo_service2 - # Photo service - photo_service1 = container.photo_service() - photo_service2 = container.photo_service() - self.assertIsInstance(photo_service1, PhotoService) - self.assertIsInstance(photo_service2, PhotoService) - self.assertIsNot(photo_service1, photo_service2) + assert isinstance(photo_service1.db, sqlite3.Connection) + assert isinstance(photo_service2.db, sqlite3.Connection) + assert photo_service1.db is photo_service2.db + assert photo_service1.db is container.database_client() + assert photo_service2.db is container.database_client() - self.assertIsInstance(photo_service1.db, sqlite3.Connection) - self.assertIsInstance(photo_service2.db, sqlite3.Connection) - self.assertIs(photo_service1.db, photo_service2.db) - self.assertIs(photo_service1.db, container.database_client()) - self.assertIs(photo_service2.db, container.database_client()) - - self.assertIs(photo_service1.s3, photo_service2.s3) - self.assertIs(photo_service1.s3, container.s3_client()) - self.assertIs(photo_service2.s3, container.s3_client()) + assert photo_service1.s3 is photo_service2.s3 + assert photo_service1.s3 is container.s3_client() + assert photo_service2.s3 is container.s3_client() -class TestSchemaMultipleContainers(unittest.TestCase): +def test_multiple_containers_schema(container: containers.DynamicContainer): + container.from_yaml_schema(f"{SAMPLES_DIR}/schemasample/container-multiple.yml") + container.core.config.from_dict({ + "database": { + "dsn": ":memory:", + }, + "aws": { + "access_key_id": "KEY", + "secret_access_key": "SECRET", + }, + "auth": { + "token_ttl": 3600, + }, + }) - def test(self): - container = containers.DynamicContainer() - container.from_yaml_schema(f"{_SAMPLES_DIR}/schemasample/container-multiple.yml") - container.core.config.from_dict({ - "database": { - "dsn": ":memory:", - }, - "aws": { - "access_key_id": "KEY", - "secret_access_key": "SECRET", - }, - "auth": { - "token_ttl": 3600, - }, - }) + # User service + user_service1 = container.services.user() + user_service2 = container.services.user() + assert isinstance(user_service1, UserService) + assert isinstance(user_service2, UserService) + assert user_service1 is not user_service2 - # User service - user_service1 = container.services.user() - user_service2 = container.services.user() - self.assertIsInstance(user_service1, UserService) - self.assertIsInstance(user_service2, UserService) - self.assertIsNot(user_service1, user_service2) + assert isinstance(user_service1.db, sqlite3.Connection) + assert isinstance(user_service2.db, sqlite3.Connection) + assert user_service1.db is user_service2.db - self.assertIsInstance(user_service1.db, sqlite3.Connection) - self.assertIsInstance(user_service2.db, sqlite3.Connection) - self.assertIs(user_service1.db, user_service2.db) + # Auth service + auth_service1 = container.services.auth() + auth_service2 = container.services.auth() + assert isinstance(auth_service1, AuthService) + assert isinstance(auth_service2, AuthService) + assert auth_service1 is not auth_service2 - # Auth service - auth_service1 = container.services.auth() - auth_service2 = container.services.auth() - self.assertIsInstance(auth_service1, AuthService) - self.assertIsInstance(auth_service2, AuthService) - self.assertIsNot(auth_service1, auth_service2) + assert isinstance(auth_service1.db, sqlite3.Connection) + assert isinstance(auth_service2.db, sqlite3.Connection) + assert auth_service1.db is auth_service2.db + assert auth_service1.db is container.gateways.database_client() + assert auth_service2.db is container.gateways.database_client() - self.assertIsInstance(auth_service1.db, sqlite3.Connection) - self.assertIsInstance(auth_service2.db, sqlite3.Connection) - self.assertIs(auth_service1.db, auth_service2.db) - self.assertIs(auth_service1.db, container.gateways.database_client()) - self.assertIs(auth_service2.db, container.gateways.database_client()) + assert auth_service1.token_ttl == 3600 + assert auth_service2.token_ttl == 3600 - self.assertEqual(auth_service1.token_ttl, 3600) - self.assertEqual(auth_service2.token_ttl, 3600) + # Photo service + photo_service1 = container.services.photo() + photo_service2 = container.services.photo() + assert isinstance(photo_service1, PhotoService) + assert isinstance(photo_service2, PhotoService) + assert photo_service1 is not photo_service2 - # Photo service - photo_service1 = container.services.photo() - photo_service2 = container.services.photo() - self.assertIsInstance(photo_service1, PhotoService) - self.assertIsInstance(photo_service2, PhotoService) - self.assertIsNot(photo_service1, photo_service2) + assert isinstance(photo_service1.db, sqlite3.Connection) + assert isinstance(photo_service2.db, sqlite3.Connection) + assert photo_service1.db is photo_service2.db + assert photo_service1.db is container.gateways.database_client() + assert photo_service2.db is container.gateways.database_client() - self.assertIsInstance(photo_service1.db, sqlite3.Connection) - self.assertIsInstance(photo_service2.db, sqlite3.Connection) - self.assertIs(photo_service1.db, photo_service2.db) - self.assertIs(photo_service1.db, container.gateways.database_client()) - self.assertIs(photo_service2.db, container.gateways.database_client()) - - self.assertIs(photo_service1.s3, photo_service2.s3) - self.assertIs(photo_service1.s3, container.gateways.s3_client()) - self.assertIs(photo_service2.s3, container.gateways.s3_client()) + assert photo_service1.s3 is photo_service2.s3 + assert photo_service1.s3 is container.gateways.s3_client() + assert photo_service2.s3 is container.gateways.s3_client() -class TestSchemaMultipleContainersReordered(unittest.TestCase): +def test_multiple_reordered_containers_schema(container: containers.DynamicContainer): + container.from_yaml_schema(f"{SAMPLES_DIR}/schemasample/container-multiple-reordered.yml") + container.core.config.from_dict({ + "database": { + "dsn": ":memory:", + }, + "aws": { + "access_key_id": "KEY", + "secret_access_key": "SECRET", + }, + "auth": { + "token_ttl": 3600, + }, + }) - def test(self): - container = containers.DynamicContainer() - container.from_yaml_schema(f"{_SAMPLES_DIR}/schemasample/container-multiple-reordered.yml") - container.core.config.from_dict({ - "database": { - "dsn": ":memory:", - }, - "aws": { - "access_key_id": "KEY", - "secret_access_key": "SECRET", - }, - "auth": { - "token_ttl": 3600, - }, - }) + # User service + user_service1 = container.services.user() + user_service2 = container.services.user() + assert isinstance(user_service1, UserService) + assert isinstance(user_service2, UserService) + assert user_service1 is not user_service2 - # User service - user_service1 = container.services.user() - user_service2 = container.services.user() - self.assertIsInstance(user_service1, UserService) - self.assertIsInstance(user_service2, UserService) - self.assertIsNot(user_service1, user_service2) + assert isinstance(user_service1.db, sqlite3.Connection) + assert isinstance(user_service2.db, sqlite3.Connection) + assert user_service1.db is user_service2.db - self.assertIsInstance(user_service1.db, sqlite3.Connection) - self.assertIsInstance(user_service2.db, sqlite3.Connection) - self.assertIs(user_service1.db, user_service2.db) + # Auth service + auth_service1 = container.services.auth() + auth_service2 = container.services.auth() + assert isinstance(auth_service1, AuthService) + assert isinstance(auth_service2, AuthService) + assert auth_service1 is not auth_service2 - # Auth service - auth_service1 = container.services.auth() - auth_service2 = container.services.auth() - self.assertIsInstance(auth_service1, AuthService) - self.assertIsInstance(auth_service2, AuthService) - self.assertIsNot(auth_service1, auth_service2) + assert isinstance(auth_service1.db, sqlite3.Connection) + assert isinstance(auth_service2.db, sqlite3.Connection) + assert auth_service1.db is auth_service2.db + assert auth_service1.db is container.gateways.database_client() + assert auth_service2.db is container.gateways.database_client() - self.assertIsInstance(auth_service1.db, sqlite3.Connection) - self.assertIsInstance(auth_service2.db, sqlite3.Connection) - self.assertIs(auth_service1.db, auth_service2.db) - self.assertIs(auth_service1.db, container.gateways.database_client()) - self.assertIs(auth_service2.db, container.gateways.database_client()) + assert auth_service1.token_ttl == 3600 + assert auth_service2.token_ttl == 3600 - self.assertEqual(auth_service1.token_ttl, 3600) - self.assertEqual(auth_service2.token_ttl, 3600) + # Photo service + photo_service1 = container.services.photo() + photo_service2 = container.services.photo() + assert isinstance(photo_service1, PhotoService) + assert isinstance(photo_service2, PhotoService) + assert photo_service1 is not photo_service2 - # Photo service - photo_service1 = container.services.photo() - photo_service2 = container.services.photo() - self.assertIsInstance(photo_service1, PhotoService) - self.assertIsInstance(photo_service2, PhotoService) - self.assertIsNot(photo_service1, photo_service2) + assert isinstance(photo_service1.db, sqlite3.Connection) + assert isinstance(photo_service2.db, sqlite3.Connection) + assert photo_service1.db is photo_service2.db + assert photo_service1.db is container.gateways.database_client() + assert photo_service2.db is container.gateways.database_client() - self.assertIsInstance(photo_service1.db, sqlite3.Connection) - self.assertIsInstance(photo_service2.db, sqlite3.Connection) - self.assertIs(photo_service1.db, photo_service2.db) - self.assertIs(photo_service1.db, container.gateways.database_client()) - self.assertIs(photo_service2.db, container.gateways.database_client()) - - self.assertIs(photo_service1.s3, photo_service2.s3) - self.assertIs(photo_service1.s3, container.gateways.s3_client()) - self.assertIs(photo_service2.s3, container.gateways.s3_client()) + assert photo_service1.s3 is photo_service2.s3 + assert photo_service1.s3 is container.gateways.s3_client() + assert photo_service2.s3 is container.gateways.s3_client() -class TestSchemaMultipleContainersWithInlineProviders(unittest.TestCase): +def test_multiple_containers_with_inline_providers_schema(container: containers.DynamicContainer): + container.from_yaml_schema(f"{SAMPLES_DIR}/schemasample/container-multiple-inline.yml") + container.core.config.from_dict({ + "database": { + "dsn": ":memory:", + }, + "aws": { + "access_key_id": "KEY", + "secret_access_key": "SECRET", + }, + "auth": { + "token_ttl": 3600, + }, + }) - def test(self): - container = containers.DynamicContainer() - container.from_yaml_schema(f"{_SAMPLES_DIR}/schemasample/container-multiple-inline.yml") - container.core.config.from_dict({ - "database": { - "dsn": ":memory:", - }, - "aws": { - "access_key_id": "KEY", - "secret_access_key": "SECRET", - }, - "auth": { - "token_ttl": 3600, - }, - }) + # User service + user_service1 = container.services.user() + user_service2 = container.services.user() + assert isinstance(user_service1, UserService) + assert isinstance(user_service2, UserService) + assert user_service1 is not user_service2 - # User service - user_service1 = container.services.user() - user_service2 = container.services.user() - self.assertIsInstance(user_service1, UserService) - self.assertIsInstance(user_service2, UserService) - self.assertIsNot(user_service1, user_service2) + assert isinstance(user_service1.db, sqlite3.Connection) + assert isinstance(user_service2.db, sqlite3.Connection) + assert user_service1.db is user_service2.db - self.assertIsInstance(user_service1.db, sqlite3.Connection) - self.assertIsInstance(user_service2.db, sqlite3.Connection) - self.assertIs(user_service1.db, user_service2.db) + # Auth service + auth_service1 = container.services.auth() + auth_service2 = container.services.auth() + assert isinstance(auth_service1, AuthService) + assert isinstance(auth_service2, AuthService) + assert auth_service1 is not auth_service2 - # Auth service - auth_service1 = container.services.auth() - auth_service2 = container.services.auth() - self.assertIsInstance(auth_service1, AuthService) - self.assertIsInstance(auth_service2, AuthService) - self.assertIsNot(auth_service1, auth_service2) + assert isinstance(auth_service1.db, sqlite3.Connection) + assert isinstance(auth_service2.db, sqlite3.Connection) + assert auth_service1.db is auth_service2.db + assert auth_service1.db is container.gateways.database_client() + assert auth_service2.db is container.gateways.database_client() - self.assertIsInstance(auth_service1.db, sqlite3.Connection) - self.assertIsInstance(auth_service2.db, sqlite3.Connection) - self.assertIs(auth_service1.db, auth_service2.db) - self.assertIs(auth_service1.db, container.gateways.database_client()) - self.assertIs(auth_service2.db, container.gateways.database_client()) + assert auth_service1.token_ttl == 3600 + assert auth_service2.token_ttl == 3600 - self.assertEqual(auth_service1.token_ttl, 3600) - self.assertEqual(auth_service2.token_ttl, 3600) + # Photo service + photo_service1 = container.services.photo() + photo_service2 = container.services.photo() + assert isinstance(photo_service1, PhotoService) + assert isinstance(photo_service2, PhotoService) + assert photo_service1 is not photo_service2 - # Photo service - photo_service1 = container.services.photo() - photo_service2 = container.services.photo() - self.assertIsInstance(photo_service1, PhotoService) - self.assertIsInstance(photo_service2, PhotoService) - self.assertIsNot(photo_service1, photo_service2) + assert isinstance(photo_service1.db, sqlite3.Connection) + assert isinstance(photo_service2.db, sqlite3.Connection) + assert photo_service1.db is photo_service2.db + assert photo_service1.db is container.gateways.database_client() + assert photo_service2.db is container.gateways.database_client() - self.assertIsInstance(photo_service1.db, sqlite3.Connection) - self.assertIsInstance(photo_service2.db, sqlite3.Connection) - self.assertIs(photo_service1.db, photo_service2.db) - self.assertIs(photo_service1.db, container.gateways.database_client()) - self.assertIs(photo_service2.db, container.gateways.database_client()) - - self.assertIs(photo_service1.s3, photo_service2.s3) - self.assertIs(photo_service1.s3, container.gateways.s3_client()) - self.assertIs(photo_service2.s3, container.gateways.s3_client()) + assert photo_service1.s3 is photo_service2.s3 + assert photo_service1.s3 is container.gateways.s3_client() + assert photo_service2.s3 is container.gateways.s3_client() -class TestSchemaBoto3Session(unittest.TestCase): +@mark.skip(reason="Boto3 tries to connect to the internet") +def test_schema_with_boto3_session(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", + }, + ) - @unittest.skip("Boto3 tries to connect to the internet") - 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") + assert container.s3_client().__class__.__name__ == "S3" + assert container.sqs_client().__class__.__name__ == "SQS"