yaoyao/tests/updater.test.ts

import { lstatSync, mkdirSync, mkdtempSync, readFileSync, rmSync, writeFileSync } from 'node:fs'
import { tmpdir } from 'node:os'
import { join } from 'node:path'
import { afterEach, describe, expect, it } from 'vitest'
import { backupPlugin, currentTarget, restorePlugin, switchCurrent, validateManifest } from '../bin/hermes-yaoyao-updater.mjs'

const roots: string[] = []
afterEach(() => {
  for (const root of roots.splice(0)) rmSync(root, { recursive: true, force: true })
})

describe('standalone updater primitives', () => {
  it('atomically switches the stable current symlink', () => {
    const root = mkdtempSync(join(tmpdir(), 'hermes-updater-'))
    roots.push(root)
    const first = join(root, 'releases', '0.2.0-a')
    const second = join(root, 'releases', '0.3.0-b')
    switchCurrent(root, first, 'first')
    expect(currentTarget(root)).toBe(first)
    switchCurrent(root, second, 'second')
    expect(currentTarget(root)).toBe(second)
    expect(lstatSync(join(root, 'current')).isSymbolicLink()).toBe(true)
  })

  it('rejects a mismatched release manifest before any service transition', () => {
    expect(() => validateManifest({
      schemaVersion: 1,
      releaseVersion: '0.3.0',
      webVersion: '0.3.1',
      pluginVersion: '1.8.0',
      gitTag: 'v0.3.0',
    })).toThrow('版本组合不一致')
  })

  it('backs up and restores plugin code without touching durable plugin data', () => {
    const root = mkdtempSync(join(tmpdir(), 'hermes-updater-plugin-'))
    roots.push(root)
    const hermesHome = join(root, 'hermes')
    const dataHome = join(root, 'web-data')
    const dashboard = join(hermesHome, 'plugins', 'yaoyao', 'dashboard')
    const durable = join(hermesHome, 'plugin-data', 'yaoyao')
    mkdirSync(dashboard, { recursive: true })
    mkdirSync(durable, { recursive: true })
    writeFileSync(join(dashboard, 'manifest.json'), JSON.stringify({ version: '1.7.1' }))
    writeFileSync(join(dashboard, 'plugin.py'), 'old plugin')
    writeFileSync(join(hermesHome, 'plugins', 'yaoyao', 'plugin.yaml'), 'version: "1.7.1"\n')
    writeFileSync(join(durable, 'archive.db'), 'durable data')

    const backup = backupPlugin({ hermesHome, dataHome }, '11111111-1111-4111-8111-111111111111')
    writeFileSync(join(dashboard, 'plugin.py'), 'new plugin')
    restorePlugin(backup.pluginRoot, backup.backupRoot, 'restore', true)

    expect(readFileSync(join(dashboard, 'plugin.py'), 'utf8')).toBe('old plugin')
    expect(readFileSync(join(durable, 'archive.db'), 'utf8')).toBe('durable data')
  })
})