Skip to content

Commit b1145de

Browse files
fix detecting codeflash import
1 parent 351dd75 commit b1145de

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

codeflash/languages/javascript/instrument.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -51,6 +51,11 @@ class StandaloneCallMatch:
5151
has_trailing_semicolon: bool
5252

5353

54+
codeflash_import_pattern = re.compile(
55+
r"(import\s+codeflash\s+from\s+['\"]codeflash['\"])|(const\s+codeflash\s*=\s*require\(['\"]codeflash['\"]\))"
56+
)
57+
58+
5459
class StandaloneCallTransformer:
5560
"""Transforms standalone func(...) calls in JavaScript test code.
5661
@@ -730,7 +735,7 @@ def _instrument_js_test_code(
730735
"""
731736
# Add codeflash helper import if not already present
732737
# Support both npm package (codeflash) and legacy local file (codeflash-jest-helper)
733-
has_codeflash_import = "codeflash" in code
738+
has_codeflash_import = codeflash_import_pattern.search(code)
734739
if not has_codeflash_import:
735740
# Detect module system: ESM uses "import ... from", CommonJS uses "require()"
736741
is_esm = bool(re.search(r"^\s*import\s+.+\s+from\s+['\"]", code, re.MULTILINE))

0 commit comments

Comments
 (0)