Skip to content

Commit 724add9

Browse files
committed
fix tests
1 parent 4cc2a36 commit 724add9

File tree

1 file changed

+4
-6
lines changed

1 file changed

+4
-6
lines changed

src/telemetry/telemetry.ts

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -7,17 +7,13 @@ import { MACHINE_METADATA } from "./constants.js";
77
import { EventCache } from "./eventCache.js";
88
import { detectContainerEnv } from "../helpers/container.js";
99
import type { DeviceId } from "../helpers/deviceId.js";
10-
import { EventEmitter } from "stream";
10+
import { EventEmitter } from "events";
1111

1212
type EventResult = {
1313
success: boolean;
1414
error?: Error;
1515
};
1616

17-
async function timeout(promise: Promise<unknown>, ms: number): Promise<void> {
18-
await Promise.race([new Promise((resolve) => setTimeout(resolve, ms)), promise]);
19-
}
20-
2117
export interface TelemetryEvents {
2218
"events-emitted": [];
2319
"events-send-failed": [];
@@ -90,7 +86,9 @@ export class Telemetry {
9086

9187
public async close(): Promise<void> {
9288
this.isBufferingEvents = false;
93-
await timeout(this.emit([]), 5_000);
89+
90+
// Wait up to 5 seconds for events to be sent before closing, but don't throw if it times out
91+
await Promise.race([new Promise((resolve) => setTimeout(resolve, 5000)), this.emit([])]);
9492
}
9593

9694
/**

0 commit comments

Comments
 (0)