src: fix remaining tests

nodejs · Jul 28, 2023 · 6d26f70 · 6d26f70
1 parent 09e0102
commit 6d26f70
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/async_context_frame.cc b/src/async_context_frame.cc
@@ -67,7 +67,9 @@ AsyncContextFrame::AsyncContextFrame(Environment* env,
 }
 
 Local<Value> AsyncContextFrame::current(Environment* env) {
-  return env->context()->GetContinuationPreservedEmbedderData();
+  return env->isolate()
+      ->GetEnteredOrMicrotaskContext()
+      ->GetContinuationPreservedEmbedderData();
 }
 
 Local<Value> AsyncContextFrame::disable(Environment* env, Local<Value> key) {
@@ -90,7 +92,9 @@ Local<Value> AsyncContextFrame::disable(Environment* env, Local<Value> key) {
 // but sometimes (such as enterWith) a direct exchange is needed.
 Local<Value> AsyncContextFrame::exchange(Environment* env, Local<Value> value) {
   auto prior = current(env);
-  env->context()->SetContinuationPreservedEmbedderData(value);
+  env->isolate()
+      ->GetEnteredOrMicrotaskContext()
+      ->SetContinuationPreservedEmbedderData(value);
   return prior;
 }
 

diff --git a/src/async_wrap.cc b/src/async_wrap.cc
@@ -635,6 +635,10 @@ void AsyncWrap::AsyncReset(Local<Object> resource, double execution_async_id,
       UNREACHABLE();
   }
 
+#if defined(NODE_USE_NATIVE_ALS) && NODE_USE_NATIVE_ALS
+  context_frame_.Reset(env()->isolate(), AsyncContextFrame::current(env()));
+#endif
+
   if (silent) return;
 
   EmitAsyncInit(env(), resource,