@checksum-ai/runtime 1.0.92 → 1.1.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@checksum-ai/runtime",
-  "version": "1.0.92",
+  "version": "1.1.1",
   "description": "Checksum.ai test runtime",
   "main": "index.js",
   "dependencies": {

package/scripts/patch.js

@@ -27,34 +27,9 @@ function amend(filePath, entryPointText, appendText) {
   fs.writeFileSync(filePath, result, "utf8");
 }
 
-// Replaces the content of the specified line in the file.
-// When "on" is true, the new line is written to the file,
-// otherwise the original line is restored.
-function replaceLine(filePath, lineNumber, originalLine, newLine) {
-  // Read the file content
-  const fileContent = fs.readFileSync(filePath, "utf8");
-
-  // Split the content into an array of lines
-  const lines = fileContent.split(/\r?\n/);
-
-  // Check if the line number is within the range of the file's line count
-  if (lineNumber < 1 || lineNumber > lines.length) {
-    throw new Error("Line number out of range");
-  }
-
-  // Replace the content of the specified line
-  lines[lineNumber - 1] = on ? newLine : originalLine;
-
-  // Join the lines back into a single string
-  const updatedContent = lines.join("\n");
-
-  // Write the modified content back to the file
-  fs.writeFileSync(filePath, updatedContent, "utf8");
-}
-
-// Replaces the content of the specified line in the file.
-// When "on" is true, the new line is written to the file,
-// otherwise the original line is restored.
+// Replaces content.
+// When "on" is true, the new content is written to the file replacing the original content,
+// otherwise the original content is restored.
 function replaceContent(filePath, originalContent, newContent) {
   // Read the file content
   const fileContent = fs.readFileSync(filePath, "utf8");
@@ -62,23 +37,14 @@ function replaceContent(filePath, originalContent, newContent) {
   // add a marker for newContent that can be later recognized for "off" state
   newContent = `/* checksumai */ ${newContent}`;
 
-  // Split the content into an array of lines
-  const lines = fileContent.split(/\r?\n/);
-
-  lines.forEach((line, index) => {
-    if (on) {
-      if (line.includes(originalContent)) {
-        lines[index] = line.replace(originalContent, newContent);
-      }
-    } else {
-      if (line.includes(newContent)) {
-        lines[index] = line.replace(newContent, originalContent);
-      }
-    }
-  });
+  if (on && fileContent.includes(newContent)) {
+    return;
+  }
 
   // Join the lines back into a single string
-  const updatedContent = lines.join("\n");
+  const updatedContent = on
+    ? fileContent.replace(originalContent, newContent)
+    : fileContent.replace(newContent, originalContent);
 
   // Write the modified content back to the file
   fs.writeFileSync(filePath, updatedContent, "utf8");
@@ -133,12 +99,158 @@ function addGenerateSelectorAndLocator(projectRoot) {
   amend(file, entryPointText2, appendText2);
 }
 
+// -------- [Runtime modifications] -------- //
+
+function expect(projectRoot) {
+  const file = join(
+    projectRoot,
+    "node_modules/playwright/lib/matchers/expect.js"
+  );
+  if (!doesFileExist(file)) {
+    return;
+  }
+  let originalContent, newContent;
+
+  originalContent = `return (...args) => {
+      const testInfo = (0, _globals.currentTestInfo)();`;
+  newContent = `return (...args) => {
+      let noSoft = false;
+      if (args[args.length-1]==='no-soft'){
+        noSoft = true;
+        args.pop();
+      }
+      const testInfo = (0, _globals.currentTestInfo)();`;
+  replaceContent(file, originalContent, newContent);
+
+  originalContent = `const rawStack = (0, _utils.captureRawStack)();`;
+  newContent = `const rawStack = (0, _utils.captureRawStack)().filter(s=>!s.includes('@checksum-ai/runtime'));`;
+  replaceContent(file, originalContent, newContent);
+
+  originalContent = `step.complete({
+          error
+        })`;
+  newContent = `step.complete({
+          error,
+          noSoft
+        })`;
+  replaceContent(file, originalContent, newContent);
+
+  originalContent = `if (!this._info.isSoft) throw error;`;
+  newContent = `if (!this._info.isSoft || noSoft) throw error;`;
+  replaceContent(file, originalContent, newContent);
+}
+
+function testInfo(projectRoot) {
+  const file = join(
+    projectRoot,
+    "node_modules/playwright/lib/worker/testInfo.js"
+  );
+  if (!doesFileExist(file)) {
+    return;
+  }
+  let originalContent, newContent;
+  let entryPointText, appendText;
+
+  originalContent = `const rawStack = (0, _utils.captureRawStack)();`;
+  newContent = `let rawStack = (0, _utils.captureRawStack)();`;
+  replaceContent(file, originalContent, newContent);
+
+  entryPointText = `if (!parentStep) parentStep = _utils.zones.zoneData('stepZone', rawStack) || undefined;`;
+  appendText = `\nrawStack = rawStack.filter(s=>!s.includes('@checksum-ai/runtime'));`;
+  amend(file, entryPointText, appendText);
+
+  entryPointText = `data.location = data.location || filteredStack[0];`;
+  appendText = `\nif (this._checksumInternal) {
+      data.location = undefined;
+      this._checksumInternal = false;
+    }
+    if (this._checksumNoLocation){
+      data.location = undefined;
+    }`;
+  amend(file, entryPointText, appendText);
+
+  originalContent = `if (!step.error) {`;
+  newContent = `if (!step.error && !step.preventInfectParentStepsWithError) {`;
+  replaceContent(file, originalContent, newContent);
+
+  originalContent = `_failWithError(error, isHardError) {`;
+  newContent = `addError(error, message) {
+    const serialized = (0, _util.serializeError)(error);
+    serialized.message = [message, serialized.message].join('\\n\\n');
+    serialized.stack = [message, serialized.stack].join('\\n\\n');
+    const step = error[stepSymbol];
+    if (step && step.boxedStack) serialized.stack = \`\${error.name}: \${error.message}\\n\${(0, _utils.stringifyStackFrames)(step.boxedStack).join('\\n')}\`;
+    this.errors.push(serialized);
+  }
+  _failWithError(error, isHardError) {`;
+  replaceContent(file, originalContent, newContent);
+
+  originalContent = `if (step.isSoft && result.error) this._failWithError`;
+  newContent = `if (!result.noSoft && step.isSoft && result.error) this._failWithError`;
+  replaceContent(file, originalContent, newContent);
+}
+
+function testType(projectRoot) {
+  const file = join(
+    projectRoot,
+    "node_modules/playwright/lib/common/testType.js"
+  );
+  if (!doesFileExist(file)) {
+    return;
+  }
+
+  let originalContent, newContent;
+
+  originalContent = `}, async () => {
+      // Make sure that internal "step" is not leaked to the user callback.
+      return await body();
+    });`;
+  newContent = `}, async (step) => {
+      if (options.obtainStep){
+        options.obtainStep(step);
+      }
+      // Make sure that internal "step" is not leaked to the user callback.
+      return await body();
+    });`;
+  replaceContent(file, originalContent, newContent);
+}
+
+function channelOwner(projectRoot) {
+  const file = join(
+    projectRoot,
+    "node_modules/playwright-core/lib/client/channelOwner.js"
+  );
+  if (!doesFileExist(file)) {
+    return;
+  }
+  let originalContent, newContent;
+  let entryPointText, appendText;
+
+  entryPointText = `async _wrapApiCall(func, isInternal = false) {`;
+  appendText = `\nif (this._checksumInternal){
+      isInternal = true;
+    }`;
+  amend(file, entryPointText, appendText);
+
+  originalContent = `const stack = (0, _stackTrace.captureRawStack)();`;
+  newContent = `const stack = (0, _stackTrace.captureRawStack)().filter(s=>!s.includes('@checksum-ai/runtime'));`;
+  replaceContent(file, originalContent, newContent);
+
+  entryPointText = `let apiName = stackTrace.apiName;`;
+  appendText = `\nif (!isInternal && this._checksumTitle){
+      apiName = this._checksumTitle;
+      this._checksumTitle = undefined;
+    }`;
+  amend(file, entryPointText, appendText);
+}
+
 // -------- [Run] -------- //
 
+const isRuntime = true || process.env.RUNTIME === "true";
 const projectRootFromEnv = process.env.PROJECT_ROOT;
 const projectPaths = projectRootFromEnv
   ? [projectRootFromEnv]
-  : ["backend", "lib", "frontend"].map((project) =>
+  : ["backend", "lib", "frontend", "runtime"].map((project) =>
       join(__dirname, "..", project)
     );
 
@@ -147,6 +259,12 @@ for (const projectPath of projectPaths) {
     if (fs.existsSync(projectPath)) {
       alwaysInjectScripts(projectPath);
       addGenerateSelectorAndLocator(projectPath);
+      if (isRuntime) {
+        expect(projectPath);
+        testInfo(projectPath);
+        testType(projectPath);
+        channelOwner(projectPath);
+      }
     }
   } catch (e) {
     console.warn("Unable to patch playwright", projectPath, e);
@@ -162,3 +280,60 @@ for (const projectPath of projectPaths) {
 //     "    await this.extendInjectedScript(consoleApiSource.source);";
 //   replaceLine(file, lineNumber, originalLine, newLine);
 // }
+
+// // Replaces the content of the specified line in the file.
+// // When "on" is true, the new line is written to the file,
+// // otherwise the original line is restored.
+// function replaceLine(filePath, lineNumber, originalLine, newLine) {
+//   // Read the file content
+//   const fileContent = fs.readFileSync(filePath, "utf8");
+
+//   // Split the content into an array of lines
+//   const lines = fileContent.split(/\r?\n/);
+
+//   // Check if the line number is within the range of the file's line count
+//   if (lineNumber < 1 || lineNumber > lines.length) {
+//     throw new Error("Line number out of range");
+//   }
+
+//   // Replace the content of the specified line
+//   lines[lineNumber - 1] = on ? newLine : originalLine;
+
+//   // Join the lines back into a single string
+//   const updatedContent = lines.join("\n");
+
+//   // Write the modified content back to the file
+//   fs.writeFileSync(filePath, updatedContent, "utf8");
+// }
+
+// // Replaces the content of the specified line in the file.
+// // When "on" is true, the new line is written to the file,
+// // otherwise the original line is restored.
+// function replaceContent(filePath, originalContent, newContent) {
+//   // Read the file content
+//   const fileContent = fs.readFileSync(filePath, "utf8");
+
+//   // add a marker for newContent that can be later recognized for "off" state
+//   newContent = `/* checksumai */ ${newContent}`;
+
+//   // Split the content into an array of lines
+//   const lines = fileContent.split(/\r?\n/);
+
+//   lines.forEach((line, index) => {
+//     if (on) {
+//       if (line.includes(originalContent)) {
+//         lines[index] = line.replace(originalContent, newContent);
+//       }
+//     } else {
+//       if (line.includes(newContent)) {
+//         lines[index] = line.replace(newContent, originalContent);
+//       }
+//     }
+//   });
+
+//   // Join the lines back into a single string
+//   const updatedContent = lines.join("\n");
+
+//   // Write the modified content back to the file
+//   fs.writeFileSync(filePath, updatedContent, "utf8");
+// }