From 1a345fc74bdba58eb13d9a098a2e17ec394e8b1f Mon Sep 17 00:00:00 2001
From: Matteo Collina <hello@matteocollina.com>
Date: Sat, 11 May 2024 09:26:22 +0200
Subject: [PATCH] Add test cases

Signed-off-by: Matteo Collina <hello@matteocollina.com>
---
 test/basic.test.js | 82 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 82 insertions(+)

diff --git a/test/basic.test.js b/test/basic.test.js
index 205bca7..df42a8c 100644
--- 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
+})