Nuxt/packages/nuxt/test/treeshake-client.test.ts

135 lines
4.6 KiB
TypeScript
Raw Normal View History

import { readFileSync } from 'node:fs'
import path from 'node:path'
import { describe, it, expect, vi } from 'vitest'
import * as VueCompilerSFC from 'vue/compiler-sfc'
import type { Plugin } from 'vite'
import { Parser } from 'acorn'
import type { Options } from '@vitejs/plugin-vue'
import _vuePlugin from '@vitejs/plugin-vue'
import { TreeShakeTemplatePlugin } from '../src/components/tree-shake'
import { fixtureDir } from './utils'
vi.mock('node:crypto', () => ({
update: vi.fn().mockReturnThis(),
digest: vi.fn().mockReturnValue('one-hash-to-rule-them-all'),
createHash: vi.fn().mockReturnThis()
}))
// mock due to differences of results between windows and linux
vi.spyOn(path, 'relative').mockImplementation((from: string, to: string) => {
if (to.includes('SomeComponent')) {
return to
}
return path.resolve(from, to)
})
function vuePlugin (options: Options) {
return {
..._vuePlugin(options),
handleHotUpdate () {},
configureDevServer () {}
}
}
const WithClientOnly = readFileSync(path.resolve(fixtureDir, './components/client/WithClientOnlySetup.vue')).toString()
const treeshakeTemplatePlugin = TreeShakeTemplatePlugin.raw({ sourcemap: false, getComponents () { return [] } }, { framework: 'rollup' }) as Plugin
const treeshake = async (source: string): Promise<string> => {
const result = await (treeshakeTemplatePlugin.transform! as Function).call({
parse: (code: string, opts: any = {}) => Parser.parse(code, {
sourceType: 'module',
ecmaVersion: 'latest',
locations: true,
...opts
})
}, source)
return typeof result === 'string' ? result : result?.code
}
async function SFCCompile (name: string, source: string, options: Options, ssr = false): Promise<string> {
const result = await (vuePlugin({
compiler: VueCompilerSFC,
...options
}).transform! as Function).call({
parse: (code: string, opts: any = {}) => Parser.parse(code, {
sourceType: 'module',
ecmaVersion: 'latest',
locations: true,
...opts
})
}, source, name, {
ssr
})
return typeof result === 'string' ? result : result?.code
}
const stateToTest: {name: string, options: Partial<Options & {devServer: {config: {server: any}}}> }[] = [
{
name: 'prod',
options: {
isProduction: true
}
},
{
name: 'dev not inlined',
options: {
isProduction: false,
devServer: {
config: {
// trigger dev behavior
server: false
}
}
}
}
]
describe('treeshake client only in ssr', () => {
vi.spyOn(process, 'cwd').mockImplementation(() => '')
for (const [index, state] of stateToTest.entries()) {
it(`should treeshake ClientOnly correctly in ${state.name}`, async () => {
// add index to avoid using vite vue plugin cache
const clientResult = await SFCCompile(`SomeComponent${index}.vue`, WithClientOnly, state.options)
const ssrResult = await SFCCompile(`SomeComponent${index}.vue`, WithClientOnly, state.options, true)
2023-02-12 19:16:42 +00:00
const treeshaken = await treeshake(ssrResult)
const [_, scopeId] = clientResult.match(/_pushScopeId\("(.*)"\)/)!
// ensure the id is correctly passed between server and client
expect(clientResult).toContain(`pushScopeId("${scopeId}")`)
2023-02-12 19:16:42 +00:00
expect(treeshaken).toContain(`<div ${scopeId}>`)
expect(clientResult).toContain('should-be-treeshaken')
2023-02-12 19:16:42 +00:00
expect(treeshaken).not.toContain('should-be-treeshaken')
2023-02-12 19:16:42 +00:00
expect(treeshaken).not.toContain("import HelloWorld from '../HelloWorld.vue'")
expect(clientResult).toContain("import HelloWorld from '../HelloWorld.vue'")
2023-02-12 19:16:42 +00:00
expect(treeshaken).not.toContain("import { Treeshaken } from 'somepath'")
expect(clientResult).toContain("import { Treeshaken } from 'somepath'")
// remove resolved import
2023-02-12 19:16:42 +00:00
expect(treeshaken).not.toContain('const _component_ResolvedImport =')
expect(clientResult).toContain('const _component_ResolvedImport =')
// expect import of ClientImport to be treeshaken but not Glob since it is also used outside <ClientOnly>
2023-02-12 19:16:42 +00:00
expect(treeshaken).not.toContain('ClientImport')
expect(treeshaken).toContain('import { Glob, } from \'#components\'')
if (state.options.isProduction === false) {
// treeshake at inlined template
2023-02-12 19:16:42 +00:00
expect(treeshaken).not.toContain('ssrRenderComponent($setup["HelloWorld"]')
expect(treeshaken).toContain('ssrRenderComponent($setup["Glob"]')
} else {
// treeshake unref
2023-02-12 19:16:42 +00:00
expect(treeshaken).not.toContain('ssrRenderComponent(_unref(HelloWorld')
expect(treeshaken).toContain('ssrRenderComponent(_unref(Glob')
}
2023-02-12 19:16:42 +00:00
expect(treeshaken).toMatchSnapshot()
})
}
})