tuono/packages/lazy-fn-vite-plugin/tests/transpileSource.test.ts

import fs from 'node:fs/promises'
import os from 'node:os'

import { it, expect, describe } from 'vitest'
import type { Plugin } from 'vite'

import { LazyLoadingPlugin } from '../src'

type ViteTransformHandler = Exclude<
  Plugin['transform'],
  // eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
  Function | undefined
>['handler']

// Create a type-safe transform method
function getTransform(): (...args: Parameters<ViteTransformHandler>) => string {
  return LazyLoadingPlugin().transform as never
}

describe('"dynamic" sources', async () => {
  const folderNames = await fs.readdir(`${process.cwd()}/tests/sources`)

  describe.each(folderNames)('%s', async (folderName) => {
    const testDirPath = `${process.cwd()}/tests/sources/${folderName}`

    const sourceRaw = await fs.readFile(`${testDirPath}/source.tsx`, 'utf-8')
    /**
     * When adding `packages/lazy-fn-vite-plugin/tests/sources/dynamic-only` only
     * the test involving that fixture were broken on Windows... but not the one in the other fixtures:
     * - packages/lazy-fn-vite-plugin/tests/sources/vanilla
     * - packages/lazy-fn-vite-plugin/tests/sources/external-dynamic
     *
     * Awkwardly this doesn't happen on  `packages/fs-router-vite-plugin/tests/generator.spec.ts`
     *
     * Too much pain and sadness to investigate this right now.
     * Might worth creating an utility function in the future if this happens again
     */
    const source = sourceRaw.replace(new RegExp(os.EOL, 'g'), '\n')

    it('should generate file for client', async () => {
      const pluginTransform = getTransform()
      const clientBundle = pluginTransform(source, 'id')

      const expectedClientSrc = `${testDirPath}/client.expected.tsx`

      await expect(clientBundle).toMatchFileSnapshot(
        expectedClientSrc,
        `${testDirPath} client build should be equal to ${expectedClientSrc}`,
      )
    })

    it('should generate file for server', async () => {
      const pluginTransform = getTransform()
      const serverBundle = pluginTransform(source, 'id', { ssr: true })

      const expectedServerSrc = `${testDirPath}/server.expected.tsx`

      await expect(serverBundle).toMatchFileSnapshot(
        expectedServerSrc,
        `${testDirPath} server build should be equal to ${expectedServerSrc}`,
      )
    })
  })
})