@centeredgesoft/runtime-mock-routes 2.8.2 → 2.9.0

package/build/src/appV2.d.ts

@@ -5,7 +5,7 @@ export declare const isSupportedMethod: (obj: any) => obj is "GET" | "POST" | "P
 export interface RuntimeRequestMethodBody {
     body: any;
     status?: any;
-    headers?: Record<string, string>;
+    headers?: any;
 }
 export interface RequestParameters {
     query: any;

package/build/src/appV2.js

@@ -157,8 +157,13 @@ const appFactory = (runtimeCollection) => {
                 res.status(status);
             }
             if (method.headers) {
-                const processedHeaders = Object.keys(method.headers).reduce((acc, curr) => (Object.assign(Object.assign({}, acc), { [curr]: handlebars_1.default.compile(method.headers[curr])(tokenParams) })), {});
-                res.set(processedHeaders);
+                if (typeof method.headers === 'function') {
+                    res.set(method.headers(tokenParams));
+                }
+                else {
+                    const processedHeaders = Object.keys(method.headers).reduce((acc, curr) => (Object.assign(Object.assign({}, acc), { [curr]: handlebars_1.default.compile(method.headers[curr])(tokenParams) })), {});
+                    res.set(processedHeaders);
+                }
             }
             let { body } = method;
             // if body is a function, execute and then process the response

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@centeredgesoft/runtime-mock-routes",
-  "version": "2.8.2",
+  "version": "2.9.0",
   "description": "",
   "main": "build/index.js",
   "scripts": {

package/src/appV2.test.ts

@@ -370,3 +370,48 @@ describe('Status as a function', () => {
         expect(secondResponse.status).toEqual(400);
     })
 })
+
+describe('Headers as a function', () => {
+    test('is true for headers that are a function', () => {
+        const seed: RuntimeRequestMethodBody = {
+            body: "test",
+            status: 200,
+            headers: () => {
+                return {'test-header': 5}
+            }
+        };
+
+        const valid = isRuntimeRequestMethodBody(seed);
+        expect(valid).toBeTruthy();
+    })
+
+    test('Executes header functions', async () => {
+        const seed: RuntimeRequestCollection = {
+            "/test": {
+                path: "/test",
+                methods: {
+                    GET: {
+                        body: {id: 1},
+                        status: 200,
+                        headers: function(rp: RequestParameters) {
+                            if(rp.query.id == "2") {
+                                return {'test-header': '2'};
+                            } else {
+                                return {'test-header': '3'}
+                            }
+                        }
+                    }
+                }
+            }
+        }
+
+        const app = appFactory(seed);
+        expect(app).toBeDefined();
+
+        var response = await request(app).get('/test?id=2');
+        expect(response.headers['test-header']).toEqual('2');
+
+        var secondResponse = await request(app).get('/test?id=3');
+        expect(secondResponse.headers['test-header']).toEqual('3');
+    })
+})

package/src/appV2.ts

@@ -36,7 +36,7 @@ export const isSupportedMethod = (obj: any): obj is SupportedMethod => Supported
 export interface RuntimeRequestMethodBody {
     body: any;
     status?: any;
-    headers?: Record<string, string>;
+    headers?: any;
 }
 
 export interface RequestParameters {
@@ -205,14 +205,18 @@ export const appFactory = (runtimeCollection?: RuntimeRequestCollection) => {
             }
 
             if (method.headers) {
-                const processedHeaders = Object.keys(method.headers).reduce(
-                    (acc, curr) => ({
-                        ...acc,
-                        [curr]: Handlebars.compile(method.headers[curr])(tokenParams)
-                    }),
-                    {} as Record<string, string>
-                );
-                res.set(processedHeaders);
+                if (typeof method.headers === 'function') {
+                    res.set(method.headers(tokenParams))
+                } else {
+                    const processedHeaders = Object.keys(method.headers).reduce(
+                        (acc, curr) => ({
+                            ...acc,
+                            [curr]: Handlebars.compile(method.headers[curr])(tokenParams)
+                        }),
+                        {} as Record<string, string>
+                    );
+                    res.set(processedHeaders);
+                }
             }
 
             let { body } = method;