chore: sleep between search retries

Author: kmruiz

tests/integration/helpers.ts

@@ -382,3 +382,7 @@ export function getDataFromUntrustedContent(content: string): string {
     }
     return match.groups.data.trim();
 }
+
+export function sleep(ms: number): Promise<void> {
+    return new Promise((resolve) => setTimeout(resolve, ms));
+}

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

@@ -1,13 +1,9 @@
 import type { Session } from "../../../../src/common/session.js";
-import { expectDefined, getDataFromUntrustedContent, getResponseElements } from "../../helpers.js";
+import { expectDefined, getDataFromUntrustedContent, getResponseElements, sleep } from "../../helpers.js";
 import { describeWithAtlas, withProject, randomId, parseTable } from "./atlasHelpers.js";
 import type { ClusterDescription20240805 } from "../../../../src/common/atlas/openapi.js";
 import { afterAll, beforeAll, describe, expect, it } from "vitest";
 
-function sleep(ms: number): Promise<void> {
-    return new Promise((resolve) => setTimeout(resolve, ms));
-}
-
 async function deleteCluster(
     session: Session,
     projectId: string,

tests/integration/tools/mongodb/search/listSearchIndexes.test.ts

@@ -6,6 +6,7 @@ import {
     validateToolMetadata,
     validateThrowsForInvalidArguments,
     databaseCollectionInvalidArgs,
+    sleep,
 } from "../../../helpers.js";
 import type { NodeDriverServiceProvider } from "@mongosh/service-provider-node-driver";
 import type { SearchIndexStatus } from "../../../../../src/tools/mongodb/search/listSearchIndexes.js";
@@ -102,16 +103,18 @@ describeWithMongoDB(
     { search: true } // use a search cluster
 );
 
+const SEARCH_RETRIES = 10_000;
 async function waitUntilSearchIsReady(provider: NodeDriverServiceProvider, abortSignal: AbortSignal): Promise<void> {
     let success = true;
-    for (let i = 0; i < 1000 && !abortSignal.aborted; i++) {
+    for (let i = 0; i < SEARCH_RETRIES && !abortSignal.aborted; i++) {
         try {
             await provider.insertOne("tmp", "test", { field1: "yay" });
             await provider.createSearchIndexes("tmp", "test", [{ definition: { mappings: { dynamic: true } } }]);
             success = true;
             break;
         } catch (err) {
             void err;
+            await sleep(100);
         }
     }
 
@@ -128,7 +131,7 @@ async function waitUntilIndexIsQueryable(
     abortSignal: AbortSignal
 ): Promise<void> {
     let success = false;
-    for (let i = 0; i < 1000 && !abortSignal.aborted; i++) {
+    for (let i = 0; i < SEARCH_RETRIES && !abortSignal.aborted; i++) {
         try {
             const [indexStatus] = await provider.getSearchIndexes(database, collection, indexName);
             if (indexStatus?.queryable === true) {
@@ -137,6 +140,7 @@ async function waitUntilIndexIsQueryable(
             }
         } catch (err) {
             void err;
+            await sleep(100);
         }
     }