python-dependency-injector/tests/unit/providers/traversal/test_dependencies_container_py3.py
2021-10-18 10:30:46 -04:00

53 lines
1.4 KiB
Python

"""DependenciesContainer provider traversal tests."""
from dependency_injector import providers
def test_traversal():
provider = providers.DependenciesContainer()
all_providers = list(provider.traverse())
assert len(all_providers) == 0
def test_traversal_default():
another_provider = providers.Provider()
provider = providers.DependenciesContainer(default=another_provider)
all_providers = list(provider.traverse())
assert len(all_providers) == 1
assert another_provider in all_providers
def test_traversal_fluent_interface():
provider = providers.DependenciesContainer()
provider1 = provider.provider1
provider2 = provider.provider2
all_providers = list(provider.traverse())
assert len(all_providers) == 2
assert provider1 in all_providers
assert provider2 in all_providers
def test_traversal_overriding():
provider1 = providers.Provider()
provider2 = providers.Provider()
provider3 = providers.DependenciesContainer(
provider1=provider1,
provider2=provider2,
)
provider = providers.DependenciesContainer()
provider.override(provider3)
all_providers = list(provider.traverse())
assert len(all_providers) == 5
assert provider1 in all_providers
assert provider2 in all_providers
assert provider3 in all_providers
assert provider.provider1 in all_providers
assert provider.provider2 in all_providers