Python: tests for async iterators/context managers

yoff · yoff · commit 3929e013505a · 2020-08-11T08:10:46.000+02:00
diff --git a/python/ql/test/experimental/dataflow/coverage/classes.py b/python/ql/test/experimental/dataflow/coverage/classes.py
@@ -1271,49 +1271,63 @@ async def atest_await():
 # # coroutine.throw(type[, value[, traceback]])
 # # coroutine.close()
 
-# # 3.4.3. Asynchronous Iterators
-# # object.__aiter__(self)
-# class With_aiter:
+# 3.4.3. Asynchronous Iterators
+# object.__aiter__(self)
+class With_aiter:
 
-#   def __aiter__(self):
-#     OK()
-#     return "" # edit to match type
+  def __aiter__(self):
+    OK()
+    return self
+
+  async def __anext__(self):
+    raise StopAsyncIteration
+
+async def atest_aiter():
+  with_aiter = With_aiter()
+  async for x in with_aiter:
+    pass
+
+# object.__anext__(self)
+class With_anext:
+
+  def __aiter__(self):
+    return self
 
-# def test_aiter():
-#   with_aiter = With_aiter()
-#   aiter(with_aiter) # edit to effect call
+  async def __anext__(self):
+    OK()
+    raise StopAsyncIteration
 
-# # object.__anext__(self)
-# class With_anext:
+async def atest_anext():
+  with_anext = With_anext()
+  async for x in with_anext:
+    pass
 
-#   def __anext__(self):
-#     OK()
-#     return "" # edit to match type
 
-# def test_anext():
-#   with_anext = With_anext()
-#   anext(with_anext) # edit to effect call
+# 3.4.4. Asynchronous Context Managers
+# object.__aenter__(self)
+class With_aenter:
 
+  async def __aenter__(self):
+    OK()
 
-# # 3.4.4. Asynchronous Context Managers
-# # object.__aenter__(self)
-# class With_aenter:
+  async def __aexit__(self, exc_type, exc_value, traceback):
+    pass
 
-#   def __aenter__(self):
-#     OK()
-#     return "" # edit to match type
+async def atest_aenter():
+  with_aenter = With_aenter()
+  async with with_aenter:
+    pass
 
-# def test_aenter():
-#   with_aenter = With_aenter()
-#   aenter(with_aenter) # edit to effect call
+# object.__aexit__(self, exc_type, exc_value, traceback)
+class With_aexit:
 
-# # object.__aexit__(self, exc_type, exc_value, traceback)
-# class With_aexit:
+  async def __aenter__(self):
+    pass
 
-#   def __aexit__(self, exc_type, exc_value, traceback):
-#     OK()
-#     return "" # edit to match type
+  async def __aexit__(self, exc_type, exc_value, traceback):
+    OK()
 
-# def test_aexit():
-#   with_aexit = With_aexit()
-#   aexit(with_aexit) # edit to effect call
+async def atest_aexit():
+  with_aexit = With_aexit()
+  async with with_aexit:
+    pass
