ets-labs/python-dependency-injector
1
1
Fork 0
mirror of https://github.com/ets-labs/python-dependency-injector.git synced 2025-07-04 20:33:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add FactoryAggrefate tests

Browse Source
This commit is contained in:
Roman Mogylatov 2021-01-31 09:37:25 -05:00
parent 79b14cf7c4
commit 171afdad71
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/unit/providers/test_traversal_py3.py
View File

@ -428,3 +428,17 @@ class FactoryTests(unittest.TestCase):
self.assertIn(provider1, all_providers) self.assertIn(provider1, all_providers)
self.assertIn(provider2, all_providers) self.assertIn(provider2, all_providers)
self.assertIn(provider3, all_providers) self.assertIn(provider3, all_providers)
class FactoryAggregateTests(unittest.TestCase):
def test_traverse(self):
factory1 = providers.Factory(dict)
factory2 = providers.Factory(list)
provider = providers.FactoryAggregate(factory1=factory1, factory2=factory2)
all_providers = list(provider.traverse())
self.assertEqual(len(all_providers), 2)
self.assertIn(factory1, all_providers)
self.assertIn(factory2, all_providers)