fix: pass previousRealm to AsyncGeneratorCompleteStep in AsyncGeneratorYield

Author: Elior Nguyen

core/engine/src/builtins/generator/mod.rs

@@ -100,6 +100,9 @@ impl GeneratorContext {
         resume_kind: GeneratorResumeKind,
         context: &mut Context,
     ) -> CompletionRecord {
+        // Capture the caller's realm for AsyncGeneratorYield (spec §27.6.3.8 step 8).
+        let caller_realm = Some(context.realm().clone());
+
         std::mem::swap(&mut context.vm.stack, &mut self.stack);
         let frame = self.call_frame.take().expect("should have a call frame");
         let fp = frame.fp;
@@ -109,6 +112,7 @@ impl GeneratorContext {
         let frame = context.vm.frame_mut();
         frame.fp = fp;
         frame.rp = rp;
+        frame.caller_realm = caller_realm;
         frame.set_exit_early(true);
 
         if let Some(value) = value {

core/engine/src/tests/async_generator.rs

@@ -1,5 +1,5 @@
 use crate::{
-    Context, JsValue, TestAction, builtins::promise::PromiseState, object::JsPromise,
+    Context, JsValue, Source, TestAction, builtins::promise::PromiseState, object::JsPromise,
     run_test_actions,
 };
 use boa_macros::js_str;
@@ -120,3 +120,49 @@ fn return_on_then_queue() {
         TestAction::assert_eq("count", JsValue::from(2)),
     ]);
 }
+
+#[test]
+fn cross_realm_async_generator_yield() {
+    // Exercises AsyncGeneratorYield spec steps 6-8 (previousRealm handling)
+    // by creating a generator in one realm and consuming it from another.
+    let mut context = Context::default();
+
+    let generator_realm = context.create_realm().unwrap();
+
+    let old_realm = context.enter_realm(generator_realm);
+    let generator = context
+        .eval(Source::from_bytes(
+            b"(async function* g() { yield 42; yield 99; })()",
+        ))
+        .unwrap();
+    context.enter_realm(old_realm);
+
+    let next_fn = generator
+        .as_object()
+        .unwrap()
+        .get(js_str!("next"), &mut context)
+        .unwrap();
+
+    let call_next = |ctx: &mut Context| -> JsValue {
+        let result = next_fn
+            .as_callable()
+            .unwrap()
+            .call(&generator, &[], ctx)
+            .unwrap();
+        ctx.run_jobs().unwrap();
+        result
+    };
+
+    assert_promise_iter_value(
+        &call_next(&mut context),
+        &JsValue::from(42),
+        false,
+        &mut context,
+    );
+    assert_promise_iter_value(
+        &call_next(&mut context),
+        &JsValue::from(99),
+        false,
+        &mut context,
+    );
+}

core/engine/src/vm/call_frame/mod.rs

@@ -77,6 +77,10 @@ pub struct CallFrame {
     /// \[\[Realm\]\]
     pub(crate) realm: Realm,
 
+    /// The caller's realm, captured during generator resume.
+    /// Used by `AsyncGeneratorYield` to pass `previousRealm` to `AsyncGeneratorCompleteStep`.
+    pub(crate) caller_realm: Option<Realm>,
+
     // SAFETY: Nothing in `CallFrameFlags` requires tracing, so this is safe.
     #[unsafe_ignore_trace]
     pub(crate) flags: CallFrameFlags,
@@ -154,6 +158,7 @@ impl CallFrame {
             active_runnable,
             environments,
             realm,
+            caller_realm: None,
             flags: CallFrameFlags::empty(),
         }
     }

core/engine/src/vm/opcode/generator/yield_stm.rs

@@ -66,13 +66,19 @@ impl AsyncGeneratorYield {
         let value = context.vm.get_register(value.into());
         let completion = Ok(value.clone());
 
-        // TODO: 6. Assert: The execution context stack has at least two elements.
-        // TODO: 7. Let previousContext be the second to top element of the execution context stack.
-        // TODO: 8. Let previousRealm be previousContext's Realm.
+        // 6. Assert: The execution context stack has at least two elements.
+        // 7. Let previousContext be the second to top element of the execution context stack.
+        // 8. Let previousRealm be previousContext's Realm.
+        let previous_realm = context.vm.frame().caller_realm.clone();
+
         // 9. Perform AsyncGeneratorCompleteStep(generator, completion, false, previousRealm).
-        if let Err(err) =
-            AsyncGenerator::complete_step(&async_generator_object, completion, false, None, context)
-        {
+        if let Err(err) = AsyncGenerator::complete_step(
+            &async_generator_object,
+            completion,
+            false,
+            previous_realm,
+            context,
+        ) {
             return context.handle_error(err);
         }
 
@@ -114,8 +120,9 @@ impl AsyncGeneratorYield {
         //     a. Set generator.[[AsyncGeneratorState]] to suspended-yield.
         r#gen.data_mut().state = AsyncGeneratorState::SuspendedYield;
 
-        //     TODO: b. Remove genContext from the execution context stack and restore the execution context that is at the top of the execution context stack as the running execution context.
-        //     TODO: c. Let callerContext be the running execution context.
+        //     b. Remove genContext from the execution context stack and restore the execution context
+        //        that is at the top of the execution context stack as the running execution context.
+        //     c. Let callerContext be the running execution context.
         //     d. Resume callerContext passing undefined. If genContext is ever resumed again, let resumptionValue be the Completion Record with which it is resumed.
         //     e. Assert: If control reaches here, then genContext is the running execution context again.
         //     f. Return ? AsyncGeneratorUnwrapYieldResumption(resumptionValue).