From 4970e9bc8c05d3573580b960bc5f59fce0de1e89 Mon Sep 17 00:00:00 2001
From: Damien George <damien.p.george@gmail.com>
Date: Tue, 4 Sep 2018 14:37:30 +1000
Subject: [PATCH] tests/basics: Add test cases for context manager raising in
 enter/exit.

---
 tests/basics/with_raise.py | 44 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 tests/basics/with_raise.py

diff --git a/tests/basics/with_raise.py b/tests/basics/with_raise.py
new file mode 100644
index 0000000000..67eab09b41
--- /dev/null
+++ b/tests/basics/with_raise.py
@@ -0,0 +1,44 @@
+# test with when context manager raises in __enter__/__exit__
+
+class CtxMgr:
+    def __init__(self, id):
+        self.id = id
+
+    def __enter__(self):
+        print("__enter__", self.id)
+        if 10 <= self.id < 20:
+            raise Exception('enter', self.id)
+        return self
+
+    def __exit__(self, a, b, c):
+        print("__exit__", self.id, repr(a), repr(b))
+        if 15 <= self.id < 25:
+            raise Exception('exit', self.id)
+
+# no raising
+try:
+    with CtxMgr(1):
+        pass
+except Exception as e:
+    print(e)
+
+# raise in enter
+try:
+    with CtxMgr(10):
+        pass
+except Exception as e:
+    print(e)
+
+# raise in enter and exit
+try:
+    with CtxMgr(15):
+        pass
+except Exception as e:
+    print(e)
+
+# raise in exit
+try:
+    with CtxMgr(20):
+        pass
+except Exception as e:
+    print(e)