add pr fix

Author: DenysKuchma

lib/plugin/aiTrace.js

@@ -8,7 +8,7 @@ import Container from '../container.js'
 import recorder from '../recorder.js'
 import event from '../event.js'
 import output from '../output.js'
-import { deleteDir } from '../utils.js'
+import { deleteDir, clearString } from '../utils.js'
 import colors from 'chalk'
 
 const supportedHelpers = Container.STANDARD_ACTING_HELPERS
@@ -97,11 +97,13 @@ export default function (config) {
   })
 
   event.dispatcher.on(event.test.before, test => {
-    const sha256hash = crypto
+    const testTitle = clearString(test.fullTitle()).slice(0, 200)
+    const uniqueHash = crypto
       .createHash('sha256')
       .update(test.file + test.title)
       .digest('hex')
-    dir = path.join(reportDir, `trace_${sha256hash}`)
+      .slice(0, 8)
+    dir = path.join(reportDir, `trace_${testTitle}_${uniqueHash}`)
     mkdirp.sync(dir)
     stepNum = 0
     error = null
@@ -162,6 +164,16 @@ export default function (config) {
       prefix: stepPrefix,
       artifacts: {},
       meta: {},
+      debugOutput: [],
+    }
+
+    if (step.startTime && step.endTime) {
+      stepData.meta.duration = ((step.endTime - step.startTime) / 1000).toFixed(2) + 's'
+    }
+
+    if (config.captureDebugOutput && debugOutput.length > 0) {
+      stepData.debugOutput = [...debugOutput]
+      debugOutput = []
     }
 
     try {
@@ -176,19 +188,27 @@ export default function (config) {
       }
 
       // Save screenshot
-      const screenshotFile = `${stepPrefix}_screenshot.png`
-      await helper.saveScreenshot(path.join(dir, screenshotFile), config.fullPageScreenshots)
-      stepData.artifacts.screenshot = screenshotFile
+      if (!step.artifacts?.screenshot) {
+        const screenshotFile = `${stepPrefix}_screenshot.png`
+        await helper.saveScreenshot(path.join(dir, screenshotFile), config.fullPageScreenshots)
+        stepData.artifacts.screenshot = screenshotFile
+      } else {
+        stepData.artifacts.screenshot = step.artifacts.screenshot
+      }
 
       // Save HTML
       if (config.captureHTML && helper.grabSource) {
-        try {
-          const html = await helper.grabSource()
-          const htmlFile = `${stepPrefix}_page.html`
-          fs.writeFileSync(path.join(dir, htmlFile), html)
-          stepData.artifacts.html = htmlFile
-        } catch (err) {
-          output.debug(`aiTrace: Could not capture HTML: ${err.message}`)
+        if (!step.artifacts?.html) {
+          try {
+            const html = await helper.grabSource()
+            const htmlFile = `${stepPrefix}_page.html`
+            fs.writeFileSync(path.join(dir, htmlFile), html)
+            stepData.artifacts.html = htmlFile
+          } catch (err) {
+            output.debug(`aiTrace: Could not capture HTML: ${err.message}`)
+          }
+        } else {
+          stepData.artifacts.html = step.artifacts.html
         }
       }
 
@@ -257,10 +277,20 @@ export default function (config) {
     steps.forEach((stepData, index) => {
       markdown += `${stepData.step}\n`
 
+      if (stepData.meta.duration) {
+        markdown += `  > duration: ${stepData.meta.duration}\n`
+      }
+
       if (stepData.meta.url) {
         markdown += `  > navigated to ${stepData.meta.url}\n`
       }
 
+      if (config.captureDebugOutput && stepData.debugOutput && stepData.debugOutput.length > 0) {
+        stepData.debugOutput.forEach(line => {
+          markdown += `  > ${line}\n`
+        })
+      }
+
       if (stepData.artifacts.html) {
         markdown += `  > [HTML](./${stepData.artifacts.html})\n`
       }