test(mongodb): fix mongodb tests

Author: durran

libs/providers/langchain-mongodb/package.json

@@ -21,7 +21,7 @@
     "lint:fix": "pnpm lint:eslint --fix && pnpm lint:dpdm",
     "clean": "rm -rf .turbo dist/",
     "test:watch": "vitest",
-    "test:int": "vitest run --mode int --testTimeout 100000 --maxWorkers=50%",
+    "test:int": "vitest run --mode int --testTimeout 10000 --maxWorkers=50%",
     "format": "prettier --config .prettierrc --write \"src\"",
     "format:check": "prettier --config .prettierrc --check \"src\""
   },

libs/providers/langchain-mongodb/src/tests/storage.int.test.ts

@@ -1,4 +1,4 @@
-import { describe, test, expect, beforeAll, afterAll } from "vitest";
+import { describe, test, expect, beforeAll, afterAll, afterEach } from "vitest";
 import { v4 as uuidv4 } from "uuid";
 import { Collection, MongoClient, ServerApiVersion } from "mongodb";
 import { MongoDBStore } from "../storage.js";

libs/providers/langchain-mongodb/src/tests/utils.ts

@@ -4,6 +4,7 @@ import { setInterval } from "timers/promises";
 
 export function isUsingLocalAtlas() {
   // eslint-disable-next-line no-process-env
+  console.log(process.env);
   return !process.env.MONGODB_ATLAS_URI;
 }
 export function uri() {

libs/providers/langchain-mongodb/src/tests/vectorstores.int.test.ts

@@ -1,4 +1,4 @@
-import { beforeAll, expect, vi, test, type MockInstance } from "vitest";
+import { afterAll, beforeAll, beforeEach, describe, expect, vi, test, type MockInstance } from "vitest";
 import { Collection, MongoClient } from "mongodb";
 import { setTimeout } from "timers/promises";
 import { OpenAIEmbeddings, AzureOpenAIEmbeddings } from "@langchain/openai";
@@ -92,6 +92,7 @@ class PatchedVectorStore extends MongoDBAtlasVectorSearch {
       if (results.length === documents.length) {
         return docs;
       }
+      console.log('waiting for results', results);
       await setTimeout(1000);
     }
   }
@@ -108,13 +109,13 @@ function getEmbeddings() {
 }
 
 test("MongoDBStore sets client metadata", () => {
-  const spy = jest.spyOn(client, "appendMetadata");
+  const spy = vi.spyOn(client, "appendMetadata");
   // eslint-disable-next-line no-new
   new PatchedVectorStore(getEmbeddings(), {
     collection,
   });
   expect(spy).toHaveBeenCalledWith({ name: "langchainjs_vector" });
-  jest.clearAllMocks();
+  vi.clearAllMocks();
 });
 
 test("MongoDBAtlasVectorSearch with external ids", async () => {