@prisma/language-server/src/__test__/completions/multi-file.test.ts

Prisma Language Server

import { test, expect } from 'vitest'
import { handleCompletionRequest } from '../../lib/MessageHandler'
import { getMultifileHelper } from '../MultifileHelper'

test('type name completion', async () => {
  const helper = await getMultifileHelper('complete-field')
  const post = helper.file('Post.prisma')

  const response = handleCompletionRequest(helper.schema, post.textDocument, {
    textDocument: {
      uri: post.uri,
    },
    position: post.lineContaining('author Us').characterAfter('U'),
  })

  const userItem = response?.items.find((item) => item.label === 'User')
  expect(userItem).not.toBeUndefined()
})

test('type name completion with no name typed', async () => {
  const helper = await getMultifileHelper('complete-field')
  const post = helper.file('Post.prisma')

  const response = handleCompletionRequest(helper.schema, post.textDocument, {
    textDocument: {
      uri: post.uri,
    },
    position: post.lineContaining('author ').characterAfter('r '),
  })

  const userItem = response?.items.find((item) => item.label === 'User')
  expect(userItem).not.toBeUndefined()
})

test('native type complete', async () => {
  const helper = await getMultifileHelper('complete-native-type')
  const post = helper.file('Post.prisma')

  const response = handleCompletionRequest(helper.schema, post.textDocument, {
    textDocument: {
      uri: post.uri,
    },
    position: post.lineContaining('Decimal').characterAfter('@db'),
  })

  expect(response?.items?.map((item) => item.label)).toMatchInlineSnapshot(`
    [
      "Decimal()",
      "Money",
    ]
  `)
})