Fix assertion.

oracle · May 9, 2023 · bdc8060 · bdc8060
1 parent 41510ec
commit bdc8060
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.../src/com.oracle.truffle.js/src/com/oracle/truffle/js/nodes/function/FunctionRootNode.java b/.../src/com.oracle.truffle.js/src/com/oracle/truffle/js/nodes/function/FunctionRootNode.java
@@ -87,7 +87,8 @@ public static FunctionRootNode create(AbstractBodyNode body, FrameDescriptor fra
                     SourceSection sourceSection, ScriptOrModule activeScriptOrModule, TruffleString internalFunctionName) {
         FunctionRootNode rootNode = new FunctionRootNode(body, frameDescriptor, functionData, sourceSection, activeScriptOrModule, internalFunctionName);
         if (functionData.getContext().getContextOptions().isTestCloneUninitialized()) {
-            assert JSNodeUtil.hasExactlyOneRootBodyTag(body) : "Function does not have exactly one RootBodyTag";
+            // async/generator functions have the root body tag in the resumption root node.
+            assert JSNodeUtil.hasExactlyOneRootBodyTag(body) || rootNode.getFunctionData().isAsync() || rootNode.getFunctionData().isGenerator() : "Function does not have exactly one RootBodyTag";
             return (FunctionRootNode) rootNode.cloneUninitialized();
         } else {
             return rootNode;