test: import-resolver

Author: devdumpling

test/unit/import-resolver.spec.ts

@@ -0,0 +1,82 @@
+import { describe, expect, it } from 'vitest'
+import type { PackageFileTree } from '../../shared/types'
+import {
+  createImportResolver,
+  flattenFileTree,
+  resolveRelativeImport,
+} from '../../server/utils/import-resolver'
+
+describe('flattenFileTree', () => {
+  it('flattens nested trees into a file set', () => {
+    const tree: PackageFileTree[] = [
+      {
+        name: 'dist',
+        path: 'dist',
+        type: 'directory',
+        children: [
+          { name: 'index.js', path: 'dist/index.js', type: 'file', size: 10 },
+          {
+            name: 'utils',
+            path: 'dist/utils',
+            type: 'directory',
+            children: [{ name: 'format.js', path: 'dist/utils/format.js', type: 'file', size: 5 }],
+          },
+        ],
+      },
+    ]
+
+    const files = flattenFileTree(tree)
+
+    expect(files.has('dist/index.js')).toBe(true)
+    expect(files.has('dist/utils/format.js')).toBe(true)
+    expect(files.has('dist/utils')).toBe(false)
+  })
+})
+
+describe('resolveRelativeImport', () => {
+  it('resolves a relative import with extension priority for JS files', () => {
+    const files = new Set<string>(['dist/utils.js', 'dist/utils.ts'])
+    const resolved = resolveRelativeImport('./utils', 'dist/index.js', files)
+
+    expect(resolved?.path).toBe('dist/utils.js')
+  })
+
+  it('resolves directory imports to index files', () => {
+    const files = new Set<string>(['dist/components/index.js'])
+    const resolved = resolveRelativeImport('./components', 'dist/index.js', files)
+
+    expect(resolved?.path).toBe('dist/components/index.js')
+  })
+
+  it('resolves parent directory paths', () => {
+    const files = new Set<string>(['dist/shared/helpers.js'])
+    const resolved = resolveRelativeImport('../shared/helpers', 'dist/pages/home.js', files)
+
+    expect(resolved?.path).toBe('dist/shared/helpers.js')
+  })
+
+  it('returns null for non-relative imports', () => {
+    const files = new Set<string>(['dist/utils.js'])
+    const resolved = resolveRelativeImport('react', 'dist/index.js', files)
+
+    expect(resolved).toBeNull()
+  })
+
+  it('returns null when no matching file is found', () => {
+    const files = new Set<string>(['dist/utils.js'])
+    const resolved = resolveRelativeImport('./missing', 'dist/index.js', files)
+
+    expect(resolved).toBeNull()
+  })
+})
+
+describe('createImportResolver', () => {
+  it('creates a resolver that returns code browser URLs', () => {
+    const files = new Set<string>(['dist/utils.js'])
+    const resolver = createImportResolver(files, 'dist/index.js', 'pkg-name', '1.2.3')
+
+    const url = resolver('./utils')
+
+    expect(url).toBe('/code/pkg-name/v/1.2.3/dist/utils.js')
+  })
+})