@ondc/automation-mock-runner 0.2.2 → 1.0.0

package/dist/lib/MockRunner.d.ts

@@ -20,7 +20,8 @@ export declare class MockRunner {
     runMeetRequirementsWithSession(actionId: string, sessionData: any): Promise<ExecutionResult>;
     getDefaultStep(api: string, actionId: string): MockPlaygroundConfigType["steps"][0];
     generateContext(actionId: string, action: string, sessionData?: any): any;
-    getSessionDataUpToStep(index: number): Record<string, any>;
+    getSessionDataUpToStep(index: number): Promise<Record<string, any>>;
+    static runGetSave(payload: any, expression: string): Promise<ExecutionResult>;
     static encodeBase64(input: string): string;
     static decodeBase64(encoded: string): string;
 }

package/dist/lib/MockRunner.js

@@ -64,7 +64,7 @@ class MockRunner {
             const index = this.config.steps.findIndex((s) => s.action_id === actionId);
             // Deep clone to avoid mutations
             const defaultPayload = JSON.parse(JSON.stringify(step.mock.defaultPayload));
-            const sessionData = this.getSessionDataUpToStep(index);
+            const sessionData = await this.getSessionDataUpToStep(index);
             // Validate inputs against schema if provided
             if (step.mock.inputs?.jsonSchema && Object.keys(inputs).length > 0) {
                 // TODO: Add JSON schema validation for inputs
@@ -160,7 +160,7 @@ class MockRunner {
             }
             const index = this.config.steps.findIndex((s) => s.action_id === actionId);
             const schema = (0, function_registry_1.getFunctionSchema)("validate");
-            const sessionData = this.getSessionDataUpToStep(index);
+            const sessionData = await this.getSessionDataUpToStep(index);
             const result = await this.getRunnerInstance().execute(MockRunner.decodeBase64(step.mock.validate), schema, [targetPayload, sessionData]);
             return result;
         }
@@ -210,7 +210,7 @@ class MockRunner {
             }
             const index = this.config.steps.findIndex((s) => s.action_id === actionId);
             const schema = (0, function_registry_1.getFunctionSchema)("meetsRequirements");
-            const sessionData = this.getSessionDataUpToStep(index);
+            const sessionData = await this.getSessionDataUpToStep(index);
             const result = await this.getRunnerInstance().execute(MockRunner.decodeBase64(step.mock.requirements), schema, [sessionData]);
             return result;
         }
@@ -372,7 +372,7 @@ class MockRunner {
             },
         };
     }
