fixup

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

test/issue-7.test.mjs

@@ -1,6 +1,6 @@
 import { before, describe, it } from 'node:test'
 import Fastify from 'fastify'
-import { app, start } from '../index.js'
+import asyncforge, { app, start } from '../index.js'
 import assert from 'node:assert/strict'
 
 let fastify
@@ -14,7 +14,16 @@ async function build (config) {
   return server
 }
 
-describe('support exiting from a context', () => {
+async function buildWithRegister (config) {
+  const server = await Fastify()
+
+  server.register(asyncforge)
+  server.decorate('config', config)
+
+  return server
+}
+
+describe('support exiting from a context / start', () => {
   before(async () => {
     fastify = await build({ foo: 'bar' })
   })
@@ -28,3 +37,18 @@ describe('support exiting from a context', () => {
     assert.equal(app(), fastify)
   })
 })
+
+describe('support exiting from a context / register', () => {
+  before(async () => {
+    fastify = await buildWithRegister({ foo: 'bar' })
+  })
+
+  it('throws', () => {
+    assert.throws(app)
+  })
+
+  it('does not throw using enterWith', () => {
+    fastify.enterWith()
+    assert.equal(app(), fastify)
+  })
+})