Support await plan (#8)

* Support `await plan`

* be a thenable

Signed-off-by: Matteo Collina <[email protected]>

---------

Signed-off-by: Matteo Collina <[email protected]>
Co-authored-by: Matteo Collina <[email protected]>

Author: jsumners-nr

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
+})

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) => {