From 89c08246ba2988e20a9bba748ef29f297f907f1a Mon Sep 17 00:00:00 2001
From: Noah <33094578+coolreader18@users.noreply.github.com>
Date: Wed, 17 Feb 2021 17:11:39 -0600
Subject: [PATCH] Fix SetupAsyncWith stack level handling

---
 src/compile.rs                                | 17 +++++
 ...pile__tests__nested_double_async_with.snap | 76 +++++++++++++++++++
 2 files changed, 93 insertions(+)
 create mode 100644 src/snapshots/rustpython_compiler_core__compile__tests__nested_double_async_with.snap

diff --git a/src/compile.rs b/src/compile.rs
index 626bec7..e25bd7c 100644
--- a/src/compile.rs
+++ b/src/compile.rs
@@ -2563,6 +2563,23 @@ if True and False and False:
             "\
 if (True and False) or (False and True):
     pass
+"
+        ));
+    }
+
+    #[test]
+    fn test_nested_double_async_with() {
+        assert_dis_snapshot!(compile_exec(
+            "\
+for stop_exc in (StopIteration('spam'), StopAsyncIteration('ham')):
+    with self.subTest(type=type(stop_exc)):
+        try:
+            async with woohoo():
+                raise stop_exc
+        except Exception as ex:
+            self.assertIs(ex, stop_exc)
+        else:
+            self.fail(f'{stop_exc} was suppressed')
 "
         ));
     }
diff --git a/src/snapshots/rustpython_compiler_core__compile__tests__nested_double_async_with.snap b/src/snapshots/rustpython_compiler_core__compile__tests__nested_double_async_with.snap
new file mode 100644
index 0000000..b6f7d64
--- /dev/null
+++ b/src/snapshots/rustpython_compiler_core__compile__tests__nested_double_async_with.snap
@@ -0,0 +1,76 @@
+---
+source: compiler/src/compile.rs
+expression: "compile_exec(\"\\\nfor stop_exc in (StopIteration('spam'), StopAsyncIteration('ham')):\n    with self.subTest(type=type(stop_exc)):\n        try:\n            async with woohoo():\n                raise stop_exc\n        except Exception as ex:\n            self.assertIs(ex, stop_exc)\n        else:\n            self.fail(f'{stop_exc} was suppressed')\n\")"
+---
+                 0 SetupLoop
+                 1 LoadNameAny          (0, StopIteration)
+                 2 LoadConst            ("spam")
+                 3 CallFunctionPositional (1)
+                 4 LoadNameAny          (1, StopAsyncIteration)
+                 5 LoadConst            ("ham")
+                 6 CallFunctionPositional (1)
+                 7 BuildTuple           (2, false)
+                 8 GetIter
+          >>     9 ForIter              (68)
+                10 StoreLocal           (2, stop_exc)
+                11 LoadNameAny          (3, self)
+                12 LoadAttr             (subTest)
+                13 LoadNameAny          (5, type)
+                14 LoadNameAny          (2, stop_exc)
+                15 CallFunctionPositional (1)
+                16 LoadConst            (("type"))
+                17 CallFunctionKeyword  (1)
+                18 SetupWith            (65)
+                19 Pop
+                20 SetupExcept          (40)
+                21 LoadNameAny          (6, woohoo)
+                22 CallFunctionPositional (0)
+                23 BeforeAsyncWith
+                24 GetAwaitable
+                25 LoadConst            (None)
+                26 YieldFrom
+                27 SetupAsyncWith       (33)
+                28 Pop
+                29 LoadNameAny          (2, stop_exc)
+                30 Raise                (Raise)
+                31 PopBlock
+                32 EnterFinally
+          >>    33 WithCleanupStart
+                34 GetAwaitable
+                35 LoadConst            (None)
+                36 YieldFrom
+                37 WithCleanupFinish
+                38 PopBlock
+                39 Jump                 (54)
+          >>    40 Duplicate
+                41 LoadNameAny          (7, Exception)
+                42 CompareOperation     (ExceptionMatch)
+                43 JumpIfFalse          (53)
+                44 StoreLocal           (8, ex)
+                45 LoadNameAny          (3, self)
+                46 LoadAttr             (assertIs)
+                47 LoadNameAny          (8, ex)
+                48 LoadNameAny          (2, stop_exc)
+                49 CallFunctionPositional (2)
+                50 Pop
+                51 PopException
+                52 Jump                 (63)
+          >>    53 Raise                (Reraise)
+          >>    54 LoadNameAny          (3, self)
+                55 LoadAttr             (fail)
+                56 LoadConst            ("")
+                57 LoadNameAny          (2, stop_exc)
+                58 FormatValue          (None)
+                59 LoadConst            (" was suppressed")
+                60 BuildString          (2)
+                61 CallFunctionPositional (1)
+                62 Pop
+          >>    63 PopBlock
+                64 EnterFinally
+          >>    65 WithCleanupStart
+                66 WithCleanupFinish
+                67 Jump                 (9)
+          >>    68 PopBlock
+                69 LoadConst            (None)
+                70 ReturnValue
+