|
1 | 1 | import { beforeEach, expect, inject, onTestFinished, test } from 'vitest' |
2 | 2 |
|
3 | 3 | import { Driver } from '@ydbjs/core' |
4 | | -import { CoordinationClient, type CoordinationSession } from '@ydbjs/coordination' |
| 4 | +import { |
| 5 | + CoordinationClient, |
| 6 | + type CoordinationSession, |
| 7 | + LeaderChangedError, |
| 8 | + LeaseReleasedError, |
| 9 | +} from '@ydbjs/coordination' |
5 | 10 |
|
6 | 11 | // #region setup |
7 | 12 | declare module 'vitest' { |
@@ -306,4 +311,59 @@ test('leader state signal aborts when the leader changes', async () => { |
306 | 311 |
|
307 | 312 | expect(capturedSignal).toBeDefined() |
308 | 313 | expect(capturedSignal!.aborted).toBe(true) |
| 314 | + expect(capturedSignal!.reason).toBeInstanceOf(LeaderChangedError) |
| 315 | +}) |
| 316 | + |
| 317 | +test('leadership.signal.reason is LeaseReleasedError after resign', async () => { |
| 318 | + let election = sessionA.election(electionName) |
| 319 | + |
| 320 | + let leadership = await election.campaign(Buffer.from('A'), AbortSignal.timeout(5000)) |
| 321 | + await leadership.resign(AbortSignal.timeout(5000)) |
| 322 | + |
| 323 | + expect(leadership.signal.reason).toBeInstanceOf(LeaseReleasedError) |
| 324 | +}) |
| 325 | + |
| 326 | +test('double resign is idempotent', async () => { |
| 327 | + let election = sessionA.election(electionName) |
| 328 | + |
| 329 | + let leadership = await election.campaign(Buffer.from('A'), AbortSignal.timeout(5000)) |
| 330 | + await leadership.resign(AbortSignal.timeout(5000)) |
| 331 | + await expect(leadership.resign(AbortSignal.timeout(5000))).resolves.toBeUndefined() |
| 332 | +}) |
| 333 | + |
| 334 | +test('N sessions race for leadership — exactly one wins', async () => { |
| 335 | + let N = 4 |
| 336 | + |
| 337 | + let sessions: CoordinationSession[] = [] |
| 338 | + for (let i = 0; i < N; i++) { |
| 339 | + // oxlint-disable-next-line no-await-in-loop |
| 340 | + sessions.push(await client.createSession(testNodePath, {}, AbortSignal.timeout(5000))) |
| 341 | + } |
| 342 | + |
| 343 | + try { |
| 344 | + // All sessions campaign simultaneously with short timeout |
| 345 | + let results = await Promise.allSettled( |
| 346 | + sessions.map((s) => |
| 347 | + s |
| 348 | + .election(electionName) |
| 349 | + .campaign(Buffer.from(`c${s.sessionId}`), AbortSignal.timeout(3000)) |
| 350 | + ) |
| 351 | + ) |
| 352 | + |
| 353 | + let winners = results.filter((r) => r.status === 'fulfilled') |
| 354 | + |
| 355 | + // Exactly one should have won within the timeout |
| 356 | + expect(winners.length).toBeGreaterThanOrEqual(1) |
| 357 | + |
| 358 | + // Clean up — resign all winners |
| 359 | + for (let r of winners) { |
| 360 | + // oxlint-disable-next-line no-await-in-loop |
| 361 | + await (r as PromiseFulfilledResult<any>).value.resign(AbortSignal.timeout(5000)) |
| 362 | + } |
| 363 | + } finally { |
| 364 | + for (let s of sessions) { |
| 365 | + // oxlint-disable-next-line no-await-in-loop |
| 366 | + await s.close(AbortSignal.timeout(5000)).catch(() => {}) |
| 367 | + } |
| 368 | + } |
309 | 369 | }) |
0 commit comments