mirror of
https://github.com/stablyai/orca.git
synced 2026-09-27 08:02:35 +00:00
Jinwoo Hong
eb2f2d52ae
* refactor(cloud): share PostgreSQL schema startup between services * feat(cloud): add durable native push notification gateway * infra(push): define dedicated gateway resources and operational checks * fix(push): bound cross-host admission and simplify gateway configuration * fix(push): validate deploy configuration and preserve topic-error registrations
46 lines
1.5 KiB
TypeScript
46 lines
1.5 KiB
TypeScript
import { EventEmitter } from 'node:events'
|
|
import { expect, it, vi } from 'vitest'
|
|
const mocks = vi.hoisted(() => ({
|
|
connect: vi.fn(),
|
|
read: vi.fn(async () => ({ status: 200, body: '' }))
|
|
}))
|
|
vi.mock('node:http2', async (original) => ({
|
|
...(await original<typeof import('node:http2')>()),
|
|
connect: mocks.connect
|
|
}))
|
|
vi.mock('./apns-stream-response.js', () => ({ readApnsStreamResponse: mocks.read }))
|
|
import { createApnsHttp2Transport } from './apns-http2-transport.js'
|
|
|
|
it('keeps the replacement cached when the draining session closes later', async () => {
|
|
const sessions: Array<
|
|
EventEmitter & {
|
|
closed: boolean
|
|
destroyed: boolean
|
|
request: ReturnType<typeof vi.fn>
|
|
close: ReturnType<typeof vi.fn>
|
|
}
|
|
> = []
|
|
mocks.connect.mockImplementation(() => {
|
|
const session = Object.assign(new EventEmitter(), {
|
|
closed: false,
|
|
destroyed: false,
|
|
request: vi.fn(() => ({})),
|
|
close: vi.fn()
|
|
})
|
|
sessions.push(session)
|
|
return session
|
|
})
|
|
const transport = createApnsHttp2Transport()
|
|
const request = { host: 'api.push.apple.com', path: '/synthetic', headers: {}, body: '{}' }
|
|
await transport(request)
|
|
sessions[0]!.closed = true
|
|
await transport(request)
|
|
sessions[0]!.emit('close')
|
|
sessions[0]!.emit('error', new Error('old-session'))
|
|
await transport(request)
|
|
expect(sessions).toHaveLength(2)
|
|
expect(sessions[1]!.request).toHaveBeenCalledTimes(2)
|
|
transport.close()
|
|
expect(sessions[1]!.close).toHaveBeenCalledOnce()
|
|
})
|