76 lines
2.7 KiB
TypeScript
76 lines
2.7 KiB
TypeScript
import { describe, it, expect } from 'vitest';
|
|
import Database from 'better-sqlite3';
|
|
import { runMigrations } from '@main/db/migrations/index.js';
|
|
|
|
describe('migrations', () => {
|
|
it('creates schema with intent + edited columns', () => {
|
|
const db = new Database(':memory:');
|
|
runMigrations(db);
|
|
const cols = db.prepare(`PRAGMA table_info(notes)`).all().map((r: any) => r.name);
|
|
expect(cols).toEqual(
|
|
expect.arrayContaining([
|
|
'id', 'raw_text', 'ai_title', 'ai_summary', 'ai_status', 'ai_error',
|
|
'ai_provider', 'ai_generated_at',
|
|
'title_edited_by_user', 'summary_edited_by_user',
|
|
'user_intent', 'intent_prompted_at',
|
|
'created_at', 'updated_at'
|
|
])
|
|
);
|
|
db.close();
|
|
});
|
|
|
|
it('is idempotent', () => {
|
|
const db = new Database(':memory:');
|
|
runMigrations(db);
|
|
const before = (db.prepare('PRAGMA user_version').get() as any).user_version;
|
|
runMigrations(db);
|
|
const after = (db.prepare('PRAGMA user_version').get() as any).user_version;
|
|
expect(after).toBe(before);
|
|
db.close();
|
|
});
|
|
});
|
|
|
|
describe('migration v3 — soft delete columns', () => {
|
|
it('adds deleted_at, last_recalled_at, recall_dismissed_at to notes', () => {
|
|
const db = new Database(':memory:');
|
|
runMigrations(db);
|
|
const cols = db.prepare(`PRAGMA table_info(notes)`).all().map((r: any) => r.name);
|
|
expect(cols).toEqual(
|
|
expect.arrayContaining(['deleted_at', 'last_recalled_at', 'recall_dismissed_at'])
|
|
);
|
|
db.close();
|
|
});
|
|
|
|
it('creates idx_notes_deleted_at index', () => {
|
|
const db = new Database(':memory:');
|
|
runMigrations(db);
|
|
const indexes = db
|
|
.prepare(`SELECT name FROM sqlite_master WHERE type='index' AND tbl_name='notes'`)
|
|
.all() as Array<{ name: string }>;
|
|
expect(indexes.map((i) => i.name)).toContain('idx_notes_deleted_at');
|
|
db.close();
|
|
});
|
|
|
|
it('user_version reaches 3', () => {
|
|
const db = new Database(':memory:');
|
|
runMigrations(db);
|
|
const row = db.prepare('PRAGMA user_version').get() as { user_version: number };
|
|
expect(row.user_version).toBe(3);
|
|
db.close();
|
|
});
|
|
|
|
it('all 3 new columns default to NULL', () => {
|
|
const db = new Database(':memory:');
|
|
runMigrations(db);
|
|
db.prepare(
|
|
`INSERT INTO notes (id, raw_text, ai_status, created_at, updated_at)
|
|
VALUES ('n1', 't', 'pending', '2026-05-01T00:00:00Z', '2026-05-01T00:00:00Z')`
|
|
).run();
|
|
const row = db.prepare('SELECT deleted_at, last_recalled_at, recall_dismissed_at FROM notes WHERE id=?').get('n1') as any;
|
|
expect(row.deleted_at).toBeNull();
|
|
expect(row.last_recalled_at).toBeNull();
|
|
expect(row.recall_dismissed_at).toBeNull();
|
|
db.close();
|
|
});
|
|
});
|