yaoyao/desktop/service-sync.test.mjs

import { test } from 'node:test'
import assert from 'node:assert/strict'
import { mkdtemp, readFile, writeFile, rm, cp, realpath } from 'node:fs/promises'
import { tmpdir } from 'node:os'
import { dirname, join, resolve } from 'node:path'
import { createServer } from 'node:net'
import { spawn } from 'node:child_process'
import { createHash } from 'node:crypto'
import { _electron as electron } from '@playwright/test'
import { LaunchAgentService, localRequest, synchronizeDesktop } from '../bin/lib/service-update.mjs'
import { sealRuntime, verifyRuntimePackage } from '../bin/lib/runtime-release.mjs'

const root = resolve(import.meta.dirname, '..')
const payloadRoot = process.env.DESKTOP_TEST_EXECUTABLE
  ? resolve(dirname(process.env.DESKTOP_TEST_EXECUTABLE), '../Resources/runtime/web-service')
  : join(root, '.desktop-build', 'web-service')
async function freePort() {
  const server = createServer(); await new Promise(done => server.listen(0, '127.0.0.1', done))
  const port = server.address().port; await new Promise(done => server.close(done)); return port
}
async function until(read, accept, timeout = 60000) {
  const deadline = Date.now() + timeout
  while (Date.now() < deadline) {
    try { const value = await read(); if (accept(value)) return value } catch {}
    await new Promise(done => setTimeout(done, 200))
  }
  throw new Error('等待同步服务状态超时')
}

test('App first launch installs an independent Web; Web remains up after quit and a newer Web is retained', { timeout: 150000 }, async () => {
  const home = await realpath(await mkdtemp(join(tmpdir(), 'yaoyao-sync-native-')))
  const port = await freePort(), upstreamPort = await freePort()
  const releaseRoot = join(home, 'updates', 'fixture-releases')
  const label = `cn.samien.yaoyao.sync-test.${createHash('sha256').update(home).digest('hex').slice(0, 16)}`
  const driver = new LaunchAgentService({ home, port, releaseRoot, label, plistPath: join(home, 'updates', 'fixture-service.plist') })
  const upstream = spawn(process.execPath, [join(root, 'tests/fixtures/fake-hermes.mjs')], {
    cwd: root, env: { ...process.env, FAKE_HERMES_PORT: String(upstreamPort), FAKE_HERMES_LOCAL_AUTH: '1' }, stdio: 'ignore',
  })
  let app
  const launch = () => electron.launch({
    ...(process.env.DESKTOP_TEST_EXECUTABLE ? { executablePath: process.env.DESKTOP_TEST_EXECUTABLE } : {}),
    args: process.env.DESKTOP_TEST_EXECUTABLE ? [] : [root], cwd: root,
    env: { ...process.env, HERMES_YAOYAO_DESKTOP_TEST_HOME: home, HERMES_YAOYAO_DESKTOP_PORT: String(port),
      HERMES_YAOYAO_DESKTOP_TEST_SYNC: '1', HERMES_YAOYAO_UPSTREAM: `http://127.0.0.1:${upstreamPort}` }, timeout: 30000,
  })
  try {
    await until(() => localRequest(`http://127.0.0.1:${upstreamPort}/api/status`), value => value.status === 200)
    app = await launch()
    const page = await app.firstWindow()
    await page.waitForURL(`http://127.0.0.1:${port}/**`, { timeout: 90000 })
    const record = JSON.parse(await readFile(join(home, 'service-instance.json'), 'utf8'))
    assert.equal(record.desktopOwned, false)
    const identity = await localRequest(`http://127.0.0.1:${port}/desktop/service`, record.token)
    const original = verifyRuntimePackage(payloadRoot)
    assert.equal(identity.body.build.artifactDigest, original.artifactDigest)
    assert.equal(identity.body.quiesced, false)
    const support = await page.evaluate(async () => {
      const bootstrap = await (await fetch('/api/app/bootstrap')).json()
      const setup = await fetch('/api/app/setup', { method: 'POST', headers: { 'Content-Type': 'application/json', 'X-CSRF-Token': bootstrap.csrfToken },
        body: JSON.stringify({ username: 'sync-fixture', password: 'sync-fixture-password' }) })
      if (!setup.ok) throw new Error('验收账号创建失败')
      return (await fetch('/api/app/system/update/status')).json()
    })
    assert.equal(support.installationMode, 'release'); assert.equal(support.supported, true)
    await app.close(); app = undefined
    assert.equal((await localRequest(`http://127.0.0.1:${port}/healthz`)).body.ok, true)
    assert.equal(driver.pid(), record.pid, 'quitting the App must leave the independent Web running')
    const saved = driver.snapshot().plist
    saved.EnvironmentVariables.YAOYAO_TEST_PRESERVE = 'keep-existing-settings'
    driver.writePlist(saved)

    // Install a later release through the same transaction engine used by Web's updater.
    const newer = join(home, 'updates', 'newer-package')
    await cp(payloadRoot, newer, { recursive: true })
    await writeFile(join(newer, 'release.json'), JSON.stringify({ schemaVersion: 1, releaseVersion: '0.3.33', webVersion: '0.3.33', gitTag: 'v0.3.33' }))
    const build = JSON.parse(await readFile(join(newer, 'build-info.json'), 'utf8'))
    await writeFile(join(newer, 'build-info.json'), JSON.stringify({ ...build, commit: 'c'.repeat(40), ancestors: [build.commit, ...build.ancestors], dirty: false }))
    sealRuntime(newer)
    await synchronizeDesktop({ home, runtimeRoot: newer, releaseRoot, driver })
    const updated = JSON.parse(await readFile(join(home, 'service-instance.json'), 'utf8'))
    assert.equal(updated.version, '0.3.33'); assert.notEqual(updated.pid, record.pid)
    assert.equal(driver.snapshot().plist.EnvironmentVariables.YAOYAO_TEST_PRESERVE, 'keep-existing-settings')
    app = await launch()
    const reopened = await app.firstWindow(); await reopened.waitForURL(`http://127.0.0.1:${port}/**`, { timeout: 90000 })
    assert.equal(driver.pid(), updated.pid, 'older App must retain the newer Web without restarting it')
    const marker = JSON.parse(await readFile(join(home, 'updates', 'desktop-sync.json'), 'utf8'))
    assert.equal(marker.action, 'newer')
    assert.equal(marker.artifactDigest, original.artifactDigest)
    await app.close(); app = undefined
    assert.equal((await localRequest(`http://127.0.0.1:${port}/healthz`)).body.ok, true)
    await driver.stop()
    app = await launch()
    const restarted = await app.firstWindow(); await restarted.waitForURL(`http://127.0.0.1:${port}/**`, { timeout: 90000 })
    const resumed = JSON.parse(await readFile(join(home, 'service-instance.json'), 'utf8'))
    assert.equal(resumed.version, '0.3.33'); assert.equal(resumed.desktopOwned, false)
    assert.notEqual(resumed.pid, updated.pid)
    await app.close(); app = undefined
  } catch (error) {
    // Only this fixture's server log, which contains startup diagnostics.
    const log = await readFile(join(home, 'service.log'), 'utf8').catch(() => '')
    if (log) process.stderr.write(log.slice(-3000))
    throw error
  } finally {
    await app?.close().catch(() => {})
    await driver.stop()
    upstream.kill('SIGTERM')
    await rm(home, { recursive: true, force: true })
  }
})