From 4c70644c82af694cc43f32c7c5b5114f969efb71 Mon Sep 17 00:00:00 2001 From: Roman Mogylatov Date: Sun, 31 Jan 2021 10:41:53 -0500 Subject: [PATCH] Add ProvidedInstance provider tests --- tests/unit/providers/test_traversal_py3.py | 25 ++++++++++++++++++++++ 1 file changed, 25 insertions(+) diff --git a/tests/unit/providers/test_traversal_py3.py b/tests/unit/providers/test_traversal_py3.py index 1cfa1b4a..bb8a5ccd 100644 --- a/tests/unit/providers/test_traversal_py3.py +++ b/tests/unit/providers/test_traversal_py3.py @@ -727,3 +727,28 @@ class SelectorTests(unittest.TestCase): self.assertIn(provider1, all_providers) self.assertIn(provider2, all_providers) self.assertIn(selector1, all_providers) + + +class ProvidedInstanceTests(unittest.TestCase): + + def test_traverse(self): + provider1 = providers.Provider() + provider = provider1.provided + + all_providers = list(provider.traverse()) + + self.assertEqual(len(all_providers), 1) + self.assertIn(provider1, all_providers) + + def test_traverse_overridden(self): + provider1 = providers.Provider() + provider2 = providers.Provider() + + provider = provider1.provided + provider.override(provider2) + + all_providers = list(provider.traverse()) + + self.assertEqual(len(all_providers), 2) + self.assertIn(provider1, all_providers) + self.assertIn(provider2, all_providers)