From e9344905fed46e82951fa5170407c95761482258 Mon Sep 17 00:00:00 2001
From: Roman Mogylatov <rmogilatov@gmail.com>
Date: Wed, 18 Nov 2020 14:40:06 -0500
Subject: [PATCH] Add tests for async generator

---
 tests/unit/providers/test_resource_py35.py | 60 +++++++++++-----------
 1 file changed, 30 insertions(+), 30 deletions(-)

diff --git a/tests/unit/providers/test_resource_py35.py b/tests/unit/providers/test_resource_py35.py
index 250377b3..4bfd7b93 100644
--- a/tests/unit/providers/test_resource_py35.py
+++ b/tests/unit/providers/test_resource_py35.py
@@ -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