python-dependency-injector/examples/containers/declarative_override_providers.py

18 lines
405 B
Python
Raw Normal View History

2020-08-14 23:27:17 +03:00
"""Declarative container provider override example."""
import sqlite3
from unittest import mock
from dependency_injector import containers, providers
class Container(containers.DeclarativeContainer):
database = providers.Singleton(sqlite3.connect, ':memory:')
container = Container(database=mock.Mock(sqlite3.Connection))
database = container.database()
assert isinstance(database, mock.Mock)