2026-10-20 18:56:50 +08:00
|
|
|
|
import { describe, it, expect } from 'vitest'
|
|
|
|
|
|
import { getDb } from '../src/db/client.js'
|
|
|
|
|
|
|
|
|
|
|
|
describe('数据库连接', () => {
|
|
|
|
|
|
it('连接本项目 PG（5434）并响应 select 1', async () => {
|
|
|
|
|
|
const db = getDb()
|
|
|
|
|
|
// postgres.js 驱动下 drizzle 的 execute 返回 RowList（行数组），
|
|
|
|
|
|
// 不是 node-postgres 风格的 { rows: [...] }，故直接取 r[0]
|
|
|
|
|
|
const r = (await db.execute('select 1 as ok')) as any
|
|
|
|
|
|
expect(Number(r[0].ok)).toBe(1)
|
|
|
|
|
|
})
|
2026-10-20 19:10:19 +08:00
|
|
|
|
|
|
|
|
|
|
it('DATABASE_URL 缺失时 getDb 早失败并给出诊断信息（不静默连 localhost）', () => {
|
|
|
|
|
|
const savedTest = process.env.DATABASE_URL_TEST
|
|
|
|
|
|
const saved = process.env.DATABASE_URL
|
|
|
|
|
|
delete process.env.DATABASE_URL_TEST
|
|
|
|
|
|
delete process.env.DATABASE_URL
|
|
|
|
|
|
try {
|
|
|
|
|
|
// postgres(undefined) 会静默回退 OS 用户@localhost:5432，必须在此前置拦截
|
|
|
|
|
|
expect(() => getDb()).toThrow(/DATABASE_URL not set/)
|
|
|
|
|
|
} finally {
|
|
|
|
|
|
if (savedTest !== undefined) process.env.DATABASE_URL_TEST = savedTest
|
|
|
|
|
|
if (saved !== undefined) process.env.DATABASE_URL = saved
|
|
|
|
|
|
}
|
|
|
|
|
|
})
|
2026-10-20 18:56:50 +08:00
|
|
|
|
})
|
