Add unit tests for REST onResponse hook functionality:Fixes #5042

Author: myrepo

test/unit/helper/json_response_onResponse_test.js

@@ -0,0 +1,68 @@
+const assert = require('assert')
+const REST = require('../../../lib/helper/REST')
+const Container = require('../../../lib/container')
+
+const TestHelper = require('../../support/TestHelper')
+const axios = require('axios')
+const fallBackURL = 'https://jsonplaceholder.typicode.com'
+//start the server using npm run test-server as in the package.json
+let api_url = TestHelper.jsonServerUrl()
+
+describe('REST onResponse Hook Wrapper', () => {
+  let rest
+
+  beforeEach(async () => {
+    Container.helpers({})
+    try {
+      await axios.get(`${api_url}`, { timeout: 1000 }) // Check if the server is reachable
+    } catch (e) {
+      api_url = fallBackURL // Fallback to alternative endpoint
+    }
+
+    rest = new REST({
+      endpoint: api_url,
+      onResponse: res => {
+        res.customFlag = true
+      },
+    })
+
+    rest._before()
+  })
+
+  afterEach(() => {
+    rest = null
+  })
+
+  it('should store response in this.response', async () => {
+    const response = await rest.sendGetRequest('/posts/1')
+    assert.ok(response, 'Expected response to be set on REST instance')
+    assert.equal(response.status, 200)
+  })
+
+  it('should call onResponse function and preserve modifications', async () => {
+    const response = await rest.sendGetRequest('/posts/1')
+    assert.ok(response.customFlag, 'Expected original onResponse to run and modify response')
+  })
+
+  it('should not fail if original onResponse is not set in the config', async () => {
+    const restNoHook = new REST({ endpoint: api_url })
+    restNoHook._before()
+
+    const response = await restNoHook.sendGetRequest('/posts/1')
+    assert.ok(response, 'Expected response to be returned')
+    assert.equal(response.status, 200)
+  })
+
+  it('should not throw if onResponse is not a function in the config', async () => {
+    const restInvalid = new REST({
+      endpoint: api_url,
+      onResponse: undefined,
+    })
+
+    restInvalid._before()
+
+    const response = await restInvalid.sendGetRequest('/posts/1')
+    assert.ok(response)
+    assert.equal(response.status, 200)
+  })
+})