smartass
92 lines
2.8 KiB
TypeScript
92 lines
2.8 KiB
TypeScript
import {
|
|
mkdirSync,
|
|
mkdtempSync,
|
|
readdirSync,
|
|
readFileSync,
|
|
rmSync,
|
|
statSync,
|
|
writeFileSync
|
|
} from 'node:fs'
|
|
import { tmpdir } from 'node:os'
|
|
import { join } from 'node:path'
|
|
import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
|
|
import { defaultStorePath, readStore, writeStore } from '../../../src/config/store.js'
|
|
import type { ConnectionConfig } from '../../../src/config/types.js'
|
|
|
|
const connection: ConnectionConfig = {
|
|
name: 'lab',
|
|
type: 'postgres',
|
|
host: 'localhost',
|
|
user: 'postgres',
|
|
readonly: false
|
|
}
|
|
|
|
describe('store', () => {
|
|
let dir: string
|
|
let path: string
|
|
|
|
beforeEach(() => {
|
|
dir = mkdtempSync(join(tmpdir(), 'dbmole-store-'))
|
|
path = join(dir, 'nested', 'connections.json')
|
|
vi.spyOn(console, 'error').mockImplementation(() => {})
|
|
})
|
|
|
|
afterEach(() => {
|
|
rmSync(dir, { recursive: true, force: true })
|
|
vi.restoreAllMocks()
|
|
})
|
|
|
|
it('reads missing file as empty list', () => {
|
|
expect(readStore(path)).toEqual([])
|
|
})
|
|
|
|
it('round-trips connections and creates parent dirs', () => {
|
|
writeStore(path, [connection])
|
|
expect(readStore(path)).toEqual([connection])
|
|
})
|
|
|
|
it('writes file with 0600 permissions and leaves no tmp files', () => {
|
|
writeStore(path, [connection])
|
|
const mode = statSync(path).mode & 0o777
|
|
expect(mode).toBe(0o600)
|
|
const leftovers = readdirSync(join(dir, 'nested')).filter((f) => f.includes('.tmp'))
|
|
expect(leftovers).toEqual([])
|
|
})
|
|
|
|
it('treats invalid JSON as empty with a warning', () => {
|
|
writeStore(path, [connection])
|
|
writeFileSync(path, '{ broken')
|
|
expect(readStore(path)).toEqual([])
|
|
expect(console.error).toHaveBeenCalled()
|
|
})
|
|
|
|
it('treats non-array JSON as empty with a warning', () => {
|
|
writeStore(path, [connection])
|
|
writeFileSync(path, '{"connections": []}')
|
|
expect(readStore(path)).toEqual([])
|
|
expect(console.error).toHaveBeenCalled()
|
|
})
|
|
|
|
it('skips invalid entries but keeps valid ones', () => {
|
|
mkdirSync(join(dir, 'nested'), { recursive: true })
|
|
writeFileSync(path, JSON.stringify([connection, { name: 'broken' }]))
|
|
expect(readStore(path)).toEqual([connection])
|
|
expect(console.error).toHaveBeenCalled()
|
|
})
|
|
|
|
it('stores pretty JSON for hand editing', () => {
|
|
writeStore(path, [connection])
|
|
expect(readFileSync(path, 'utf8')).toContain('\n "name": "lab"')
|
|
})
|
|
})
|
|
|
|
describe('defaultStorePath', () => {
|
|
it('honors DBMOLE_STORE override', () => {
|
|
expect(defaultStorePath({ DBMOLE_STORE: '/tmp/custom.json' })).toBe('/tmp/custom.json')
|
|
})
|
|
|
|
it('falls back to ~/.config/dbmole/connections.json', () => {
|
|
expect(defaultStorePath({})).toMatch(/\.config\/dbmole\/connections\.json$/)
|
|
})
|
|
})
|