From ae73c0147ffc481206a4e7bc29e0d942cbba511e Mon Sep 17 00:00:00 2001
From: mike12345567 <me@michaeldrury.co.uk>
Date: Mon, 20 Jan 2025 15:41:08 +0000
Subject: [PATCH] Adding test checks.

---
 packages/string-templates/src/environment.ts        | 11 +++++++++++
 packages/string-templates/src/helpers/javascript.ts |  5 ++++-
 packages/string-templates/test/jsLogging.spec.ts    |  7 +++++++
 3 files changed, 22 insertions(+), 1 deletion(-)
 create mode 100644 packages/string-templates/src/environment.ts

diff --git a/packages/string-templates/src/environment.ts b/packages/string-templates/src/environment.ts
new file mode 100644
index 0000000000..ede52591b1
--- /dev/null
+++ b/packages/string-templates/src/environment.ts
@@ -0,0 +1,11 @@
+function isJest() {
+  return (
+    process.env.NODE_ENV === "jest" ||
+    (process.env.JEST_WORKER_ID != null &&
+      process.env.JEST_WORKER_ID !== "null")
+  )
+}
+
+export function isTest() {
+  return isJest()
+}
diff --git a/packages/string-templates/src/helpers/javascript.ts b/packages/string-templates/src/helpers/javascript.ts
index 997ca7b6ec..91f2f9a0ce 100644
--- a/packages/string-templates/src/helpers/javascript.ts
+++ b/packages/string-templates/src/helpers/javascript.ts
@@ -10,6 +10,7 @@ import { iifeWrapper } from "../iife"
 import { JsTimeoutError, UserScriptError } from "../errors"
 import { cloneDeep } from "lodash/fp"
 import { Log, LogType } from "../types"
+import { isTest } from "../environment"
 
 // The method of executing JS scripts depends on the bundle being built.
 // This setter is used in the entrypoint (either index.js or index.mjs).
@@ -126,7 +127,9 @@ export function processJS(handlebars: string, context: any) {
       const jsLineCount = frontendWrapJS(js).split(js)[0].split("\n").length
       const buildLogResponse = (type: LogType) => {
         return (...props: any[]) => {
-          console[type](...props)
+          if (!isTest()) {
+            console[type](...props)
+          }
           props.forEach((prop, index) => {
             if (typeof prop === "object") {
               props[index] = JSON.stringify(prop)
diff --git a/packages/string-templates/test/jsLogging.spec.ts b/packages/string-templates/test/jsLogging.spec.ts
index f328b76c7c..44b3b392ba 100644
--- a/packages/string-templates/test/jsLogging.spec.ts
+++ b/packages/string-templates/test/jsLogging.spec.ts
@@ -43,4 +43,11 @@ describe("Javascript", () => {
     expect(output.logs[0].line).toEqual(1)
     expect(output.logs[0].type).toEqual("warn")
   })
+
+  it("should return the type working with error", () => {
+    const output = processJS(`console.error("error"); return 1`)
+    expect(output.logs[0].log).toEqual(["error"])
+    expect(output.logs[0].line).toEqual(1)
+    expect(output.logs[0].type).toEqual("error")
+  })
 })