Simplify in-memory test file management by removing UUID references

Author: tech-sushant

src/tools/add-percy-snapshots.ts

@@ -4,22 +4,22 @@ import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { percyWebSetupInstructions } from "../tools/sdk-utils/percy-web/handler.js";
 
 export async function updateTestsWithPercyCommands(args: {
-  uuid: string;
   index: number;
 }): Promise<CallToolResult> {
-  const { uuid, index } = args;
+  const { index } = args;
   const stored = storedPercyResults.get();
-
-  if (!stored || !stored.uuid || stored.uuid !== uuid || !stored[uuid]) {
-    throw new Error(`No test files found in memory for UUID: ${uuid}`);
+  if (!stored || !stored.testFiles) {
+    throw new Error(
+      `No test files found in memory. Please call listTestFiles first.`,
+    );
   }
 
-  const fileStatusMap = stored[uuid];
+  const fileStatusMap = stored.testFiles;
   const filePaths = Object.keys(fileStatusMap);
 
   if (index < 0 || index >= filePaths.length) {
     throw new Error(
-      `Invalid index: ${index}. There are ${filePaths.length} files for UUID: ${uuid}`,
+      `Invalid index: ${index}. There are ${filePaths.length} files available.`,
     );
   }
 
@@ -30,9 +30,8 @@ export async function updateTestsWithPercyCommands(args: {
     percyWebSetupInstructions,
   );
 
-  // Mark this file as updated (true) in the unified structure
   const updatedStored = { ...stored };
-  updatedStored[uuid][filePaths[index]] = true; // true = updated
+  updatedStored.testFiles[filePaths[index]] = true; // true = updated
   storedPercyResults.set(updatedStored);
 
   return {

src/tools/list-test-files.ts

@@ -2,7 +2,6 @@ import { listTestFiles } from "./percy-snapshot-utils/detect-test-files.js";
 import { storedPercyResults } from "../lib/inmemory-store.js";
 import { updateFileAndStep } from "./percy-snapshot-utils/utils.js";
 import { percyWebSetupInstructions } from "./sdk-utils/percy-web/handler.js";
-import crypto from "crypto";
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 
 export async function addListTestFiles(): Promise<CallToolResult> {
@@ -56,30 +55,26 @@ export async function addListTestFiles(): Promise<CallToolResult> {
     };
   }
 
-  // For multiple files, use the UUID workflow
-  const uuid = crypto.randomUUID();
-
-  // Store files in the unified structure with initial status false (not updated)
+  // For multiple files, store directly in testFiles
   const fileStatusMap: { [key: string]: boolean } = {};
   testFiles.forEach((file) => {
     fileStatusMap[file] = false; // false = not updated, true = updated
   });
 
-  // Update storedPercyResults with single UUID for the project
+  // Update storedPercyResults with test files
   const updatedStored = { ...storedResults };
-  updatedStored.uuid = uuid; // Store the UUID reference
-  updatedStored[uuid] = fileStatusMap; // Store files under the UUID key
+  updatedStored.testFiles = fileStatusMap;
   storedPercyResults.set(updatedStored);
 
   return {
     content: [
       {
         type: "text",
-        text: `The Test files are stored in memory with id ${uuid} and the total number of tests files found is ${testFiles.length}. You can use this UUID to retrieve the tests file paths later.`,
+        text: `The Test files are stored in memory and the total number of tests files found is ${testFiles.length}.`,
       },
       {
         type: "text",
-        text: `You can now use the tool addPercySnapshotCommands to update the test file with Percy commands for visual testing with the UUID ${uuid}`,
+        text: `You can now use the tool addPercySnapshotCommands to update the test file with Percy commands for visual testing.`,
       },
     ],
   };

src/tools/percy-snapshot-utils/constants.ts

@@ -3,9 +3,6 @@ import { SDKSupportedLanguage } from "../sdk-utils/common/types.js";
 import { DetectionConfig } from "./types.js";
 
 export const UpdateTestFileWithInstructionsParams = {
-  uuid: z
-    .string()
-    .describe("UUID referencing the in-memory array of test file paths"),
   index: z.number().describe("Index of the test file to update"),
 };
 

src/tools/run-percy-scan.ts

@@ -73,15 +73,14 @@ function checkForUpdatedFiles(
   const projectMatches = stored?.projectName === projectName;
   return (
     projectMatches &&
-    stored?.uuid &&
-    stored[stored.uuid] &&
-    Object.values(stored[stored.uuid]).some((status) => status === true)
+    stored?.testFiles &&
+    Object.values(stored.testFiles).some((status) => status === true)
   );
 }
 
 function getUpdatedFiles(stored: any): string[] {
   const updatedFiles: string[] = [];
-  const fileStatusMap = stored[stored.uuid!];
+  const fileStatusMap = stored.testFiles;
 
   Object.entries(fileStatusMap).forEach(([filePath, status]) => {
     if (status === true) {
@@ -105,7 +104,7 @@ function generateUpdatedFilesSteps(
 
   return [
     `Run only the updated files with Percy:\n` +
-      `Example: ${exampleCommand} -- <file1> <file2> ...`,
+      `Example: ${exampleCommand} <file1> <file2> ...`,
     `Updated files to run:\n${filesToRun.join("\n")}`,
   ];
 }

src/tools/sdk-utils/common/constants.ts

@@ -19,7 +19,7 @@ export const PERCY_REPLACE_REGEX =
   /Invoke listTestFiles\(\) with the provided directories[\s\S]*?- DO NOT STOP until you add commands in all the files or you reach end of the files\./;
 
 export const PERCY_SNAPSHOT_INSTRUCTION = `
-Invoke listTestFiles() with the provided directories from user to gather all test files in memory and obtain the generated UUID ---STEP---
+Invoke listTestFiles() with the provided directories from user to gather all test files in memory ---STEP---
 Process files in STRICT sequential order using tool addPercySnapshotCommands() with below instructions:
 - Start with index 0
 - Then index 1  

src/tools/sdk-utils/handler.ts

@@ -57,7 +57,7 @@ export async function setUpPercyHandler(
       integrationType: input.integrationType,
       folderPaths: input.folderPaths || [],
       filePaths: input.filePaths || [],
-      uuid: null,
+      testFiles: {},
     });
 
     const authorization = getBrowserStackAuth(config);