File tree Expand file tree Collapse file tree
codeflash/languages/javascript Expand file tree Collapse file tree Original file line number Diff line number Diff 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+
5459class 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 ))
You can’t perform that action at this time.
0 commit comments