From 0aaae89328fb5ab0b9957edb01e9f1ded6704597 Mon Sep 17 00:00:00 2001 From: Roman Mogylatov Date: Sun, 3 Jan 2021 17:38:20 -0500 Subject: [PATCH] Cover shutdown of not initialized resource --- tests/unit/providers/test_resource_py35.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/tests/unit/providers/test_resource_py35.py b/tests/unit/providers/test_resource_py35.py index 0b0fb55a..95d37778 100644 --- a/tests/unit/providers/test_resource_py35.py +++ b/tests/unit/providers/test_resource_py35.py @@ -169,6 +169,15 @@ class ResourceTests(unittest.TestCase): self.assertEqual(_init.init_counter, 2) self.assertEqual(_init.shutdown_counter, 2) + def test_shutdown_of_not_initialized(self): + def _init(): + yield + + provider = providers.Resource(_init) + + result = provider.shutdown() + self.assertIsNone(result) + def test_initialized(self): provider = providers.Resource(init_fn) self.assertFalse(provider.initialized) @@ -461,6 +470,16 @@ class AsyncResourceTest(AsyncTestCase): self.assertEqual(_init.init_counter, 2) self.assertEqual(_init.shutdown_counter, 2) + def test_shutdown_of_not_initialized(self): + async def _init(): + yield + + provider = providers.Resource(_init) + provider.enable_async_mode() + + result = self._run(provider.shutdown()) + self.assertIsNone(result) + def test_concurrent_init(self): resource = object()