import { describe, it, expect, afterEach } from 'vitest'
import { DatabaseSync } from 'node:sqlite'
import { migrate } from '../../src/main/db/migrate'
import type { SqliteDb } from '../../src/main/db/types'

describe('migrate v7', () => {
  let db: SqliteDb
  afterEach(() => db?.close())

  it('UT-MIG-07: v6 database migrates to v7 with snapshot table and knowledge columns', () => {
    db = new DatabaseSync(':memory:')
    migrate(db)
    const version = db.prepare('SELECT MAX(version) AS v FROM schema_version').get() as { v: number }
    expect(version.v).toBe(10)
    const tables = db
      .prepare("SELECT name FROM sqlite_master WHERE type='table' AND name='chapter_writing_snapshots'")
      .get()
    expect(tables).toBeTruthy()
    const cols = db.prepare('PRAGMA table_info(knowledge_entries)').all() as Array<{ name: string }>
    expect(cols.some((c) => c.name === 'extraction_confidence')).toBe(true)
    expect(cols.some((c) => c.name === 'merge_target_id')).toBe(true)
    expect(cols.some((c) => c.name === 'extraction_batch_id')).toBe(true)
  })
})