OLD | NEW |
| (Empty) |
1 // Copyright 2016 the V8 project authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 // Flags: --harmony-async-await --allow-natives-syntax | |
6 | |
7 function assertEqualsAsync(expected, run, msg) { | |
8 var actual; | |
9 var hadValue = false; | |
10 var hadError = false; | |
11 var promise = run(); | |
12 | |
13 if (typeof promise !== "object" || typeof promise.then !== "function") { | |
14 throw new MjsUnitAssertionError( | |
15 "Expected " + run.toString() + | |
16 " to return a Promise, but it returned " + PrettyPrint(promise)); | |
17 } | |
18 | |
19 promise.then(function(value) { hadValue = true; actual = value; }, | |
20 function(error) { hadError = true; actual = error; }); | |
21 | |
22 assertFalse(hadValue || hadError); | |
23 | |
24 %RunMicrotasks(); | |
25 | |
26 if (hadError) throw actual; | |
27 | |
28 assertTrue( | |
29 hadValue, "Expected '" + run.toString() + "' to produce a value"); | |
30 | |
31 assertEquals(expected, actual, msg); | |
32 }; | |
33 | |
34 // Rename a function so that it can help omit things from stack trace. | |
35 function test(fn) { | |
36 return Object.defineProperty(fn, "name", { | |
37 enumerable: false, | |
38 configurable: true, | |
39 value: "@" + fn.name, | |
40 writable: false | |
41 }); | |
42 } | |
43 | |
44 function getStack(error) { | |
45 var stack = error.stack.split('\n'). | |
46 filter(function(line) { | |
47 return /^\s*at @?[a-zA-Z0-9_]/.test(line); | |
48 }). | |
49 map(line => line.replace(/^\s*at (@?[a-zA-Z0-9_\.\[\]]+)(.*)/, "$1")); | |
50 | |
51 // remove `Promise.then()` invocation by assertEqualsAsync() | |
52 if (stack[2] === "assertEqualsAsync") return []; | |
53 | |
54 return stack.reverse(); | |
55 } | |
56 | |
57 var log = []; | |
58 class FakePromise extends Promise { | |
59 constructor(executor) { | |
60 var stack = getStack(new Error("Getting Callstack")); | |
61 if (stack.length) { | |
62 var first = -1; | |
63 for (var i = 0; i < stack.length; ++i) { | |
64 if (stack[i][0] === '@') { | |
65 first = i; | |
66 break; | |
67 } | |
68 } | |
69 while (first > 0) stack.shift(), --first; | |
70 if (stack.length) { | |
71 log.push("@@Species: [" + stack.join(" > ") + "]"); | |
72 } | |
73 } | |
74 return new Promise(executor); | |
75 } | |
76 }; | |
77 | |
78 Object.defineProperty(Promise, Symbol.species, { | |
79 value: FakePromise, | |
80 configurable: true, | |
81 enumerable: false, | |
82 writable: false | |
83 }); | |
84 | |
85 // Internal `AsyncFunctionAwait` only --- no @@species invocations. | |
86 async function asyncFn() { return await "foo"; } | |
87 assertEqualsAsync("foo", test(function testInternalOnly() { return asyncFn(); }, | |
88 "should not call Promise[@@Species]")); | |
89 assertEquals([], log); | |
90 | |
91 log.length = 0; | |
92 assertEqualsAsync( | |
93 "foo", | |
94 test(function testThenOnReturnedPromise() { | |
95 return asyncFn().then(x => (log.push("Then: " + x), x)); | |
96 }), | |
97 "should call Promise[@@Species] after non-internal Then"); | |
98 assertEquals([ | |
99 "@@Species: [@testThenOnReturnedPromise > Promise.then > FakePromise]", | |
100 "Then: foo" | |
101 ], log); | |
OLD | NEW |