@ondc/automation-mock-runner 1.3.54 → 1.3.56

package/dist/lib/MockRunner.d.ts

@@ -123,5 +123,7 @@ export declare class MockRunner {
     static encodeBase64(input: string): string;
     static decodeBase64(encoded: string): string;
     private static resolveBaseActionId;
+    private resolveStep;
+    private allActionIds;
     private static getIdFromSession;
 }

package/dist/lib/MockRunner.js

@@ -162,12 +162,10 @@ class MockRunner {
                 actionId,
                 sessionKeys: Object.keys(sessionData),
             });
-            const step = this.config.steps.find((s) => s.action_id === baseActionId);
+            const step = this.resolveStep(baseActionId);
             if (!step) {
-                const availableActions = this.config.steps.map((s) => s.action_id);
-                throw new errors_1.ActionNotFoundError(actionId, availableActions);
+                throw new errors_1.ActionNotFoundError(actionId, this.allActionIds());
             }
-            const index = this.config.steps.findIndex((s) => s.action_id === baseActionId);
             // Deep clone to avoid mutations
             const defaultPayload = JSON.parse(JSON.stringify(step.mock.defaultPayload));
             const context = this.generateContext(step.action_id, step.api, sessionData);
@@ -248,7 +246,7 @@ class MockRunner {
     async runValidatePayloadWithSession(actionId, targetPayload, sessionData) {
         try {
             const baseActionId = MockRunner.resolveBaseActionId(actionId);
-            const step = this.config.steps.find((s) => s.action_id === baseActionId);
+            const step = this.resolveStep(baseActionId);
             if (!step) {
                 throw new Error(`Action step with ID ${actionId} not found.`);
             }
@@ -300,7 +298,7 @@ class MockRunner {
     async runMeetRequirementsWithSession(actionId, sessionData) {
         try {
             const baseActionId = MockRunner.resolveBaseActionId(actionId);
-            const step = this.config.steps.find((s) => s.action_id === baseActionId);
+            const step = this.resolveStep(baseActionId);
             if (!step) {
                 throw new Error(`Action step with ID ${actionId} not found.`);
             }
@@ -423,7 +421,7 @@ class MockRunner {
         // GENERATED#1#on_search_full_page_gcr
         // get the last by splitting on # and taking the last part
         const baseActionId = MockRunner.resolveBaseActionId(actionId);
-        const step = this.config.steps.find((s) => s.action_id === baseActionId);
+        const step = this.resolveStep(baseActionId);
         // Determine the message_id based on responseFor logic
         let messageId = (0, uuid_1.v4)();
         if (step?.responseFor) {
@@ -635,6 +633,18 @@ class MockRunner {
     static resolveBaseActionId(actionId) {
         return actionId.split("#").slice(-1)[0];
     }
+    // Resolve a step by base action id across main steps then extra steps.
+    resolveStep(baseActionId) {
+        return (this.config.steps.find((s) => s.action_id === baseActionId) ??
+            this.config.extra_steps?.steps.find((s) => s.action_id === baseActionId));
+    }
+    // All known action ids (main + extra), for error messages.
+    allActionIds() {
+        return [
+            ...this.config.steps.map((s) => s.action_id),
+            ...(this.config.extra_steps?.steps.map((s) => s.action_id) ?? []),
+        ];
+    }
     static getIdFromSession(sessionData, key) {
         if (sessionData === undefined) {
             return undefined;

package/dist/lib/configHelper.js

@@ -166,6 +166,17 @@ async function createOptimizedMockConfig(config) {
             },
         };
     }));
+    const optimizedExtraSteps = config.extra_steps ? await Promise.all(config.extra_steps.steps.map(async (step) => {
+        return {
+            ...step,
+            mock: {
+                ...step.mock,
+                generate: await getMinifiedCode(step.mock.generate),
+                validate: await getMinifiedCode(step.mock.validate),
+                requirements: await getMinifiedCode(step.mock.requirements),
+            },
+        };
+    })) : [];
     const optimizedConfig = {
         meta: config.meta,
         transaction_history: [],
@@ -173,6 +184,9 @@ async function createOptimizedMockConfig(config) {
         validationLib: config.validationLib,
         transaction_data: config.transaction_data,
         steps: optimizedSteps,
+        extra_steps: {
+            steps: optimizedExtraSteps,
+        }
     };
     return optimizedConfig;
 }

package/dist/test/MockRunner.test.js

@@ -719,4 +719,58 @@ describe("MockRunner", () => {
             expect(result.success).toBe(true);
         });
     });
+    describe("extra steps support in *WithSession methods", () => {
+        let runner;
+        beforeEach(async () => {
+            const baseConfig = {
+                meta: {
+                    domain: "ONDC:TRV14",
+                    version: "2.0.0",
+                    flowId: "extra-steps-test",
+                },
+                transaction_data: {
+                    transaction_id: "extra-steps-txn-id",
+                    latest_timestamp: "1970-01-01T00:00:00.000Z",
+                },
+                steps: [],
+                transaction_history: [],
+                validationLib: "",
+                helperLib: "",
+            };
+            const base = new MockRunner_1.MockRunner(baseConfig, true);
+            base.getConfig().steps.push(base.getDefaultStep("search", "search_0"));
+            const optimized = await (0, configHelper_1.createOptimizedMockConfig)(base.getConfig());
+            // createOptimizedMockConfig drops extra_steps, so attach an extra step
+            // (absent from main steps) after optimizing the main steps.
+            optimized.extra_steps = {
+                steps: [base.getDefaultStep("on_status", "on_status_0")],
+            };
+            runner = new MockRunner_1.MockRunner(optimized, true);
+        });
+        it("runGeneratePayloadWithSession resolves a step from extra_steps", async () => {
+            const result = await runner.runGeneratePayloadWithSession("on_status_0", {
+                transaction_id: "some-txn",
+            });
+            expect(result.success).toBe(true);
+        });
+        it("runValidatePayloadWithSession resolves a step from extra_steps", async () => {
+            const result = await runner.runValidatePayloadWithSession("on_status_0", { context: {}, message: {} }, {});
+            expect(result.success).toBe(true);
+        });
+        it("runMeetRequirementsWithSession resolves a step from extra_steps", async () => {
+            const result = await runner.runMeetRequirementsWithSession("on_status_0", {});
+            expect(result.success).toBe(true);
+        });
+        it("main steps still resolve (main takes precedence over extras)", async () => {
+            const result = await runner.runGeneratePayloadWithSession("search_0", {
+                transaction_id: "some-txn",
+            });
+            expect(result.success).toBe(true);
+        });
+        it("an action id in neither main nor extra returns failure", async () => {
+            const result = await runner.runGeneratePayloadWithSession("not_a_real_action", {});
+            expect(result.success).toBe(false);
+            expect(result.error?.message).toContain("not_a_real_action");
+        });
+    });
 });

package/package.json

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