From c755df9f913170b5b36f61bc5acdb69cfa4713bb Mon Sep 17 00:00:00 2001
From: Martin McKeaveney <martin@shogunsystems.co.uk>
Date: Tue, 9 Mar 2021 12:39:32 +0000
Subject: [PATCH] resourceId tests

---
 packages/server/src/middleware/resourceId.js  |   2 +
 .../src/middleware/tests/resourceId.spec.js   | 105 ++++++++++++++++++
 2 files changed, 107 insertions(+)

diff --git a/packages/server/src/middleware/resourceId.js b/packages/server/src/middleware/resourceId.js
index 4351901dad..4216131119 100644
--- a/packages/server/src/middleware/resourceId.js
+++ b/packages/server/src/middleware/resourceId.js
@@ -36,6 +36,8 @@ class ResourceIdGetter {
   }
 }
 
+module.exports.ResourceIdGetter = ResourceIdGetter
+
 module.exports.paramResource = main => {
   return new ResourceIdGetter("params").mainResource(main).build()
 }
diff --git a/packages/server/src/middleware/tests/resourceId.spec.js b/packages/server/src/middleware/tests/resourceId.spec.js
index e69de29bb2..35e6e5af50 100644
--- a/packages/server/src/middleware/tests/resourceId.spec.js
+++ b/packages/server/src/middleware/tests/resourceId.spec.js
@@ -0,0 +1,105 @@
+const {
+  paramResource,
+  paramSubResource,
+  bodyResource,
+  bodySubResource,
+  ResourceIdGetter
+} = require("../resourceId")
+
+class TestConfiguration {
+  constructor(middleware) {
+    this.middleware = middleware
+    this.ctx = {
+      request: {},
+    }
+    this.next = jest.fn()
+  }
+
+  setParams(params) {
+    this.ctx.params = params
+  }
+
+  setBody(body) {
+    this.ctx.body = body
+  }
+
+  executeMiddleware() {
+    return this.middleware(this.ctx, this.next)
+  }
+}
+
+describe("resourceId middleware", () => {
+  it("calls next() when there is no request object to parse", () => {
+    const config = new TestConfiguration(paramResource("main"))
+
+    config.executeMiddleware()
+
+    expect(config.next).toHaveBeenCalled()
+    expect(config.ctx.resourceId).toBeUndefined()
+  })
+
+  it("generates a resourceId middleware for context query parameters", () => {
+    const config = new TestConfiguration(paramResource("main"))
+    config.setParams({
+      main: "test"
+    })
+
+    config.executeMiddleware()
+
+    expect(config.ctx.resourceId).toEqual("test")
+  })
+
+  it("generates a resourceId middleware for context query sub parameters", () => {
+    const config = new TestConfiguration(paramSubResource("main", "sub"))
+    config.setParams({
+      main: "main",
+      sub: "test"
+    })
+
+    config.executeMiddleware()
+
+    expect(config.ctx.resourceId).toEqual("main")
+    expect(config.ctx.subResourceId).toEqual("test")
+  })
+
+  it("generates a resourceId middleware for context request body", () => {
+    const config = new TestConfiguration(bodyResource("main"))
+    config.setBody({
+      main: "test"
+    })
+
+    config.executeMiddleware()
+
+    expect(config.ctx.resourceId).toEqual("test")
+  })
+
+  it("generates a resourceId middleware for context request body sub fields", () => {
+    const config = new TestConfiguration(bodySubResource("main", "sub"))
+    config.setBody({
+      main: "main",
+      sub: "test"
+    })
+
+    config.executeMiddleware()
+
+    expect(config.ctx.resourceId).toEqual("main")
+    expect(config.ctx.subResourceId).toEqual("test")
+  })
+
+  it("parses resourceIds correctly for custom middlewares", () => {
+    const middleware = new ResourceIdGetter("body")
+      .mainResource("custom")
+      .subResource("customSub")
+      .build()
+    config = new TestConfiguration(middleware)
+    config.setBody({
+      custom: "test",
+      customSub: "subtest"
+    })
+
+    config.executeMiddleware()
+
+    expect(config.ctx.resourceId).toEqual("test")
+    expect(config.ctx.subResourceId).toEqual("subtest")
+  })
+})
\ No newline at end of file