mirror of
https://github.com/ets-labs/python-dependency-injector.git
synced 2025-07-04 20:33:13 +03:00
Migrate selector provider tests
This commit is contained in:
Roman Mogylatov
parent
7774ed391d
commit
d9ba72f950
|
|
@ -1,220 +1,209 @@
|
||||||
"""Dependency injector selector provider unit tests."""
|
"""Selector provider tests."""
|
||||||
|
|
||||||
import functools
|
import functools
|
||||||
import itertools
|
import itertools
|
||||||
import sys
|
import sys
|
||||||
|
|
||||||
import unittest
|
|
||||||
|
|
||||||
from dependency_injector import providers, errors
|
from dependency_injector import providers, errors
|
||||||
from pytest import raises
|
from pytest import fixture, mark, raises
|
||||||
|
|
||||||
|
|
||||||
class SelectorTests(unittest.TestCase):
|
@fixture
|
||||||
|
def switch():
|
||||||
|
return providers.Configuration()
|
||||||
|
|
||||||
selector = providers.Configuration()
|
|
||||||
|
|
||||||
def test_is_provider(self):
|
@fixture
|
||||||
assert providers.is_provider(providers.Selector(self.selector)) is True
|
def one():
|
||||||
|
return providers.Object(1)
|
||||||
|
|
||||||
def test_init_optional(self):
|
|
||||||
one = providers.Object(1)
|
|
||||||
two = providers.Object(2)
|
|
||||||
|
|
||||||
provider = providers.Selector()
|
@fixture
|
||||||
provider.set_selector(self.selector)
|
def two():
|
||||||
provider.set_providers(one=one, two=two)
|
return providers.Object(2)
|
||||||
|
|
||||||
assert provider.providers == {"one": one, "two": two}
|
|
||||||
with self.selector.override("one"):
|
|
||||||
assert provider() == one()
|
|
||||||
with self.selector.override("two"):
|
|
||||||
assert provider() == two()
|
|
||||||
|
|
||||||
def test_set_selector_returns_self(self):
|
@fixture
|
||||||
provider = providers.Selector()
|
def selector_type():
|
||||||
assert provider.set_selector(self.selector) is provider
|
return "default"
|
||||||
|
|
||||||
def test_set_providers_returns_self(self):
|
|
||||||
provider = providers.Selector()
|
|
||||||
assert provider.set_providers(one=providers.Provider()) is provider
|
|
||||||
|
|
||||||
def test_provided_instance_provider(self):
|
@fixture
|
||||||
provider = providers.Selector(self.selector)
|
def selector(selector_type, switch, one, two):
|
||||||
assert isinstance(provider.provided, providers.ProvidedInstance)
|
if selector_type == "default":
|
||||||
|
return providers.Selector(switch, one=one, two=two)
|
||||||
def test_call(self):
|
elif selector_type == "empty":
|
||||||
provider = providers.Selector(
|
return providers.Selector()
|
||||||
self.selector,
|
elif selector_type == "sys-streams":
|
||||||
one=providers.Object(1),
|
return providers.Selector(
|
||||||
two=providers.Object(2),
|
switch,
|
||||||
)
|
|
||||||
|
|
||||||
with self.selector.override("one"):
|
|
||||||
assert provider() == 1
|
|
||||||
|
|
||||||
with self.selector.override("two"):
|
|
||||||
assert provider() == 2
|
|
||||||
|
|
||||||
def test_call_undefined_provider(self):
|
|
||||||
provider = providers.Selector(
|
|
||||||
self.selector,
|
|
||||||
one=providers.Object(1),
|
|
||||||
two=providers.Object(2),
|
|
||||||
)
|
|
||||||
|
|
||||||
with self.selector.override("three"):
|
|
||||||
with raises(errors.Error):
|
|
||||||
provider()
|
|
||||||
|
|
||||||
def test_call_selector_is_none(self):
|
|
||||||
provider = providers.Selector(
|
|
||||||
self.selector,
|
|
||||||
one=providers.Object(1),
|
|
||||||
two=providers.Object(2),
|
|
||||||
)
|
|
||||||
|
|
||||||
with self.selector.override(None):
|
|
||||||
with raises(errors.Error):
|
|
||||||
provider()
|
|
||||||
|
|
||||||
def test_call_any_callable(self):
|
|
||||||
provider = providers.Selector(
|
|
||||||
functools.partial(next, itertools.cycle(["one", "two"])),
|
|
||||||
one=providers.Object(1),
|
|
||||||
two=providers.Object(2),
|
|
||||||
)
|
|
||||||
|
|
||||||
assert provider() == 1
|
|
||||||
assert provider() == 2
|
|
||||||
assert provider() == 1
|
|
||||||
assert provider() == 2
|
|
||||||
|
|
||||||
def test_call_with_context_args(self):
|
|
||||||
provider = providers.Selector(
|
|
||||||
self.selector,
|
|
||||||
one=providers.Callable(lambda *args, **kwargs: (args, kwargs)),
|
|
||||||
)
|
|
||||||
|
|
||||||
with self.selector.override("one"):
|
|
||||||
args, kwargs = provider(1, 2, three=3, four=4)
|
|
||||||
|
|
||||||
assert args == (1, 2)
|
|
||||||
assert kwargs == {"three": 3, "four": 4}
|
|
||||||
|
|
||||||
def test_getattr(self):
|
|
||||||
provider_one = providers.Object(1)
|
|
||||||
provider_two = providers.Object(2)
|
|
||||||
|
|
||||||
provider = providers.Selector(
|
|
||||||
self.selector,
|
|
||||||
one=provider_one,
|
|
||||||
two=provider_two,
|
|
||||||
)
|
|
||||||
|
|
||||||
assert provider.one is provider_one
|
|
||||||
assert provider.two is provider_two
|
|
||||||
|
|
||||||
def test_getattr_attribute_error(self):
|
|
||||||
provider_one = providers.Object(1)
|
|
||||||
provider_two = providers.Object(2)
|
|
||||||
|
|
||||||
provider = providers.Selector(
|
|
||||||
self.selector,
|
|
||||||
one=provider_one,
|
|
||||||
two=provider_two,
|
|
||||||
)
|
|
||||||
|
|
||||||
with raises(AttributeError):
|
|
||||||
_ = provider.provider_three
|
|
||||||
|
|
||||||
def test_call_overridden(self):
|
|
||||||
provider = providers.Selector(self.selector, sample=providers.Object(1))
|
|
||||||
overriding_provider1 = providers.Selector(self.selector, sample=providers.Object(2))
|
|
||||||
overriding_provider2 = providers.Selector(self.selector, sample=providers.Object(3))
|
|
||||||
|
|
||||||
provider.override(overriding_provider1)
|
|
||||||
provider.override(overriding_provider2)
|
|
||||||
|
|
||||||
with self.selector.override("sample"):
|
|
||||||
assert provider() == 3
|
|
||||||
|
|
||||||
def test_providers_attribute(self):
|
|
||||||
provider_one = providers.Object(1)
|
|
||||||
provider_two = providers.Object(2)
|
|
||||||
|
|
||||||
provider = providers.Selector(
|
|
||||||
self.selector,
|
|
||||||
one=provider_one,
|
|
||||||
two=provider_two,
|
|
||||||
)
|
|
||||||
assert provider.providers == {"one": provider_one, "two": provider_two}
|
|
||||||
|
|
||||||
def test_deepcopy(self):
|
|
||||||
provider = providers.Selector(self.selector)
|
|
||||||
|
|
||||||
provider_copy = providers.deepcopy(provider)
|
|
||||||
|
|
||||||
assert provider is not provider_copy
|
|
||||||
assert isinstance(provider, providers.Selector)
|
|
||||||
|
|
||||||
def test_deepcopy_from_memo(self):
|
|
||||||
provider = providers.Selector(self.selector)
|
|
||||||
provider_copy_memo = providers.Selector(self.selector)
|
|
||||||
|
|
||||||
provider_copy = providers.deepcopy(
|
|
||||||
provider,
|
|
||||||
memo={id(provider): provider_copy_memo},
|
|
||||||
)
|
|
||||||
|
|
||||||
assert provider_copy is provider_copy_memo
|
|
||||||
|
|
||||||
def test_deepcopy_overridden(self):
|
|
||||||
provider = providers.Selector(self.selector)
|
|
||||||
object_provider = providers.Object(object())
|
|
||||||
|
|
||||||
provider.override(object_provider)
|
|
||||||
|
|
||||||
provider_copy = providers.deepcopy(provider)
|
|
||||||
object_provider_copy = provider_copy.overridden[0]
|
|
||||||
|
|
||||||
assert provider is not provider_copy
|
|
||||||
assert isinstance(provider, providers.Selector)
|
|
||||||
|
|
||||||
assert object_provider is not object_provider_copy
|
|
||||||
assert isinstance(object_provider_copy, providers.Object)
|
|
||||||
|
|
||||||
def test_deepcopy_with_sys_streams(self):
|
|
||||||
provider = providers.Selector(
|
|
||||||
self.selector,
|
|
||||||
stdin=providers.Object(sys.stdin),
|
stdin=providers.Object(sys.stdin),
|
||||||
stdout=providers.Object(sys.stdout),
|
stdout=providers.Object(sys.stdout),
|
||||||
stderr=providers.Object(sys.stderr),
|
stderr=providers.Object(sys.stderr),
|
||||||
|
|
||||||
)
|
)
|
||||||
|
else:
|
||||||
|
raise ValueError("Unknown selector type \"{0}\"".format(selector_type))
|
||||||
|
|
||||||
provider_copy = providers.deepcopy(provider)
|
|
||||||
|
|
||||||
assert provider is not provider_copy
|
def test_is_provider(selector):
|
||||||
assert isinstance(provider_copy, providers.Selector)
|
assert providers.is_provider(selector) is True
|
||||||
|
|
||||||
with self.selector.override("stdin"):
|
|
||||||
assert provider() is sys.stdin
|
|
||||||
|
|
||||||
with self.selector.override("stdout"):
|
@mark.parametrize("selector_type", ["empty"])
|
||||||
assert provider() is sys.stdout
|
def test_init_optional(selector, switch, one, two):
|
||||||
|
selector.set_selector(switch)
|
||||||
|
selector.set_providers(one=one, two=two)
|
||||||
|
|
||||||
with self.selector.override("stderr"):
|
assert selector.providers == {"one": one, "two": two}
|
||||||
assert provider() is sys.stderr
|
with switch.override("one"):
|
||||||
|
assert selector() == one()
|
||||||
|
with switch.override("two"):
|
||||||
|
assert selector() == two()
|
||||||
|
|
||||||
def test_repr(self):
|
|
||||||
provider = providers.Selector(
|
|
||||||
self.selector,
|
|
||||||
one=providers.Object(1),
|
|
||||||
two=providers.Object(2),
|
|
||||||
)
|
|
||||||
|
|
||||||
assert "<dependency_injector.providers.Selector({0}".format(repr(self.selector)) in repr(provider)
|
def test_set_selector_returns_self(selector, switch):
|
||||||
assert "one={0}".format(repr(provider.one)) in repr(provider)
|
assert selector.set_selector(switch) is selector
|
||||||
assert "two={0}".format(repr(provider.two)) in repr(provider)
|
|
||||||
assert "at {0}".format(hex(id(provider))) in repr(provider)
|
|
||||||
|
def test_set_providers_returns_self(selector, one):
|
||||||
|
assert selector.set_providers(one=one) is selector
|
||||||
|
|
||||||
|
|
||||||
|
def test_provided_instance_provider(selector):
|
||||||
|
assert isinstance(selector.provided, providers.ProvidedInstance)
|
||||||
|
|
||||||
|
|
||||||
|
def test_call(selector, switch):
|
||||||
|
with switch.override("one"):
|
||||||
|
assert selector() == 1
|
||||||
|
|
||||||
|
with switch.override("two"):
|
||||||
|
assert selector() == 2
|
||||||
|
|
||||||
|
|
||||||
|
def test_call_undefined_provider(selector, switch):
|
||||||
|
with switch.override("three"):
|
||||||
|
with raises(errors.Error):
|
||||||
|
selector()
|
||||||
|
|
||||||
|
|
||||||
|
def test_call_selector_is_none(selector, switch):
|
||||||
|
with switch.override(None):
|
||||||
|
with raises(errors.Error):
|
||||||
|
selector()
|
||||||
|
|
||||||
|
|
||||||
|
@mark.parametrize("selector_type", ["empty"])
|
||||||
|
def test_call_any_callable(selector):
|
||||||
|
selector.set_selector(functools.partial(next, itertools.cycle(["one", "two"])))
|
||||||
|
selector.set_providers(
|
||||||
|
one=providers.Object(1),
|
||||||
|
two=providers.Object(2),
|
||||||
|
)
|
||||||
|
|
||||||
|
assert selector() == 1
|
||||||
|
assert selector() == 2
|
||||||
|
assert selector() == 1
|
||||||
|
assert selector() == 2
|
||||||
|
|
||||||
|
|
||||||
|
@mark.parametrize("selector_type", ["empty"])
|
||||||
|
def test_call_with_context_args(selector, switch):
|
||||||
|
selector.set_selector(switch)
|
||||||
|
selector.set_providers(one=providers.Callable(lambda *args, **kwargs: (args, kwargs)))
|
||||||
|
|
||||||
|
with switch.override("one"):
|
||||||
|
args, kwargs = selector(1, 2, three=3, four=4)
|
||||||
|
|
||||||
|
assert args == (1, 2)
|
||||||
|
assert kwargs == {"three": 3, "four": 4}
|
||||||
|
|
||||||
|
|
||||||
|
def test_getattr(selector, one, two):
|
||||||
|
assert selector.one is one
|
||||||
|
assert selector.two is two
|
||||||
|
|
||||||
|
|
||||||
|
def test_getattr_attribute_error(selector):
|
||||||
|
with raises(AttributeError):
|
||||||
|
_ = selector.provider_three
|
||||||
|
|
||||||
|
|
||||||
|
def test_call_overridden(selector, switch):
|
||||||
|
overriding_provider1 = providers.Selector(switch, one=providers.Object(2))
|
||||||
|
overriding_provider2 = providers.Selector(switch, one=providers.Object(3))
|
||||||
|
|
||||||
|
selector.override(overriding_provider1)
|
||||||
|
selector.override(overriding_provider2)
|
||||||
|
|
||||||
|
with switch.override("one"):
|
||||||
|
assert selector() == 3
|
||||||
|
|
||||||
|
|
||||||
|
def test_providers_attribute(selector, one, two):
|
||||||
|
assert selector.providers == {"one": one, "two": two}
|
||||||
|
|
||||||
|
|
||||||
|
def test_deepcopy(selector):
|
||||||
|
provider_copy = providers.deepcopy(selector)
|
||||||
|
|
||||||
|
assert provider_copy is not selector
|
||||||
|
assert isinstance(selector, providers.Selector)
|
||||||
|
|
||||||
|
assert provider_copy.selector is not selector.selector
|
||||||
|
assert isinstance(provider_copy.selector, providers.Configuration)
|
||||||
|
|
||||||
|
assert provider_copy.one is not selector.one
|
||||||
|
assert isinstance(provider_copy.one, providers.Object)
|
||||||
|
assert provider_copy.one.provides == 1
|
||||||
|
|
||||||
|
assert provider_copy.two is not selector.two
|
||||||
|
assert isinstance(provider_copy.two, providers.Object)
|
||||||
|
assert provider_copy.two.provides == 2
|
||||||
|
|
||||||
|
|
||||||
|
def test_deepcopy_from_memo(selector):
|
||||||
|
provider_copy = providers.deepcopy(
|
||||||
|
selector,
|
||||||
|
memo={id(selector): selector},
|
||||||
|
)
|
||||||
|
assert provider_copy is selector
|
||||||
|
|
||||||
|
|
||||||
|
def test_deepcopy_overridden(selector):
|
||||||
|
object_provider = providers.Object(object())
|
||||||
|
|
||||||
|
selector.override(object_provider)
|
||||||
|
|
||||||
|
provider_copy = providers.deepcopy(selector)
|
||||||
|
object_provider_copy = provider_copy.overridden[0]
|
||||||
|
|
||||||
|
assert selector is not provider_copy
|
||||||
|
assert isinstance(selector, providers.Selector)
|
||||||
|
|
||||||
|
assert object_provider is not object_provider_copy
|
||||||
|
assert isinstance(object_provider_copy, providers.Object)
|
||||||
|
|
||||||
|
|
||||||
|
@mark.parametrize("selector_type", ["sys-streams"])
|
||||||
|
def test_deepcopy_with_sys_streams(selector, switch):
|
||||||
|
provider_copy = providers.deepcopy(selector)
|
||||||
|
|
||||||
|
assert selector is not provider_copy
|
||||||
|
assert isinstance(provider_copy, providers.Selector)
|
||||||
|
|
||||||
|
with switch.override("stdin"):
|
||||||
|
assert selector() is sys.stdin
|
||||||
|
|
||||||
|
with switch.override("stdout"):
|
||||||
|
assert selector() is sys.stdout
|
||||||
|
|
||||||
|
with switch.override("stderr"):
|
||||||
|
assert selector() is sys.stderr
|
||||||
|
|
||||||
|
|
||||||
|
def test_repr(selector, switch):
|
||||||
|
assert "<dependency_injector.providers.Selector({0}".format(repr(switch)) in repr(selector)
|
||||||
|
assert "one={0}".format(repr(selector.one)) in repr(selector)
|
||||||
|
assert "two={0}".format(repr(selector.two)) in repr(selector)
|
||||||
|
assert "at {0}".format(hex(id(selector))) in repr(selector)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue
Block a user