File Explorer

1/*2 * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.3 * SPDX-License-Identifier: Apache-2.0.4 */5 6import * as cancel from "./cancel";7import {EventEmitter} from "events";8 9jest.setTimeout(10000);10 11test('Simple cancel test - after await', async () => {12    let controller : cancel.CancelController = new cancel.CancelController();13 14    let emptyPromise : Promise<void> = new Promise<void>((resolve, reject) => {15       controller.addListener(() => { resolve(); });16    });17 18    setTimeout(() => {controller.cancel();}, 1000);19 20    await emptyPromise;21 22    // @ts-ignore23    expect(controller.emitter.listenerCount(cancel.EVENT_NAME)).toEqual(0);24});25 26test('Simple cancel test - before await', async () => {27    let controller : cancel.CancelController = new cancel.CancelController();28    controller.cancel();29 30    let emptyPromise : Promise<void> = new Promise<void>((resolve, reject) => {31        controller.addListener(() => { resolve(); });32    });33 34    await emptyPromise;35 36    // @ts-ignore37    expect(controller.emitter.listenerCount(cancel.EVENT_NAME)).toEqual(0);38});39 40test('Cancellable next event promise - event resolve', async () => {41    const eventName : string = "DISASTER";42    let controller : cancel.CancelController = new cancel.CancelController();43    let emitter : EventEmitter = new EventEmitter();44 45    let nextEventPromise : Promise<string> = cancel.newCancellablePromiseFromNextEvent({46        cancelController: controller,47        emitter : emitter,48        eventName : eventName,49        eventDataTransformer : (value : any) => { return value.toString(); },50        cancelMessage : "Won't happen"51    });52 53    setTimeout(() => { emitter.emit(eventName, 1); });54 55    let result : string = await nextEventPromise;56    expect(result).toEqual("1");57 58    // @ts-ignore59    expect(controller.emitter.listenerCount(cancel.EVENT_NAME)).toEqual(0);60    expect(emitter.listenerCount(eventName)).toEqual(0);61});62 63test('Cancellable next event promise - cancel reject', async () => {64    const eventName : string = "DISASTER";65    let controller : cancel.CancelController = new cancel.CancelController();66    let emitter : EventEmitter = new EventEmitter();67 68    let nextEventPromise : Promise<string> = cancel.newCancellablePromiseFromNextEvent({69        cancelController: controller,70        emitter : emitter,71        eventName : eventName,72        eventDataTransformer : (value : any) => { return value.toString(); },73        cancelMessage : "Will happen"74    });75 76    setTimeout(() => { controller.cancel(); });77 78    await expect(nextEventPromise).rejects.toMatch("Will happen");79 80    // @ts-ignore81    expect(controller.emitter.listenerCount(cancel.EVENT_NAME)).toEqual(0);82    expect(emitter.listenerCount(eventName)).toEqual(0);83});