fix: add timeout tests

Author: gagik

packages/device-id/README.md

@@ -1,3 +1,3 @@
 # @mongodb-js/device-id
 
-Create a consistent, implementation-agnostic hash from a given raw machine ID. The machine ID should originate from `node-machine-id` or `native-machine-id` depending on the platform. The hash is generated using SHA-256 and is designed to be consistent with the Atlas CLI.
+Create a consistent, implementation-agnostic hash from a given raw machine ID resolution function. The machine ID resolution function should come from `node-machine-id` or `native-machine-id` depending on the platform. The hash is generated using SHA-256 and is designed to be consistent with the Atlas CLI.

packages/device-id/package.json

@@ -24,8 +24,11 @@
   "license": "Apache-2.0",
   "main": "dist/index.js",
   "exports": {
-    "require": "./dist/index.js",
-    "import": "./dist/.esm-wrapper.mjs"
+    ".": {
+      "require": "./dist/index.js",
+      "import": "./dist/.esm-wrapper.mjs",
+      "types": "./dist/index.d.ts"
+    }
   },
   "types": "./dist/index.d.ts",
   "scripts": {

packages/device-id/src/get-device-id.spec.ts

@@ -82,4 +82,72 @@ describe('getDeviceId', function () {
 
     expect(resultA).to.equal(resultB);
   });
+
+  it('handles timeout when getting machine id', async () => {
+    let timeoutId: NodeJS.Timeout;
+    const getMachineId = () =>
+      new Promise<string>((resolve) => {
+        timeoutId = setTimeout(() => resolve('delayed-id'), 10_000);
+      });
+
+    let errorCalled = false;
+    const result = await getDeviceId({
+      getMachineId,
+      isNodeMachineId: false,
+      onError: () => {
+        errorCalled = true;
+      },
+      timeout: 1,
+    }).value;
+
+    clearTimeout(timeoutId!);
+    expect(result).to.equal('unknown');
+  });
+
+  it('handles external promise resolution', async () => {
+    let timeoutId: NodeJS.Timeout;
+    const getMachineId = () =>
+      new Promise<string>((resolve) => {
+        timeoutId = setTimeout(() => resolve('delayed-id'), 10_000);
+      });
+
+    const { resolve, value } = getDeviceId({
+      getMachineId,
+      isNodeMachineId: false,
+    });
+
+    resolve('external-id');
+
+    const result = await value;
+
+    clearTimeout(timeoutId!);
+    expect(result).to.be.a('string');
+    expect(result).to.equal('external-id');
+    expect(result).to.not.equal('unknown');
+  });
+
+  it('handles external promise rejection', async () => {
+    let timeoutId: NodeJS.Timeout;
+    const getMachineId = () =>
+      new Promise<string>((resolve) => {
+        timeoutId = setTimeout(() => resolve('delayed-id'), 10_000);
+      });
+
+    const error = new Error('External rejection');
+
+    const { reject, value } = getDeviceId({
+      getMachineId,
+      isNodeMachineId: false,
+    });
+
+    reject(error);
+
+    clearTimeout(timeoutId!);
+    try {
+      await value;
+      expect.fail('Expected promise to be rejected');
+    } catch (e) {
+      expect(e).to.equal(error);
+    }
+  });
 });