Enable test_contextlib

IronLanguages · Jun 21, 2022 · 5ba0513 · 5ba0513
1 parent 2e5bb60
commit 5ba0513
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 12 deletions.
diff --git a/Src/IronPython/Runtime/Exceptions/PythonExceptions.BaseException.cs b/Src/IronPython/Runtime/Exceptions/PythonExceptions.BaseException.cs
@@ -278,9 +278,6 @@ public BaseException? __context__ {
                 get {
                     return _context ?? __cause__;
                 }
-                internal set {
-                    _context = value;
-                }
             }
 
             public bool __suppress_context__ { get; set; }
@@ -363,7 +360,7 @@ protected internal virtual void InitializeFromClr(System.Exception/*!*/ exceptio
 
             internal Exception CreateClrExceptionWithCause(BaseException? cause, BaseException? context, bool suppressContext) {
                 _cause = cause;
-                _context = context;
+                if (context != this) _context = context;
                 __suppress_context__ = suppressContext;
                 _traceback = null;
 

diff --git a/Src/IronPythonTest/Cases/CPythonCasesManifest.ini b/Src/IronPythonTest/Cases/CPythonCasesManifest.ini
@@ -275,10 +275,6 @@ Ignore=true
 Ignore=true
 Reason=ImportError: Cannot import name SemLock
 
-[CPython.test_contextlib]
-Ignore=true
-Reason=Hangs
-
 [CPython.test_copy]
 IsolationLevel=ENGINE
 MaxRecursion=100

diff --git a/Tests/test_exceptions.py b/Tests/test_exceptions.py
@@ -87,11 +87,8 @@ def test_finally_continue_nested_finally_fails(self):
         finally:
             continue
 '''
-        try:
+        with self.assertRaises(SyntaxError):
             compile(t, '<test>', 'exec')
-            self.fail("Should raise SyntaxError")
-        except SyntaxError:
-            pass
 
     def test_bigint_division(self):
         def divide(a, b):
@@ -1046,4 +1043,28 @@ def f():
 
         self.assertRaises(error, f)
 
+    def test_reraise_context(self):
+        # reraising an exception should preserve the context
+        ex1 = Exception(1)
+        try:
+            try:
+                raise ex1
+            except:
+                raise ex1
+        except Exception as e:
+            self.assertIsNone(e.__context__)
+
+        ex1 = Exception(1)
+        ex2 = Exception(2)
+        try:
+            try:
+                try:
+                    raise ex2
+                except:
+                    raise ex1
+            except:
+                raise ex1
+        except Exception as e:
+            self.assertIs(e.__context__, ex2)
+
 run_test(__name__)