Add tests for async generator

This commit is contained in:
Roman Mogylatov 2020-11-18 14:40:06 -05:00
parent 2d2e3a9f01
commit e9344905fe

View File

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