yaoyao/tests/server/serviceInstance.test.ts

// @vitest-environment node
import { afterEach, beforeEach, describe, expect, it } from 'vitest'
import { mkdtempSync, readFileSync, rmSync } from 'node:fs'
import { tmpdir } from 'node:os'
import { join } from 'node:path'
import Koa from 'koa'
import request from 'supertest'
import { acquireServiceInstance } from '../../src/server/serviceInstance'

let home: string
let instances: ReturnType<typeof acquireServiceInstance>[]
beforeEach(() => { home = mkdtempSync(join(tmpdir(), 'yaoyao-service-lock-')); instances = [] })
afterEach(() => { for (const instance of instances) instance.release(); rmSync(home, { recursive: true, force: true }) })
const acquire = (desktop = true) => { const instance = acquireServiceInstance(home, '0.3.32', desktop); instances.push(instance); return instance }

describe('service ownership', () => {
  it('excludes a second process connection and releases ownership explicitly', () => {
    const first = acquire()
    expect(() => acquire()).toThrow('另一个')
    first.release()
    const next = acquire()
    expect(next.record.instanceId).not.toBe(first.record.instanceId)
    first.release()
    expect(JSON.parse(readFileSync(join(home, 'service-instance.json'), 'utf8')).instanceId).toBe(next.record.instanceId)
  })
  it('uses a private capability and exact local service identity, not a public port probe', async () => {
    const instance = acquire();instance.publish('http://127.0.0.1:12345')
    let stops = 0
    const app = new Koa().use(instance.middleware(async () => { stops++ }))
    await request(app.callback()).get('/desktop/service').expect(403)
    await request(app.callback()).get('/desktop/service').set('x-yaoyao-desktop-token', instance.record.token).set('Origin', 'http://evil.test').expect(403)
    const result = await request(app.callback()).get('/desktop/service').set('x-yaoyao-desktop-token', instance.record.token).expect(200)
    expect(result.body).toMatchObject({ protocol: 1, pid: process.pid, instanceId: instance.record.instanceId })
    expect(result.body.token).toBeUndefined()
    await request(app.callback()).delete('/desktop/service').set('x-yaoyao-desktop-token', instance.record.token).expect(200)
    await new Promise(resolve => setImmediate(resolve))
    expect(stops).toBe(1)
  })
  it('refuses stopping an externally managed CLI service', async () => {
    const instance = acquire(false)
    const app = new Koa().use(instance.middleware(async () => { throw new Error('must not stop') }))
    await request(app.callback()).delete('/desktop/service').set('x-yaoyao-desktop-token', instance.record.token).expect(405)
  })
})