Add test throwing exception with inner exception in iterator

rickardraysearch · abessen · commit d119c6c2c278 · 2018-06-25T16:28:50.000-04:00
(cherry picked from commit 4956b3e)
diff --git a/src/tests/test_exceptions.py b/src/tests/test_exceptions.py
@@ -350,12 +350,30 @@ def test_iteration_exception():
     from Python.Test import ExceptionTest
     from System import OverflowException
 
-    val = ExceptionTest.ThrowExceptionInIterator().__iter__()
+    exception = OverflowException("error")
+
+    val = ExceptionTest.ThrowExceptionInIterator(exception).__iter__()
+    assert next(val) == 1
+    assert next(val) == 2
+    with pytest.raises(OverflowException) as cm:
+        next(val)
+
+    exc = cm.value
+
+    assert exc == exception
+
+def test_iteration_innerexception():
+    from Python.Test import ExceptionTest
+    from System import OverflowException
+
+    exception = System.Exception("message", OverflowException("error"))
+
+    val = ExceptionTest.ThrowExceptionInIterator(exception).__iter__()
     assert next(val) == 1
     assert next(val) == 2
     with pytest.raises(OverflowException) as cm:
         next(val)
 
-    exc = cm.value 
+    exc = cm.value
 
-    assert isinstance(exc, OverflowException)
+    assert exc == exception.InnerException
