From 171afdad71332f1089732d14e81d76e432d2f851 Mon Sep 17 00:00:00 2001 From: Roman Mogylatov Date: Sun, 31 Jan 2021 09:37:25 -0500 Subject: [PATCH] Add FactoryAggrefate tests --- tests/unit/providers/test_traversal_py3.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/unit/providers/test_traversal_py3.py b/tests/unit/providers/test_traversal_py3.py index f61baa28..3c1a3a42 100644 --- a/tests/unit/providers/test_traversal_py3.py +++ b/tests/unit/providers/test_traversal_py3.py @@ -428,3 +428,17 @@ class FactoryTests(unittest.TestCase): self.assertIn(provider1, all_providers) self.assertIn(provider2, 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)