ets-labs/python-dependency-injector
1
1
Fork 0
mirror of https://github.com/ets-labs/python-dependency-injector.git synced 2025-11-04 09:57:37 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add tests for async generator

Browse Source
This commit is contained in:
Roman Mogylatov 2020-11-18 14:40:06 -05:00
parent 2d2e3a9f01
commit e9344905fe
1 changed files with 30 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

60
tests/unit/providers/test_resource_py35.py
View File

@ -367,36 +367,36 @@ class AsyncResourceTests(AsyncTestCase):
self.assertEqual(_init.counter, 1)
provider.shutdown()
#
# def test_init_generator(self):
# def _init():
# _init.init_counter += 1
# yield
# _init.shutdown_counter += 1
#
# _init.init_counter = 0
# _init.shutdown_counter = 0
#
# provider = providers.Resource(_init)
#
# result1 = provider()
# self.assertIsNone(result1)
# self.assertEqual(_init.init_counter, 1)
# self.assertEqual(_init.shutdown_counter, 0)
#
# provider.shutdown()
# self.assertEqual(_init.init_counter, 1)
# self.assertEqual(_init.shutdown_counter, 1)
#
# result2 = provider()
# self.assertIsNone(result2)
# self.assertEqual(_init.init_counter, 2)
# self.assertEqual(_init.shutdown_counter, 1)
#
# provider.shutdown()
# self.assertEqual(_init.init_counter, 2)
# self.assertEqual(_init.shutdown_counter, 2)
#
def test_async_init_generator(self):
async def _init():
_init.init_counter += 1
yield
_init.shutdown_counter += 1
_init.init_counter = 0
_init.shutdown_counter = 0
provider = providers.Resource(_init)
result1 = provider()
self.assertIsNone(result1)
self.assertEqual(_init.init_counter, 1)
self.assertEqual(_init.shutdown_counter, 0)
provider.shutdown()
self.assertEqual(_init.init_counter, 1)
self.assertEqual(_init.shutdown_counter, 1)
result2 = provider()
self.assertIsNone(result2)
self.assertEqual(_init.init_counter, 2)
self.assertEqual(_init.shutdown_counter, 1)
provider.shutdown()
self.assertEqual(_init.init_counter, 2)
self.assertEqual(_init.shutdown_counter, 2)
# def test_init_class(self):
# class TestResource(resources.Resource):
# init_counter = 0