fix(browser): esm injector doesn't replace class expressions (#3641)

vitest-dev · Jun 21, 2023 · 5c0ac4a · 5c0ac4a
1 parent 36c06d1
commit 5c0ac4a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/packages/browser/src/node/esmInjector.ts b/packages/browser/src/node/esmInjector.ts
@@ -241,7 +241,7 @@ export function injectVitestModule(code: string, id: string, parse: (code: strin
           s.prependRight(topNode.start, `const ${id.name} = ${binding};\n`)
         }
       }
-      else {
+      else if (parent.type !== 'ClassExpression') {
         s.update(id.start, id.end, binding)
       }
     },

diff --git a/test/core/test/injector-esm.test.ts b/test/core/test/injector-esm.test.ts
@@ -933,3 +933,13 @@ function test() {
     }"
   `)
 })
+
+test('avoid binding ClassExpression', () => {
+  const result = injectSimpleCode(
+    'import Foo, {Bar} from \'./foo\';console.log(Foo, Bar);const obj = {foo: class Foo{}, bar: class Bar{}}',
+  )
+  expect(result).toMatchInlineSnapshot(`
+    "import { __vi_inject__ as __vi_esm_0__ } from './foo'
+    console.log(__vi_esm_0__.default, __vi_esm_0__.Bar);const obj = {foo: class Foo{}, bar: class Bar{}}"
+  `)
+})