Support await plan (#8)

jsumners-nr · mcollina · web-flow · commit 03b31d9d387d · 2025-03-18T07:05:00.000-07:00
* Support `await plan`

* be a thenable

Signed-off-by: Matteo Collina &lt;hello@matteocollina.com&gt;

---------

Signed-off-by: Matteo Collina &lt;hello@matteocollina.com&gt;
Co-authored-by: Matteo Collina &lt;hello@matteocollina.com&gt;
diff --git a/test.js b/test.js
@@ -83,3 +83,13 @@ test('wait', async (t) => {
 
   await completed
 })
+
+test('await plan', async (t) => {
+  const plan = tspl(t, { plan: 1 })
+
+  setImmediate(() => {
+    plan.strictEqual(1, 1)
+  })
+
+  await plan
+})
diff --git a/tspl.js b/tspl.js
@@ -45,6 +45,12 @@ function tspl (t, opts = {}) {
     end
   }
 
+  Object.defineProperty(res, 'then', {
+    get () {
+      return completed.then.bind(completed)
+    }
+  })
+
   for (const method of Object.keys(assert)) {
     if (method.match(/^[a-z]/)) {
       res[method] = (...args) => {
