automation_model 1.0.775-dev → 1.0.775-stage

package/lib/route.js

@@ -2,264 +2,580 @@ import fs from "fs/promises";
 import path from "path";
 import objectPath from "object-path";
 import { tmpdir } from "os";
-let loadedRoutes = null;
-async function loadRoutes() {
-    if (loadedRoutes !== null)
-        return loadedRoutes;
+import createDebug from "debug";
+import { existsSync } from "fs";
+import { replaceWithLocalTestData } from "./utils.js";
+const debug = createDebug("automation_model:route");
+async function loadRoutes(context, template) {
+    if (context.loadedRoutes instanceof Map && context.loadedRoutes.has(template)) {
+        return context.loadedRoutes.get(template) || [];
+    }
     try {
         let dir = path.join(process.cwd(), "data", "routes");
-        if (process.env.TEMP_RUN) {
+        if (process.env.TEMP_RUN === "true") {
             dir = path.join(tmpdir(), "blinq_temp_routes");
         }
         if (!(await folderExists(dir))) {
-            loadedRoutes = [];
-            return loadedRoutes;
+            context.loadedRoutes = new Map();
+            context.loadedRoutes.set(template, []);
+            return context.loadedRoutes.get(template) || [];
         }
         const files = await fs.readdir(dir);
         const jsonFiles = files.filter((f) => f.endsWith(".json"));
-        const allRoutes = [];
+        const allRoutes = new Map();
         for (const file of jsonFiles) {
-            const content = await fs.readFile(path.join(dir, file), "utf-8");
-            const routeObj = JSON.parse(content);
-            allRoutes.push(routeObj);
+            let content = await fs.readFile(path.join(dir, file), "utf-8");
+            try {
+                const routeObj = JSON.parse(content);
+                const template = routeObj.template;
+                if (!allRoutes.has(template)) {
+                    allRoutes.set(template, []);
+                }
+                allRoutes.get(template)?.push(routeObj);
+            }
+            catch (error) {
+                debug("Error parsing route file:", error);
+                continue;
+            }
         }
-        loadedRoutes = allRoutes;
+        context.loadedRoutes = allRoutes;
+        debug(`Loaded ${allRoutes.size} route definitions from ${dir}`);
     }
     catch (error) {
         console.error("Error loading routes:", error);
-        loadedRoutes = [];
+        context.loadedRoutes = new Map();
+    }
+    return context.loadedRoutes.get(template) || [];
+}
+export function pathFilter(savedPath, actualPath) {
+    if (typeof savedPath !== "string")
+        return false;
+    if (savedPath.includes("*")) {
+        // Escape regex special characters in savedPath
+        const escapedPath = savedPath.replace(/[.+?^${}()|[\]\\]/g, "\\$&");
+        // Treat it as a wildcard
+        const regex = new RegExp(escapedPath.replace(/\*/g, ".*"));
+        return regex.test(actualPath);
+    }
+    else {
+        return savedPath === actualPath;
     }
-    return loadedRoutes;
+}
+export function queryParamsFilter(savedQueryParams, actualQueryParams) {
+    if (!savedQueryParams)
+        return true;
+    for (const [key, value] of Object.entries(savedQueryParams)) {
+        if (value === "*") {
+            // If the saved query param is a wildcard, it matches anything
+            continue;
+        }
+        if (actualQueryParams.get(key) !== value) {
+            return false;
+        }
+    }
+    return true;
+}
+export function methodFilter(savedMethod, actualMethod) {
+    if (!savedMethod)
+        return true;
+    if (savedMethod === "*") {
+        const httpMethodRegex = /^(GET|POST|PUT|DELETE|PATCH|OPTIONS|HEAD)$/;
+        return httpMethodRegex.test(actualMethod);
+    }
+    return savedMethod === actualMethod;
 }
 function matchRoute(routeItem, req) {
+    const debug = createDebug("automation_model:route:matchRoute");
     const url = new URL(req.request().url());
-    const methodMatch = !routeItem.filters.method || routeItem.filters.method === req.request().method();
-    const pathMatch = routeItem.filters.path === url.pathname;
     const queryParams = routeItem.filters.queryParams;
-    const queryMatch = !queryParams || Object.entries(queryParams).every(([key, value]) => url.searchParams.get(key) === value);
-    return methodMatch && pathMatch && queryMatch;
+    const methodMatch = methodFilter(routeItem.filters.method, req.request().method());
+    const pathMatch = pathFilter(routeItem.filters.path, url.pathname);
+    debug("Path match", pathMatch, routeItem.filters.path, url.pathname);
+    const queryParamsMatch = queryParamsFilter(queryParams, url.searchParams);
+    return methodMatch && pathMatch && queryParamsMatch;
+}
+function handleAbortRequest(action, context) {
+    if (context.tracking.timer)
+        clearTimeout(context.tracking.timer);
+    const errorCode = action.config?.errorCode ?? "failed";
+    console.log(`[abort_request] Aborting with error code: ${errorCode}`);
+    context.route.abort(errorCode);
+    context.abortActionPerformed = true;
+    context.tracking.completed = true;
+    return {
+        type: action.type,
+        description: JSON.stringify(action.config),
+        status: "success",
+        message: `Request aborted with code: ${errorCode}`,
+    };
+}
+function handleStatusCodeVerification(action, context) {
+    const isSuccess = String(context.status) === String(action.config);
+    return {
+        type: action.type,
+        description: JSON.stringify(action.config),
+        status: isSuccess ? "success" : "fail",
+        message: `Status code verification ${isSuccess ? "passed" : "failed"}. Expected ${action.config}, got ${context.status}`,
+    };
 }
-let debug = false;
-export async function registerBeforeStepRoutes(context, stepName) {
+function handleJsonModify(action, context) {
+    if (!context.json) {
+        return {
+            type: action.type,
+            description: JSON.stringify(action.config),
+            status: "fail",
+            message: "JSON modification failed. Response is not JSON",
+        };
+    }
+    objectPath.set(context.json, action.config.path, action.config.modifyValue);
+    context.finalBody = JSON.parse(JSON.stringify(context.json));
+    return {
+        type: action.type,
+        description: JSON.stringify(action.config),
+        status: "success",
+        message: `JSON modified at path '${action.config.path}'`,
+    };
+}
+function handleJsonWholeModify(action, context) {
+    if (!context.json) {
+        return {
+            type: action.type,
+            description: JSON.stringify(action.config),
+            status: "fail",
+            message: "JSON modification failed. Response is not JSON",
+        };
+    }
+    try {
+        const parsedConfig = typeof action.config === "string" ? JSON.parse(action.config) : action.config;
+        context.json = parsedConfig;
+        context.finalBody = JSON.parse(JSON.stringify(context.json));
+        return {
+            type: action.type,
+            description: JSON.stringify(action.config),
+            status: "success",
+            message: "Whole JSON body was replaced.",
+        };
+    }
+    catch (e) {
+        const message = `JSON modification failed. Invalid JSON in config: ${e instanceof Error ? e.message : String(e)}`;
+        return { type: action.type, description: JSON.stringify(action.config), status: "fail", message };
+    }
+}
+function handleStatusCodeChange(action, context) {
+    context.status = Number(action.config);
+    return {
+        type: action.type,
+        description: JSON.stringify(action.config),
+        status: "success",
+        message: `Status code changed to ${context.status}`,
+    };
+}
+function handleChangeText(action, context) {
+    if (context.isBinary) {
+        return {
+            type: action.type,
+            description: JSON.stringify(action.config),
+            status: "fail",
+            message: "Change text action failed. Body is not text.",
+        };
+    }
+    context.body = action.config;
+    context.finalBody = context.body;
+    return {
+        type: action.type,
+        description: JSON.stringify(action.config),
+        status: "success",
+        message: "Response body text was replaced.",
+    };
+}
+function handleAssertJson(action, context) {
+    if (!context.json) {
+        return {
+            type: action.type,
+            description: JSON.stringify(action.config),
+            status: "fail",
+            message: "JSON assertion failed. Response is not JSON.",
+        };
+    }
+    const actual = objectPath.get(context.json, action.config.path);
+    const expected = action.config.expectedValue;
+    const isSuccess = JSON.stringify(actual) === JSON.stringify(expected);
+    return {
+        type: action.type,
+        description: JSON.stringify(action.config),
+        status: isSuccess ? "success" : "fail",
+        message: isSuccess
+            ? `JSON assertion passed for path '${action.config.path}'.`
+            : `JSON assertion failed for path '${action.config.path}': expected ${JSON.stringify(expected)}, got ${JSON.stringify(actual)}`,
+    };
+}
+function handleAssertWholeJson(action, context) {
+    if (!context.json) {
+        return {
+            type: action.type,
+            description: JSON.stringify(action.config),
+            status: "fail",
+            message: "Whole JSON assertion failed. Response is not JSON.",
+        };
+    }
+    const originalJSON = JSON.stringify(context.json, null, 2);
+    let isSuccess = false;
+    let message = "";
+    if ("contains" in action.config) {
+        isSuccess = originalJSON.includes(action.config.contains);
+        message = isSuccess
+            ? "Whole JSON assertion passed."
+            : `Whole JSON assertion failed. Expected to contain: "${action.config.contains}".`;
+    }
+    else {
+        isSuccess = originalJSON === action.config.equals;
+        message = isSuccess
+            ? "Whole JSON assertion passed."
+            : `Whole JSON assertion failed. Expected exact match: "${action.config.equals}".`;
+    }
+    return {
+        type: action.type,
+        description: JSON.stringify(action.config),
+        status: isSuccess ? "success" : "fail",
+        message,
+    };
+}
+function handleAssertText(action, context) {
+    if (typeof context.body !== "string") {
+        return {
+            type: action.type,
+            description: JSON.stringify(action.config),
+            status: "fail",
+            message: "Text assertion failed. Body is not text.",
+        };
+    }
+    let isSuccess = false;
+    let message = "";
+    if ("contains" in action.config) {
+        isSuccess = context.body.includes(action.config.contains);
+        message = isSuccess
+            ? "Text assertion passed."
+            : `Text assertion failed. Expected to contain: "${action.config.contains}".`;
+    }
+    else {
+        isSuccess = context.body === action.config.equals;
+        message = isSuccess
+            ? "Text assertion passed."
+            : `Text assertion failed. Expected exact match: "${action.config.equals}".`;
+    }
+    return {
+        type: action.type,
+        description: JSON.stringify(action.config),
+        status: isSuccess ? "success" : "fail",
+        message,
+    };
+}
+function handleStubAction(stubAction, route, tracking) {
+    let actionStatus = "success";
+    const description = JSON.stringify(stubAction.config);
+    const request = route.request();
+    let stubActionPerformed = false;
+    debug(`Stub action found for ${request.url()}. Skipping fetch.`);
+    if (tracking.timer)
+        clearTimeout(tracking.timer);
+    const fullFillConfig = {};
+    if (!tracking.actionResults)
+        tracking.actionResults = [];
+    if (stubAction.config.path) {
+        const filePath = path.join(process.cwd(), "data", "fixtures", stubAction.config.path);
+        debug(`Stub action file path: ${filePath}`);
+        if (existsSync(filePath)) {
+            fullFillConfig.path = filePath;
+            debug(`Stub action fulfilled with file: ${filePath}`);
+        }
+        else {
+            actionStatus = "fail";
+            tracking.actionResults.push({
+                type: "stub_request",
+                description,
+                status: actionStatus,
+                message: `Stub action failed for ${tracking.url}: File not found at ${filePath}`,
+            });
+            stubActionPerformed = true;
+        }
+    }
+    if (!fullFillConfig.path) {
+        if (stubAction.config.statusCode) {
+            fullFillConfig.status = Number(stubAction.config.statusCode);
+        }
+        if (stubAction.config.contentType) {
+            if (stubAction.config.contentType === "application/json") {
+                fullFillConfig.contentType = "application/json";
+                if (stubAction.config.body) {
+                    try {
+                        fullFillConfig.json = JSON.parse(stubAction.config.body);
+                    }
+                    catch (e) {
+                        debug(`Invalid JSON in stub action body: ${stubAction.config.body}, `, e instanceof Error ? e.message : String(e));
+                        debug("Invalid JSON, defaulting to empty object");
+                        fullFillConfig.json = {};
+                    }
+                }
+            }
+            else {
+                fullFillConfig.contentType = stubAction.config.contentType;
+                fullFillConfig.body = stubAction.config.body || "";
+            }
+        }
+        if (!fullFillConfig.json && !fullFillConfig.body) {
+            if (stubAction.config.body) {
+                fullFillConfig.body = stubAction.config.body;
+            }
+        }
+    }
+    if (actionStatus === "success") {
+        try {
+            route.fulfill(fullFillConfig);
+            stubActionPerformed = true;
+            tracking.completed = true;
+            tracking.actionResults.push({
+                type: "stub_request",
+                description,
+                status: actionStatus,
+                message: `Stub action executed for ${request.url()}`,
+            });
+        }
+        catch (e) {
+            actionStatus = "fail";
+            debug(`Failed to fulfill stub request for ${request.url()}`, e);
+            tracking.actionResults.push({
+                type: "stub_request",
+                description,
+                status: actionStatus,
+                message: `Stub action failed for ${request.url()}: ${e instanceof Error ? e.message : String(e)}`,
+            });
+        }
+    }
+    return stubActionPerformed;
+}
+export async function registerBeforeStepRoutes(context, stepName, world) {
+    const debug = createDebug("automation_model:route:registerBeforeStepRoutes");
     const page = context.web.page;
     if (!page)
         throw new Error("context.web.page is missing");
     const stepTemplate = _stepNameToTemplate(stepName);
-    const routes = await loadRoutes();
-    const matchedRouteDefs = routes.filter((r) => r.template === stepTemplate);
-    const allRouteItems = matchedRouteDefs.flatMap((r) => r.routes);
+    debug("stepTemplate", stepTemplate);
+    const routes = await loadRoutes(context, stepTemplate);
+    debug("Routes", routes);
+    const allRouteItems = routes.flatMap((r) => r.routes);
+    debug("All route items", allRouteItems);
     if (!context.__routeState) {
         context.__routeState = { matched: [] };
     }
-    // Pre-register all mandatory routes
-    for (const item of allRouteItems) {
+    for (let i = 0; i < allRouteItems.length; i++) {
+        let item = allRouteItems[i];
+        debug(`Setting up mandatory route with timeout ${item.timeout}ms: ${JSON.stringify(item.filters)}`);
+        let content = JSON.stringify(item);
+        try {
+            content = await replaceWithLocalTestData(content, context.web.world, true, false, content, context.web, false);
+            allRouteItems[i] = JSON.parse(content); // Modify the original array
+            item = allRouteItems[i];
+            debug(`After replacing test data: ${JSON.stringify(allRouteItems[i])}`);
+        }
+        catch (error) {
+            debug("Error replacing test data:", error);
+        }
         if (item.mandatory) {
+            const path = item.filters.path;
+            const queryParams = Object.entries(item.filters.queryParams || {})
+                .map(([key, value]) => `${key}=${value}`)
+                .join("&");
             const tracking = {
                 routeItem: item,
-                url: "",
+                url: `${path}${queryParams ? `?${queryParams}` : ""}`,
                 completed: false,
                 startedAt: Date.now(),
                 actionResults: [],
             };
-            // tracking.timer = setTimeout(() => {
-            //   if (!tracking.completed) {
-            //     console.error(`[MANDATORY] Request to ${item.filters.path} did not complete within ${item.timeout}ms`);
-            //   }
-            // }, item.timeout);
             context.__routeState.matched.push(tracking);
         }
     }
-    page.route("**/*", async (route) => {
-        const request = route.request();
-        // print the url if debug is enabled
-        if (debug) {
-            console.log(`Intercepting request: ${request.method()} ${request.url()}`);
-        }
-        const matchedItem = allRouteItems.find((item) => matchRoute(item, route));
-        if (!matchedItem)
-            return route.continue();
-        if (debug) {
-            console.log(`Matched route item: ${JSON.stringify(matchedItem)}`);
-        }
-        // Find pre-registered tracker
-        let tracking = context.__routeState.matched.find((t) => t.routeItem === matchedItem && !t.completed);
-        // If not mandatory, register dynamically
-        if (!tracking) {
-            tracking = {
-                routeItem: matchedItem,
-                url: request.url(),
-                completed: false,
-                startedAt: Date.now(),
-                actionResults: [],
-            };
-            context.__routeState.matched.push(tracking);
-        }
-        else {
-            tracking.url = request.url();
-        }
-        let response;
-        try {
-            response = await route.fetch();
-        }
-        catch (e) {
-            console.error("Fetch failed for", request.url(), e);
-            if (tracking?.timer)
-                clearTimeout(tracking.timer);
-            return route.abort();
-        }
-        let status = response.status();
-        let headers = response.headers();
-        const isBinary = !headers["content-type"]?.includes("application/json") && !headers["content-type"]?.includes("text");
-        let body;
-        if (isBinary) {
-            body = await response.body(); // returns a Buffer
-        }
-        else {
-            body = await response.text();
-        }
-        let json;
-        try {
-            // check if the body is string
-            if (typeof body === "string") {
-                json = JSON.parse(body);
+    debug("New allrouteItems", JSON.stringify(allRouteItems));
+    let message = null;
+    try {
+        page.route("**/*", async (route) => {
+            const debug = createDebug("automation_model:route:intercept");
+            const request = route.request();
+            debug(`Intercepting request: ${request.method()} ${request.url()}`);
+            debug("All route items", allRouteItems);
+            const matchedItem = allRouteItems.find((item) => matchRoute(item, route));
+            if (!matchedItem)
+                return route.continue();
+            debug(`Matched route item: ${JSON.stringify(matchedItem)}`);
+            debug("Initial context route state", JSON.stringify(context.__routeState, null, 2));
+            let tracking = context.__routeState.matched.find((t) => JSON.stringify(t.routeItem) === JSON.stringify(matchedItem) && !t.completed);
+            debug("Tracking", tracking);
+            let stubActionPerformed = false;
+            if (!tracking) {
+                debug("Tracking not found, creating tracking");
+                tracking = {
+                    routeItem: matchedItem,
+                    url: request.url(),
+                    completed: false,
+                    startedAt: Date.now(),
+                    actionResults: [],
+                };
+                debug("Created tracking", tracking);
+                context.__routeState.matched.push(tracking);
+                debug("Current route state", context.__routeState);
             }
-        }
-        catch (_) { }
-        const actionResults = [];
-        for (const action of matchedItem.actions) {
-            let actionStatus = "success";
-            const description = JSON.stringify(action.config);
-            switch (action.type) {
-                case "abort_request":
+            else {
+                tracking.url = request.url();
+                debug("Updating tracking", tracking);
+            }
+            const stubAction = matchedItem.actions.find((a) => a.type === "stub_request");
+            if (stubAction) {
+                stubActionPerformed = handleStubAction(stubAction, route, tracking);
+            }
+            if (!stubActionPerformed) {
+                let response;
+                try {
+                    response = await route.fetch();
+                }
+                catch (e) {
+                    console.error("Fetch failed for", request.url(), e);
                     if (tracking?.timer)
                         clearTimeout(tracking.timer);
-                    const errorCode = action.config?.errorCode ?? "failed";
-                    console.log(`[abort_request] Aborting  with error code: ${errorCode}`);
-                    await route.abort(errorCode);
-                    tracking.completed = true;
-                    actionResults.push({
-                        type: action.type,
-                        description: JSON.stringify(action.config),
-                        status: "success",
-                    });
-                    return;
-                case "status_code_verification":
-                    if (status !== action.config) {
-                        console.error(`[status_code_verification] Expected ${action.config}, got ${status}`);
-                        actionStatus = "fail";
-                    }
-                    else {
-                        console.log(`[status_code_verification] Passed`);
-                    }
-                    break;
-                case "json_modify":
-                    if (!json) {
-                        console.error(`[json_modify] Response is not JSON`);
-                        actionStatus = "fail";
-                    }
-                    else {
-                        for (const mod of action.config) {
-                            objectPath.set(json, mod.path, mod.value);
-                            console.log(`[json_modify] Modified path ${mod.path} to ${mod.value}`);
-                        }
-                        console.log(`[json_modify] Modified JSON`);
-                    }
-                    break;
-                case "status_code_change":
-                    status = action.config;
-                    console.log(`[status_code_change] Status changed to ${status}`);
-                    break;
-                case "change_text":
-                    if (!headers["content-type"]?.includes("text/html")) {
-                        console.error(`[change_text] Content-Type is not text/html`);
-                        actionStatus = "fail";
-                    }
-                    else {
-                        body = action.config;
-                        console.log(`[change_text] HTML body replaced`);
-                    }
-                    break;
-                case "assert_json":
-                    if (!json) {
-                        console.error(`[assert_json] Response is not JSON`);
-                        actionStatus = "fail";
-                    }
-                    else {
-                        for (const check of action.config) {
-                            const actual = getValue(json, check.path);
-                            const expected = check.expected;
-                            if (JSON.stringify(actual) !== JSON.stringify(expected)) {
-                                console.error(`[assert_json] Path ${check.path}: expected ${expected}, got ${actual}`);
-                                actionStatus = "fail";
-                            }
-                            else {
-                                console.log(`[assert_json] Assertion passed for path ${check.path}`);
-                            }
-                        }
-                    }
-                    break;
-                case "assert_text":
-                    if (typeof body !== "string") {
-                        console.error(`[assert_text] Body is not text`);
-                        actionStatus = "fail";
+                    return route.abort();
+                }
+                const headers = response.headers();
+                const isBinary = !headers["content-type"]?.includes("application/json") && !headers["content-type"]?.includes("text");
+                const body = isBinary ? await response.body() : await response.text();
+                let json;
+                try {
+                    if (typeof body === "string")
+                        json = JSON.parse(body);
+                }
+                catch (_) { }
+                const actionHandlerContext = {
+                    route,
+                    tracking,
+                    status: response.status(),
+                    body,
+                    json,
+                    isBinary,
+                    finalBody: json ?? body,
+                    abortActionPerformed: false,
+                };
+                const actionResults = [];
+                for (const action of matchedItem.actions) {
+                    let result;
+                    switch (action.type) {
+                        case "abort_request":
+                            result = handleAbortRequest(action, actionHandlerContext);
+                            break;
+                        case "status_code_verification":
+                            result = handleStatusCodeVerification(action, actionHandlerContext);
+                            break;
+                        case "json_modify":
+                            result = handleJsonModify(action, actionHandlerContext);
+                            break;
+                        case "json_whole_modify":
+                            result = handleJsonWholeModify(action, actionHandlerContext);
+                            break;
+                        case "status_code_change":
+                            result = handleStatusCodeChange(action, actionHandlerContext);
+                            break;
+                        case "change_text":
+                            result = handleChangeText(action, actionHandlerContext);
+                            break;
+                        case "assert_json":
+                            result = handleAssertJson(action, actionHandlerContext);
+                            break;
+                        case "assert_whole_json":
+                            result = handleAssertWholeJson(action, actionHandlerContext);
+                            break;
+                        case "assert_text":
+                            result = handleAssertText(action, actionHandlerContext);
+                            break;
+                        default:
+                            console.warn(`Unknown action type`);
                     }
-                    else {
-                        if (action.config.contains && !body.includes(action.config.contains)) {
-                            console.error(`[assert_text] Expected to contain: "${action.config.contains}"`);
-                            actionStatus = "fail";
-                        }
-                        else if (action.config.equals && body !== action.config.equals) {
-                            console.error(`[assert_text] Expected exact match`);
-                            actionStatus = "fail";
+                    if (result)
+                        actionResults.push(result);
+                }
+                tracking.completed = true;
+                tracking.actionResults = actionResults;
+                if (tracking.timer)
+                    clearTimeout(tracking.timer);
+                if (!actionHandlerContext.abortActionPerformed) {
+                    try {
+                        const isJSON = headers["content-type"]?.includes("application/json");
+                        if (isJSON) {
+                            await route.fulfill({
+                                status: actionHandlerContext.status,
+                                json: actionHandlerContext.finalBody,
+                                headers,
+                            });
                         }
                         else {
-                            console.log(`[assert_text] Assertion passed`);
+                            await route.fulfill({
+                                status: actionHandlerContext.status,
+                                body: actionHandlerContext.finalBody,
+                                headers,
+                            });
                         }
                     }
-                    break;
-                default:
-                    console.warn(`Unknown action type: ${action.type}`);
+                    catch (e) {
+                        console.error("Failed to fulfill route:", e);
+                    }
+                }
             }
-            actionResults.push({ type: action.type, description, status: actionStatus });
-        }
-        tracking.completed = true;
-        tracking.actionResults = actionResults;
-        if (tracking.timer)
-            clearTimeout(tracking.timer);
-        const responseBody = isBinary ? body : json ? JSON.stringify(json) : body;
-        await route.fulfill({ status, body: responseBody, headers });
-    });
+        });
+    }
+    catch (error) {
+        console.log(JSON.stringify(error));
+    }
 }
-export async function registerAfterStepRoutes(context) {
+export async function registerAfterStepRoutes(context, world) {
     const state = context.__routeState;
+    debug("state in afterStepRoutes", JSON.stringify(state));
     if (!state)
         return [];
     const mandatoryRoutes = state.matched.filter((tracked) => tracked.routeItem.mandatory);
+    debug("mandatoryRoutes in afterStepRoutes", mandatoryRoutes);
     if (mandatoryRoutes.length === 0) {
         context.__routeState = null;
         return [];
     }
     const maxTimeout = Math.max(...mandatoryRoutes.map((r) => r.routeItem.timeout));
     const startTime = Date.now();
+    const mandatoryRouteReached = mandatoryRoutes.map((r) => true);
+    debug("mandatoryRouteReached initialized to", mandatoryRouteReached);
     await new Promise((resolve) => {
         const interval = setInterval(() => {
             const now = Date.now();
             const allCompleted = mandatoryRoutes.every((r) => r.completed);
+            debug("allCompleted in afterStepRoutes", allCompleted);
             const allTimedOut = mandatoryRoutes.every((r) => r.completed || now - startTime >= r.routeItem.timeout);
+            debug("allTimedOut in afterStepRoutes", allTimedOut);
             for (const r of mandatoryRoutes) {
                 const elapsed = now - startTime;
+                // debug(`Elapsed time for route ${r.url}: ${elapsed}ms`);
                 if (!r.completed && elapsed >= r.routeItem.timeout) {
-                    console.error(`[MANDATORY] Request to ${r.routeItem.filters.path} did not complete within ${r.routeItem.timeout}ms (elapsed: ${elapsed})`);
+                    mandatoryRouteReached[mandatoryRoutes.indexOf(r)] = false;
+                    debug(`Route ${r.url} timed out after ${elapsed}ms`);
                 }
             }
             if (allCompleted || allTimedOut) {
+                debug("allCompleted", allCompleted, "allTimedOut", allTimedOut);
                 clearInterval(interval);
                 resolve();
             }
         }, 100);
     });
-    const results = mandatoryRoutes.map((tracked) => {
+    context.results = mandatoryRoutes.map((tracked) => {
         const { routeItem, url, completed, actionResults = [] } = tracked;
+        debug("tracked in afterStepRoutes", {
+            url,
+            completed,
+            actionResults,
+        });
         const actions = actionResults.map((ar) => {
             let status = ar.status;
             if (!completed)
@@ -268,8 +584,10 @@ export async function registerAfterStepRoutes(context) {
                 type: ar.type,
                 description: ar.description,
                 status,
+                message: ar.message || null,
             };
         });
+        debug("actions in afterStepRoutes", actions);
         let overallStatus;
         if (!completed) {
             overallStatus = "timeout";
@@ -294,10 +612,29 @@ export async function registerAfterStepRoutes(context) {
         console.error("Failed to unroute:", e);
     }
     context.__routeState = null;
-    context.routeResults = results;
-    return results;
+    if (context.results && context.results.length > 0) {
+        if (world && world.attach) {
+            await world.attach(JSON.stringify(context.results), "application/json+intercept-results");
+        }
+    }
+    const hasFailed = context.results.some((r) => r.overallStatus === "fail" || r.overallStatus === "timeout");
+    if (hasFailed) {
+        const errorMessage = context.results
+            .filter((r) => r.overallStatus === "fail" || r.overallStatus === "timeout")
+            .map((r) => `Route to ${r.url} failed with status: ${r.overallStatus}`)
+            .join("\n");
+        throw new Error(`Route verification failed:\n${errorMessage}`);
+    }
+    const hasTimedOut = context.results.some((r) => r.overallStatus === "timeout");
+    if (hasTimedOut) {
+        const timeoutMessage = context.results
+            .filter((r) => r.overallStatus === "timeout")
+            .map((r) => `Mandatory Route to ${r.url} timed out after ${r.actions[0]?.description}`)
+            .join("\n");
+        throw new Error(`Mandatory Route verification timed out:\n${timeoutMessage}`);
+    }
+    return context.results;
 }
-// Helper functions
 const toCucumberExpression = (text) => text.replaceAll("/", "\\\\/").replaceAll("(", "\\\\(").replaceAll("{", "\\\\{");
 function extractQuotedText(inputString) {
     const regex = /("[^"]*")/g;