Add test for POST request with body (#11)

Author: kibertoad

package.json

@@ -20,7 +20,7 @@
   },
   "dependencies": {
     "asynchronous-local-storage": "^1.0.1",
-    "fastify-plugin": "^2.2.0"
+    "fastify-plugin": "^2.3.0"
   },
   "peerDependencies": {
     "fastify": "^3.0.0"
@@ -31,7 +31,7 @@
     "eslint": "^7.6.0",
     "eslint-config-prettier": "^6.11.0",
     "eslint-plugin-prettier": "^3.1.4",
-    "jest": "^26.2.2",
+    "jest": "^26.4.0",
     "prettier": "^2.0.5",
     "tsd": "^0.13.1",
     "typescript": "3.9.7"

test/requestContextPlugin.spec.js

@@ -2,7 +2,7 @@ const fastify = require('fastify')
 const { fastifyRequestContextPlugin } = require('../lib/requestContextPlugin')
 const { TestService } = require('./internal/testService')
 
-function initApp(endpoint) {
+function initAppGet(endpoint) {
   const app = fastify({ logger: true })
   app.register(fastifyRequestContextPlugin)
 
@@ -11,13 +11,22 @@ function initApp(endpoint) {
   return app.ready()
 }
 
+function initAppPost(endpoint) {
+  const app = fastify({ logger: true })
+  app.register(fastifyRequestContextPlugin)
+
+  app.post('/', endpoint)
+
+  return app.ready()
+}
+
 describe('requestContextPlugin', () => {
   let app
   afterEach(() => {
     return app.close()
   })
 
-  it('correctly preserves values within single request', () => {
+  it('correctly preserves values within single GET request', () => {
     expect.assertions(2)
 
     let testService
@@ -50,7 +59,7 @@ describe('requestContextPlugin', () => {
             }
           }
 
-          initApp(route).then((_app) => {
+          initAppGet(route).then((_app) => {
             app = _app
             testService = new TestService(app)
             const response1Promise = app
@@ -89,4 +98,77 @@ describe('requestContextPlugin', () => {
       return promiseRequest2
     })
   })
+
+  it('correctly preserves values within single POST request', () => {
+    expect.assertions(2)
+
+    let testService
+    let responseCounter = 0
+    return new Promise((resolveResponsePromise) => {
+      const promiseRequest2 = new Promise((resolveRequest2Promise) => {
+        const promiseRequest1 = new Promise((resolveRequest1Promise) => {
+          const route = (req, reply) => {
+            function prepareReply() {
+              testService.processRequest(requestId).then(() => {
+                const storedValue = req.requestContext.get('testKey')
+                reply.status(204).send({
+                  storedValue,
+                })
+              })
+            }
+
+            const requestId = Number.parseInt(req.body.requestId)
+            req.requestContext.set('testKey', `testValue${requestId}`)
+
+            // We don't want to read values until both requests wrote their values to see if there is a racing condition
+            if (requestId === 1) {
+              resolveRequest1Promise()
+              return promiseRequest2.then(prepareReply)
+            }
+
+            if (requestId === 2) {
+              resolveRequest2Promise()
+              return promiseRequest1.then(prepareReply)
+            }
+          }
+
+          initAppPost(route).then((_app) => {
+            app = _app
+            testService = new TestService(app)
+            const response1Promise = app
+              .inject()
+              .post('/')
+              .body({ requestId: 1 })
+              .end()
+              .then((response) => {
+                expect(response.json().storedValue).toBe('testValue1')
+                responseCounter++
+                if (responseCounter === 2) {
+                  resolveResponsePromise()
+                }
+              })
+
+            const response2Promise = app
+              .inject()
+              .post('/')
+              .body({ requestId: 2 })
+              .end()
+              .then((response) => {
+                expect(response.json().storedValue).toBe('testValue2')
+                responseCounter++
+                if (responseCounter === 2) {
+                  resolveResponsePromise()
+                }
+              })
+
+            return Promise.all([response1Promise, response2Promise])
+          })
+        })
+
+        return promiseRequest1
+      })
+
+      return promiseRequest2
+    })
+  })
 })