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

32 lines
867 B
Python

"""Dict provider traversal tests."""
from dependency_injector import providers
def test_traverse_kwargs():
provider1 = providers.Object("bar")
provider2 = providers.Object("baz")
provider = providers.Dict(foo="foo", bar=provider1, baz=provider2)
all_providers = list(provider.traverse())
assert len(all_providers) == 2
assert provider1 in all_providers
assert provider2 in all_providers
def test_traverse_overridden():
provider1 = providers.Object("bar")
provider2 = providers.Object("baz")
provider3 = providers.Dict(bar=provider1, baz=provider2)
provider = providers.Dict(foo="foo")
provider.override(provider3)
all_providers = list(provider.traverse())
assert len(all_providers) == 3
assert provider1 in all_providers
assert provider2 in all_providers
assert provider3 in all_providers