python-dependency-injector/tests/unit/providers/test_dict_py2_py3.py
2021-09-30 20:09:42 -04:00

232 lines
7.9 KiB
Python

"""Dependency injector dict provider unit tests."""
import sys
import unittest
from dependency_injector import providers
class DictTests(unittest.TestCase):
def test_is_provider(self):
self.assertTrue(providers.is_provider(providers.Dict()))
def test_provided_instance_provider(self):
provider = providers.Dict()
self.assertIsInstance(provider.provided, providers.ProvidedInstance)
def test_init_with_non_string_keys(self):
a1 = object()
a2 = object()
provider = providers.Dict({a1: "i1", a2: "i2"})
dict1 = provider()
dict2 = provider()
self.assertEqual(dict1, {a1: "i1", a2: "i2"})
self.assertEqual(dict2, {a1: "i1", a2: "i2"})
self.assertIsNot(dict1, dict2)
def test_init_with_string_and_non_string_keys(self):
a1 = object()
provider = providers.Dict({a1: "i1"}, a2="i2")
dict1 = provider()
dict2 = provider()
self.assertEqual(dict1, {a1: "i1", "a2": "i2"})
self.assertEqual(dict2, {a1: "i1", "a2": "i2"})
self.assertIsNot(dict1, dict2)
def test_call_with_init_keyword_args(self):
provider = providers.Dict(a1="i1", a2="i2")
dict1 = provider()
dict2 = provider()
self.assertEqual(dict1, {"a1": "i1", "a2": "i2"})
self.assertEqual(dict2, {"a1": "i1", "a2": "i2"})
self.assertIsNot(dict1, dict2)
def test_call_with_context_keyword_args(self):
provider = providers.Dict(a1="i1", a2="i2")
self.assertEqual(
provider(a3="i3", a4="i4"),
{"a1": "i1", "a2": "i2", "a3": "i3", "a4": "i4"},
)
def test_call_with_provider(self):
provider = providers.Dict(
a1=providers.Factory(str, "i1"),
a2=providers.Factory(str, "i2"),
)
self.assertEqual(provider(), {"a1": "i1", "a2": "i2"})
def test_fluent_interface(self):
provider = providers.Dict() \
.add_kwargs(a1="i1", a2="i2")
self.assertEqual(provider(), {"a1": "i1", "a2": "i2"})
def test_add_kwargs(self):
provider = providers.Dict() \
.add_kwargs(a1="i1") \
.add_kwargs(a2="i2")
self.assertEqual(provider.kwargs, {"a1": "i1", "a2": "i2"})
def test_add_kwargs_non_string_keys(self):
a1 = object()
a2 = object()
provider = providers.Dict() \
.add_kwargs({a1: "i1"}) \
.add_kwargs({a2: "i2"})
self.assertEqual(provider.kwargs, {a1: "i1", a2: "i2"})
def test_add_kwargs_string_and_non_string_keys(self):
a2 = object()
provider = providers.Dict() \
.add_kwargs(a1="i1") \
.add_kwargs({a2: "i2"})
self.assertEqual(provider.kwargs, {"a1": "i1", a2: "i2"})
def test_set_kwargs(self):
provider = providers.Dict() \
.add_kwargs(a1="i1", a2="i2") \
.set_kwargs(a3="i3", a4="i4")
self.assertEqual(provider.kwargs, {"a3": "i3", "a4": "i4"})
def test_set_kwargs_non_string_keys(self):
a3 = object()
a4 = object()
provider = providers.Dict() \
.add_kwargs(a1="i1", a2="i2") \
.set_kwargs({a3: "i3", a4: "i4"})
self.assertEqual(provider.kwargs, {a3: "i3", a4: "i4"})
def test_set_kwargs_string_and_non_string_keys(self):
a3 = object()
provider = providers.Dict() \
.add_kwargs(a1="i1", a2="i2") \
.set_kwargs({a3: "i3"}, a4="i4")
self.assertEqual(provider.kwargs, {a3: "i3", "a4": "i4"})
def test_clear_kwargs(self):
provider = providers.Dict() \
.add_kwargs(a1="i1", a2="i2") \
.clear_kwargs()
self.assertEqual(provider.kwargs, {})
def test_call_overridden(self):
provider = providers.Dict(a1="i1", a2="i2")
overriding_provider1 = providers.Dict(a2="i2", a3="i3")
overriding_provider2 = providers.Dict(a3="i3", a4="i4")
provider.override(overriding_provider1)
provider.override(overriding_provider2)
instance1 = provider()
instance2 = provider()
self.assertIsNot(instance1, instance2)
self.assertEqual(instance1, {"a3": "i3", "a4": "i4"})
self.assertEqual(instance2, {"a3": "i3", "a4": "i4"})
def test_deepcopy(self):
provider = providers.Dict(a1="i1", a2="i2")
provider_copy = providers.deepcopy(provider)
self.assertIsNot(provider, provider_copy)
self.assertEqual(provider.kwargs, provider_copy.kwargs)
self.assertIsInstance(provider, providers.Dict)
def test_deepcopy_from_memo(self):
provider = providers.Dict(a1="i1", a2="i2")
provider_copy_memo = providers.Dict(a1="i1", a2="i2")
provider_copy = providers.deepcopy(
provider,
memo={id(provider): provider_copy_memo},
)
self.assertIs(provider_copy, provider_copy_memo)
def test_deepcopy_kwargs(self):
provider = providers.Dict()
dependent_provider1 = providers.Factory(list)
dependent_provider2 = providers.Factory(dict)
provider.add_kwargs(d1=dependent_provider1, d2=dependent_provider2)
provider_copy = providers.deepcopy(provider)
dependent_provider_copy1 = provider_copy.kwargs["d1"]
dependent_provider_copy2 = provider_copy.kwargs["d2"]
self.assertNotEqual(provider.kwargs, provider_copy.kwargs)
self.assertIs(dependent_provider1.cls, dependent_provider_copy1.cls)
self.assertIsNot(dependent_provider1, dependent_provider_copy1)
self.assertIs(dependent_provider2.cls, dependent_provider_copy2.cls)
self.assertIsNot(dependent_provider2, dependent_provider_copy2)
def test_deepcopy_kwargs_non_string_keys(self):
a1 = object()
a2 = object()
dependent_provider1 = providers.Factory(list)
dependent_provider2 = providers.Factory(dict)
provider = providers.Dict({a1: dependent_provider1, a2: dependent_provider2})
provider_copy = providers.deepcopy(provider)
dependent_provider_copy1 = provider_copy.kwargs[a1]
dependent_provider_copy2 = provider_copy.kwargs[a2]
self.assertNotEqual(provider.kwargs, provider_copy.kwargs)
self.assertIs(dependent_provider1.cls, dependent_provider_copy1.cls)
self.assertIsNot(dependent_provider1, dependent_provider_copy1)
self.assertIs(dependent_provider2.cls, dependent_provider_copy2.cls)
self.assertIsNot(dependent_provider2, dependent_provider_copy2)
def test_deepcopy_overridden(self):
provider = providers.Dict()
object_provider = providers.Object(object())
provider.override(object_provider)
provider_copy = providers.deepcopy(provider)
object_provider_copy = provider_copy.overridden[0]
self.assertIsNot(provider, provider_copy)
self.assertEqual(provider.kwargs, provider_copy.kwargs)
self.assertIsInstance(provider, providers.Dict)
self.assertIsNot(object_provider, object_provider_copy)
self.assertIsInstance(object_provider_copy, providers.Object)
def test_deepcopy_with_sys_streams(self):
provider = providers.Dict()
provider.add_kwargs(stdin=sys.stdin, stdout=sys.stdout, stderr=sys.stderr)
provider_copy = providers.deepcopy(provider)
self.assertIsNot(provider, provider_copy)
self.assertIsInstance(provider_copy, providers.Dict)
self.assertIs(provider.kwargs["stdin"], sys.stdin)
self.assertIs(provider.kwargs["stdout"], sys.stdout)
self.assertIs(provider.kwargs["stderr"], sys.stderr)
def test_repr(self):
provider = providers.Dict(a1=1, a2=2)
self.assertEqual(repr(provider),
"<dependency_injector.providers."
"Dict({0}) at {1}>".format(
repr(provider.kwargs),
hex(id(provider))))