-    getSessionDataUpToStep(index) {
+    async getSessionDataUpToStep(index) {
         const config = this.config;
         if (index < 0 || index > config.steps.length) {
             this.logger.warn("Invalid step index for session data", {
@@ -396,6 +396,10 @@ class MockRunner {
             const saveData = config.steps[i]?.mock.saveData ?? {};
             for (const key in saveData) {
                 const path = saveData[key];
+                const isAppend = key.startsWith("APPEND#");
+                const isEval = key.startsWith("EVAL#");
+                const evalExpression = key.split("#")[2];
+                const actualKey = isAppend ? key.split("#")[1] : key;
                 try {
                     // Validate JSONPath expression
                     if (!path || typeof path !== "string") {
@@ -407,10 +411,10 @@ class MockRunner {
                         continue;
                     }
                     const values = jsonpath_1.default.query(histItem.payload, path);
-                    const value = values[0];
-                    if (value !== undefined) {
-                        sessionData[key] = value;
-                        this.logger.debug("Session data extracted", {
+                    if (isEval && evalExpression) {
+                        // Evaluate the expression using the extracted values
+                        sessionData[actualKey] = await MockRunner.runGetSave(values, evalExpression);
+                        this.logger.debug("Session data extracted via EVAL", {
                             step: i,
                             key,
                             path,
@@ -418,12 +422,25 @@ class MockRunner {
                         });
                     }
                     else {
-                        this.logger.debug("No value found for JSONPath", {
-                            step: i,
-                            key,
-                            path,
-                        });
-                        sessionData[key] = null;
+                        if (values !== undefined) {
+                            sessionData[actualKey] = isAppend
+                                ? (sessionData[actualKey] || []).concat(values)
+                                : values;
+                            this.logger.debug("Session data extracted", {
+                                step: i,
+                                key,
+                                path,
+                                hasValue: true,
+                            });
+                        }
+                        else {
+                            this.logger.debug("No value found for JSONPath", {
+                                step: i,
+                                key,
+                                path,
+                            });
+                            sessionData[actualKey] = null;
+                        }
                     }
                 }
                 catch (error) {
@@ -438,6 +455,12 @@ class MockRunner {
         });
         return sessionData;
     }
+    static async runGetSave(payload, expression) {
+        const evalExpression = MockRunner.decodeBase64(expression);
+        const runner = runner_factory_1.RunnerFactory.createRunner();
+        const schema = (0, function_registry_1.getFunctionSchema)("getSave");
+        return await runner.execute(evalExpression, schema, [payload]);
+    }
     static encodeBase64(input) {
         const bytes = new TextEncoder().encode(input);
         return btoa(String.fromCharCode(...bytes));

package/dist/lib/constants/function-registry.d.ts

@@ -1,4 +1,4 @@
-export type FunctionNamesType = "generate" | "validate" | "meetsRequirements";
+export type FunctionNamesType = "generate" | "validate" | "meetsRequirements" | "getSave";
 export interface FunctionParameter {
     name: string;
     type: string;

package/dist/lib/constants/function-registry.js

@@ -122,6 +122,25 @@ function meetsRequirements(sessionData) {
 ${body}
 }`,
     },
+    getSave: {
+        name: "getSave",
+        parameters: [
+            {
+                name: "payload",
+                type: "Object",
+                description: "The API request or response payload",
+                required: true,
+            },
+        ],
+        returnType: {
+            type: "any",
+            description: "Returns any value",
+        },
+        description: "Evaluates custom code to get save value",
+        timeout: 3000,
+        defaultBody: "return null;",
+        template: (body) => body,
+    },
 };
 function getFunctionSchema(property) {
     const item = exports.FUNCTION_REGISTRY[property];

package/dist/test/MockRunner.test.js

@@ -137,13 +137,29 @@ describe("MockRunner", () => {
         });
     });
     describe("Session Data Management", () => {
-        it("should extract session data correctly", () => {
-            const sessionData = mockRunner.getSessionDataUpToStep(0);
+        it("should extract session data correctly", async () => {
+            const sessionData = await mockRunner.getSessionDataUpToStep(0);
             expect(sessionData).toEqual({});
         });
-        it("should handle invalid step indices", () => {
-            const sessionData = mockRunner.getSessionDataUpToStep(-1);
+        it("should handle invalid step indices", async () => {
+            const sessionData = await mockRunner.getSessionDataUpToStep(-1);
             expect(sessionData).toEqual({});
         });
     });
+    describe("Eval type in saveData", () => {
+        it("should evaluate expressions in saveData correctly", async () => {
+            const fun = MockRunner_1.MockRunner.encodeBase64(`async function getSave(payload){
+				console.log('Payload in getSave:', payload);
+				return payload.a + payload.b;
+			}`);
+            const ob = {
+                a: 1,
+                b: 2,
+            };
+            const res = await MockRunner_1.MockRunner.runGetSave(ob, fun);
+            console.log(JSON.stringify(res, null, 2));
+            console.log(Object.keys(ob).length);
+            expect(res.result).toBe(ob.a + ob.b);
+        });
+    });
 });

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@ondc/automation-mock-runner",
-	"version": "0.2.2",
+	"version": "1.0.0",
 	"description": "A TypeScript library for ONDC automation mock runner",
 	"main": "dist/index.js",
 	"types": "dist/index.d.ts",