mirror of
				https://github.com/ets-labs/python-dependency-injector.git
				synced 2025-10-31 16:07:51 +03:00 
			
		
		
		
	
		
			
				
	
	
		
			232 lines
		
	
	
		
			7.9 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			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))))
 |