test(router-plugin): split addHmr snapshot testing (#4612)

Author: SeanCassiere

packages/router-plugin/tests/add-hmr.test.ts

@@ -0,0 +1,62 @@
+import { readFile, readdir } from 'node:fs/promises'
+import path from 'node:path'
+import { describe, expect, it } from 'vitest'
+
+import { compileCodeSplitReferenceRoute } from '../src/core/code-splitter/compilers'
+import { defaultCodeSplitGroupings } from '../src/core/constants'
+import { frameworks } from './constants'
+
+function getFrameworkDir(framework: string) {
+  const files = path.resolve(__dirname, `./add-hmr/test-files/${framework}`)
+  const snapshots = path.resolve(__dirname, `./add-hmr/snapshots/${framework}`)
+  return { files, snapshots }
+}
+
+describe('add-hmr works', () => {
+  describe.each(frameworks)('FRAMEWORK=%s', async (framework) => {
+    const dirs = getFrameworkDir(framework)
+    const filenames = await readdir(dirs.files)
+
+    it.each(filenames)(
+      `should add hmr in "reference" for "%s"`,
+      async (filename) => {
+        const file = await readFile(path.join(dirs.files, filename))
+        const code = file.toString()
+
+        const compileResult = compileCodeSplitReferenceRoute({
+          code,
+          filename,
+          id: filename,
+          addHmr: true,
+          codeSplitGroupings: defaultCodeSplitGroupings,
+          targetFramework: framework,
+        })
+
+        await expect(compileResult.code).toMatchFileSnapshot(
+          path.join(dirs.snapshots, filename.replace('.tsx', '@true.tsx')),
+        )
+      },
+    )
+
+    it.each(filenames)(
+      `should NOT add hmr in "reference" for "%s"`,
+      async (filename) => {
+        const file = await readFile(path.join(dirs.files, filename))
+        const code = file.toString()
+
+        const compileResult = compileCodeSplitReferenceRoute({
+          code,
+          filename,
+          id: filename,
+          addHmr: false,
+          codeSplitGroupings: defaultCodeSplitGroupings,
+          targetFramework: framework,
+        })
+
+        await expect(compileResult.code).toMatchFileSnapshot(
+          path.join(dirs.snapshots, filename.replace('.tsx', '@false.tsx')),
+        )
+      },
+    )
+  })
+})