Add test cases (#10)

* Add test cases Signed-off-by: Matteo Collina <[email protected]> * readme update Signed-off-by: Matteo Collina <[email protected]> --------- Signed-off-by: Matteo Collina <[email protected]>
mcollina · May 18, 2024 · 7cb879b · 7cb879b
1 parent 31c224e
commit 7cb879b
Show file tree

Hide file tree

Showing 2 changed files with 87 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -1,5 +1,8 @@
 # fastify-asyncforge
 
+Provide easy [AsyncLocalStorage](https://nodejs.org/api/async_context.html) magic to your
+[Fastify](https://fastify.dev) apps. It's based on [asyncforge](http://npm.im/asyncforge).
+
 ## Install
 
 ```sh
@@ -18,6 +21,8 @@ const app = fastify({
   logger: true
 })
 
+// It's fundamental that `start` is called before any other plugins are registered, otherwise the helpers
+// would not work as expected
 await start(app)
 
 app.decorate('foo', 'bar')

diff --git a/test/basic.test.js b/test/basic.test.js
@@ -89,3 +89,85 @@ test('fastify.enterWith', async (t) => {
 
   await p.complete
 })
+
+test('with additional onRequest hook', async (t) => {
+  const p = tspl(t, { plan: 15 })
+  const fastify = Fastify()
+
+  await fastifyAsyncForge.start(fastify)
+
+  p.strictEqual(logger(), fastify.log)
+  p.strictEqual(app(), fastify)
+
+  fastify.addHook('onRequest', async function b (_request, _reply) {
+    p.strictEqual(app(), this)
+    p.strictEqual(request(), _request)
+    p.strictEqual(reply(), _reply)
+    p.strictEqual(logger(), _request.log)
+  })
+
+  fastify.get('/', {
+    onRequest: async function a (_request, _reply) {
+      p.strictEqual(app(), this)
+      p.strictEqual(request(), _request)
+      p.strictEqual(reply(), _reply)
+      p.strictEqual(logger(), _request.log)
+    }
+  }, async function (_request, _reply) {
+    p.strictEqual(app(), this)
+    p.strictEqual(request(), _request)
+    p.strictEqual(reply(), _reply)
+    p.strictEqual(logger(), _request.log)
+    return { hello: 'world' }
+  })
+
+  const res = await fastify.inject({
+    method: 'GET',
+    url: '/'
+  })
+
+  p.strictEqual(res.statusCode, 200)
+
+  await p.complete
+})
+
+test('onRequest hook added before registration fails', async (t) => {
+  const p = tspl(t, { plan: 15 })
+  const fastify = Fastify()
+
+  fastify.addHook('onRequest', async function b (_request, _reply) {
+    p.strictEqual(app(), this)
+    p.strictEqual(logger(), fastify.log)
+    p.strictEqual(request(), undefined)
+    p.strictEqual(reply(), undefined)
+  })
+
+  await fastifyAsyncForge.start(fastify)
+
+  p.strictEqual(logger(), fastify.log)
+  p.strictEqual(app(), fastify)
+
+  fastify.get('/', {
+    onRequest: async function a (_request, _reply) {
+      p.strictEqual(app(), this)
+      p.strictEqual(request(), _request)
+      p.strictEqual(reply(), _reply)
+      p.strictEqual(logger(), _request.log)
+    }
+  }, async function (_request, _reply) {
+    p.strictEqual(app(), this)
+    p.strictEqual(request(), _request)
+    p.strictEqual(reply(), _reply)
+    p.strictEqual(logger(), _request.log)
+    return { hello: 'world' }
+  })
+
+  const res = await fastify.inject({
+    method: 'GET',
+    url: '/'
+  })
+
+  p.strictEqual(res.statusCode, 200)
+
+  await p.complete
+})