fix: styles

Author: fmenezes

src/common/utils.ts

@@ -0,0 +1,3 @@
+export function sleep(ms: number) {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}

src/tools/atlas/metadata/connectCluster.ts

@@ -2,6 +2,7 @@ import { z } from "zod";
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
 import { AtlasToolBase } from "../atlasTool.js";
 import { ToolArgs, OperationType } from "../../tool.js";
+import { sleep } from "../../../common/utils.js";
 
 function generateSecurePassword(): string {
     // TODO: use a better password generator
@@ -67,13 +68,13 @@ export class ConnectClusterTool extends AtlasToolBase {
             },
         });
 
-        setTimeout(async () => {
+        void sleep(expiryMs).then(async () => {
             // disconnect after 12 hours
             if (this.session.serviceProvider) {
-                await this.session.serviceProvider?.close(true);
+                await this.session.serviceProvider.close(true);
                 this.session.serviceProvider = undefined;
             }
-        }, expiryMs);
+        });
 
         const connectionString =
             (cluster.connectionStrings.standardSrv || cluster.connectionStrings.standard || "").replace(

tests/integration/tools/atlas/atlasHelpers.ts

@@ -5,10 +5,6 @@ import { setupIntegrationTest, IntegrationTest } from "../../helpers.js";
 
 export type IntegrationTestFunction = (integration: IntegrationTest) => void;
 
-export function sleep(ms: number) {
-    return new Promise((resolve) => setTimeout(resolve, ms));
-}
-
 export function describeWithAtlas(name: string, fn: IntegrationTestFunction) {
     const testDefinition = () => {
         const integration = setupIntegrationTest();

tests/integration/tools/atlas/clusters.test.ts

@@ -1,7 +1,8 @@
 import { Session } from "../../../../src/session.js";
 import { expectDefined } from "../../helpers.js";
-import { describeWithAtlas, withProject, sleep, randomId } from "./atlasHelpers.js";
+import { describeWithAtlas, withProject, randomId } from "./atlasHelpers.js";
 import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
+import { sleep } from "../../../../src/common/utils.js";
 
 async function deleteAndWaitCluster(session: Session, projectId: string, clusterName: string) {
     await session.apiClient.deleteCluster({
@@ -145,10 +146,12 @@ describeWithAtlas("clusters", (integration) => {
                             groupId: projectId,
                         },
                     },
-                    body: [{
-                        comment: "MCP test",
-                        cidrBlock: "0.0.0.0/0"
-                    }]
+                    body: [
+                        {
+                            comment: "MCP test",
+                            cidrBlock: "0.0.0.0/0",
+                        },
+                    ],
                 });
             });