mirror of
https://github.com/stablyai/orca.git
synced 2026-11-02 01:57:57 +00:00
Jinwoo Hong
aec426c2fa
Adds Discord as a second connector adapter on the shared core: signed interactions for answer buttons, one gateway worker (same image, single leader by database lock, commits on the lock's connection, checkpoints for resume), thread posts through the shared post queue, owner-only control, and teammate context only from linked accounts. Inert until the Discord settings are configured. A narrow gitleaks allowlist entry covers connector Discord env var names in config errors.
71 lines
2.6 KiB
TypeScript
71 lines
2.6 KiB
TypeScript
import { randomUUID } from 'node:crypto'
|
|
import pg from 'pg'
|
|
import { describe, expect, it } from 'vitest'
|
|
import { openInMemoryCloudDatabase } from './cloud-database.js'
|
|
import { acquireLeadership } from './cloud-leader-lock.js'
|
|
|
|
const postgresUrl = process.env.ORCA_CLOUD_DATABASE_TEST_URL
|
|
const database = await openInMemoryCloudDatabase([])
|
|
|
|
describe('acquireLeadership', () => {
|
|
it('leads at once without a database', async () => {
|
|
const leadership = await acquireLeadership({
|
|
database,
|
|
key: 'gateway',
|
|
signal: new AbortController().signal
|
|
})
|
|
expect(
|
|
await leadership.transaction((transaction) => transaction.query('SELECT 1 AS one'))
|
|
).toEqual([{ one: 1 }])
|
|
await leadership.release()
|
|
await expect(leadership.lost).resolves.toBeUndefined()
|
|
})
|
|
|
|
it.skipIf(!postgresUrl)(
|
|
'lets one holder lead until it releases or its session dies',
|
|
async () => {
|
|
const key = `gateway-${randomUUID()}`
|
|
const options = { databaseUrl: postgresUrl, database, key, pollMs: 50, checkMs: 50 }
|
|
const first = await acquireLeadership({ ...options, signal: AbortSignal.timeout(5_000) })
|
|
let secondLed = false
|
|
const second = acquireLeadership({ ...options, signal: AbortSignal.timeout(5_000) }).then(
|
|
(leadership) => {
|
|
secondLed = true
|
|
return leadership
|
|
}
|
|
)
|
|
await new Promise((resolve) => setTimeout(resolve, 200))
|
|
expect(secondLed).toBe(false)
|
|
await first.release()
|
|
const next = await second
|
|
|
|
// Killing the leader's session frees the lock and the holder notices.
|
|
const admin = new pg.Client({ connectionString: postgresUrl })
|
|
await admin.connect()
|
|
await admin.query(
|
|
`SELECT pg_terminate_backend(pid) FROM pg_locks
|
|
WHERE locktype = 'advisory' AND objid = hashtext($1::text)::oid AND granted`,
|
|
[key]
|
|
)
|
|
await admin.end()
|
|
await next.lost
|
|
// Its transactions ran on the lock's connection, so none can commit after the loss.
|
|
await expect(
|
|
next.transaction((transaction) => transaction.query('SELECT 1'))
|
|
).rejects.toThrow()
|
|
const third = await acquireLeadership({ ...options, signal: AbortSignal.timeout(5_000) })
|
|
await third.release()
|
|
}
|
|
)
|
|
|
|
it.skipIf(!postgresUrl)('stops waiting when aborted', async () => {
|
|
const key = `gateway-${randomUUID()}`
|
|
const options = { databaseUrl: postgresUrl, database, key, pollMs: 50 }
|
|
const first = await acquireLeadership({ ...options, signal: AbortSignal.timeout(5_000) })
|
|
await expect(
|
|
acquireLeadership({ ...options, signal: AbortSignal.timeout(200) })
|
|
).rejects.toThrow()
|
|
await first.release()
|
|
})
|
|
})
|