bing
d4122c8f95
Add pomodoro timer with goal notifications, writing streak milestones, and knowledge injection logging with UI previews across AI writing flows. Co-authored-by: Cursor <cursoragent@cursor.com>
58 lines
2.0 KiB
TypeScript
58 lines
2.0 KiB
TypeScript
import { describe, it, expect, afterEach } from 'vitest'
|
|
import { DatabaseSync } from 'node:sqlite'
|
|
import { migrate } from '../../src/main/db/migrate'
|
|
import schemaV1 from '../../src/main/db/schema-v1.sql?raw'
|
|
import type { SqliteDb } from '../../src/main/db/types'
|
|
|
|
function tableExists(db: SqliteDb, name: string): boolean {
|
|
const row = db.prepare("SELECT name FROM sqlite_master WHERE type='table' AND name=?").get(name)
|
|
return row != null
|
|
}
|
|
|
|
describe('migrate v2', () => {
|
|
let db: SqliteDb
|
|
|
|
afterEach(() => {
|
|
db?.close()
|
|
})
|
|
|
|
it('applies v1 and v2 on fresh database', () => {
|
|
db = new DatabaseSync(':memory:')
|
|
migrate(db)
|
|
|
|
expect(tableExists(db, 'volumes')).toBe(true)
|
|
expect(tableExists(db, 'chapters')).toBe(true)
|
|
expect(tableExists(db, 'outline_items')).toBe(true)
|
|
expect(tableExists(db, 'settings')).toBe(true)
|
|
expect(tableExists(db, 'inspiration')).toBe(true)
|
|
expect(tableExists(db, 'snapshots')).toBe(true)
|
|
expect(tableExists(db, 'bookmarks')).toBe(true)
|
|
expect(tableExists(db, 'search_fts')).toBe(true)
|
|
|
|
const version = db.prepare('SELECT MAX(version) AS v FROM schema_version').get() as { v: number }
|
|
expect(version.v).toBe(8)
|
|
})
|
|
|
|
it('migrates existing v1 database to v2', () => {
|
|
db = new DatabaseSync(':memory:')
|
|
db.exec(`CREATE TABLE IF NOT EXISTS schema_version (
|
|
version INTEGER PRIMARY KEY,
|
|
applied_at TEXT NOT NULL DEFAULT (datetime('now')),
|
|
description TEXT
|
|
)`)
|
|
db.exec(schemaV1)
|
|
db.prepare('INSERT INTO schema_version (version, description) VALUES (?, ?)').run(1, 'initial schema')
|
|
|
|
migrate(db)
|
|
|
|
expect(tableExists(db, 'outline_items')).toBe(true)
|
|
const version = db.prepare('SELECT MAX(version) AS v FROM schema_version').get() as { v: number }
|
|
expect(version.v).toBe(8)
|
|
|
|
const cols = db.prepare('PRAGMA table_info(chapters)').all() as { name: string }[]
|
|
const colNames = cols.map((c) => c.name)
|
|
expect(colNames).toContain('publish_status')
|
|
expect(colNames).toContain('summary')
|
|
})
|
|
})
|