mirror of
https://github.com/stablyai/orca.git
synced 2026-09-22 16:02:32 +00:00
Jinwoo Hong
3eec77c11a
Phase 6 of the relay split: the relay's deploy/operate surface moves under cloud/ with 24 cloud-* workflows gated on ORCA_CLOUD_OPERATIONS_ENABLED, the Cloud SQL rollout lease action, the relay Terraform root (dual-accept identities for both repositories), scripts, docs, CODEOWNERS, and a terraform validate job in Cloud Verify.
88 lines
3.0 KiB
TypeScript
88 lines
3.0 KiB
TypeScript
import pg from 'pg'
|
|
import { afterAll, beforeAll, describe, expect, it } from 'vitest'
|
|
import { RelayAssignmentStore } from './assignment-store.js'
|
|
import { openRelayDatabase, type RelayDatabase } from './database.js'
|
|
|
|
const databaseUrl = process.env.ORCA_RELAY_TEST_POSTGRES_URL
|
|
const describePostgres = databaseUrl ? describe : describe.skip
|
|
const schema = 'relay_deployment_status_test'
|
|
const cell = {
|
|
id: 'deployment-status-postgres',
|
|
url: 'https://deployment-status-postgres.example.com',
|
|
capacityRequests: 20
|
|
}
|
|
|
|
describePostgres('PostgreSQL deployment status', () => {
|
|
let database: RelayDatabase | undefined
|
|
|
|
beforeAll(async () => {
|
|
const client = new pg.Client({ connectionString: databaseUrl })
|
|
await client.connect()
|
|
try {
|
|
await client.query(`DROP SCHEMA IF EXISTS ${schema} CASCADE`)
|
|
await client.query(`CREATE SCHEMA ${schema}`)
|
|
} finally {
|
|
await client.end()
|
|
}
|
|
const url = new URL(databaseUrl!)
|
|
url.searchParams.set('options', `-c search_path=${schema}`)
|
|
database = await openRelayDatabase({ databaseUrl: url.toString(), dataDir: '' })
|
|
})
|
|
|
|
afterAll(async () => {
|
|
await database?.close()
|
|
const client = new pg.Client({ connectionString: databaseUrl })
|
|
await client.connect()
|
|
try {
|
|
await client.query(`DROP SCHEMA IF EXISTS ${schema} CASCADE`)
|
|
} finally {
|
|
await client.end()
|
|
}
|
|
})
|
|
|
|
it('classifies pending controls and restart blockers in one PostgreSQL query', async () => {
|
|
const store = new RelayAssignmentStore(database!, () => 100)
|
|
await store.reconcileCells([cell])
|
|
await store.assign({ userId: 'status-user', relayHostId: 'a1b2c3d4e5f6' })
|
|
|
|
expect(await store.cellDeploymentStatus(cell.id)).toMatchObject({
|
|
activityLeases: 1,
|
|
activityRequestUnits: 1,
|
|
reservedRequests: 1,
|
|
restartBlockingActivityLeases: 0,
|
|
restartBlockingActivityRequestUnits: 0,
|
|
restartBlockingReservedRequests: 0
|
|
})
|
|
|
|
await database!.query(
|
|
`INSERT INTO relay_assignment_activity_leases
|
|
(user_id, relay_host_id, activity_id, activity_kind, cell_id,
|
|
request_units, expires_at, updated_at)
|
|
VALUES ('blocking-user', 'b1c2d3e4f5a6', 'invite:test', 'invite', ?, 1, 90100, 100),
|
|
('malformed-user', 'c1d2e3f4a5b6', 'control-pending:bad', 'control', ?, 2, 90100, 100)`,
|
|
[cell.id, cell.id]
|
|
)
|
|
await database!.query(
|
|
`UPDATE relay_cells SET reserved_requests = reserved_requests + 3 WHERE cell_id = ?`,
|
|
[cell.id]
|
|
)
|
|
|
|
expect(await store.cellDeploymentStatus(cell.id)).toMatchObject({
|
|
activityLeases: 3,
|
|
activityRequestUnits: 4,
|
|
reservedRequests: 4,
|
|
restartBlockingActivityLeases: 2,
|
|
restartBlockingActivityRequestUnits: 3,
|
|
restartBlockingReservedRequests: 3
|
|
})
|
|
|
|
await database!.query(
|
|
`UPDATE relay_cells SET reserved_requests = 0 WHERE cell_id = ?`,
|
|
[cell.id]
|
|
)
|
|
expect(await store.cellDeploymentStatus(cell.id)).toMatchObject({
|
|
restartBlockingReservedRequests: -1
|
|
})
|
|
})
|
|
})
|