smartass
38 lines
1.2 KiB
TypeScript
38 lines
1.2 KiB
TypeScript
import { mkdtempSync, rmSync } from 'node:fs'
|
|
import { tmpdir } from 'node:os'
|
|
import { join } from 'node:path'
|
|
import { afterEach, beforeEach, describe, expect, it } from 'vitest'
|
|
import { createRegistry } from '../../src/config/registry.js'
|
|
import { createServer } from '../../src/server.js'
|
|
import { connectClient, fakeManager } from './helpers.js'
|
|
|
|
describe('createServer', () => {
|
|
let dir: string
|
|
|
|
beforeEach(() => {
|
|
dir = mkdtempSync(join(tmpdir(), 'dbmole-server-'))
|
|
})
|
|
|
|
afterEach(() => {
|
|
rmSync(dir, { recursive: true, force: true })
|
|
})
|
|
|
|
it('registers all nine tools', async () => {
|
|
const registry = createRegistry({ storePath: join(dir, 'connections.json'), env: {} })
|
|
const server = createServer(registry, fakeManager())
|
|
const client = await connectClient(server)
|
|
const { tools } = await client.listTools()
|
|
expect(tools.map((tool) => tool.name).sort()).toEqual([
|
|
'add_connection',
|
|
'describe_table',
|
|
'execute_sql',
|
|
'list_connections',
|
|
'list_databases',
|
|
'list_tables',
|
|
'remove_connection',
|
|
'test_connection',
|
|
'update_connection'
|
|
])
|
|
})
|
|
})
|