Add a test case

dispatchrun · Nov 8, 2023 · 4d48581 · 4d48581
1 parent 2f8a3d2
commit 4d48581
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/compiler/coroutine_test.go b/compiler/coroutine_test.go
@@ -198,6 +198,12 @@ func TestCoroutineYield(t *testing.T) {
 			coro:   func() { VarArgs(3) },
 			yields: []int{0, 1, 2},
 		},
+
+		{
+			name:   "closure yield",
+			coro:   func() { YieldFromClosure(3) },
+			yields: []int{3},
+		},
 	}
 
 	// This emulates the installation of function type information by the

diff --git a/compiler/testdata/coroutine.go b/compiler/testdata/coroutine.go
@@ -547,3 +547,13 @@ func varArgs(args ...int) {
 		coroutine.Yield[int, any](arg)
 	}
 }
+
+func YieldFromClosure(n int) {
+	closure(n)()
+}
+
+func closure(arg int) func() {
+	return func() {
+		coroutine.Yield[int, any](arg)
+	}
+}