Merge pull request #14295 from Automattic/IslandRhythms/gh-14279

add type for createCollections
Automattic · Jan 25, 2024 · 33c624e · 33c624e
2 parents 6b3f998 + 56f0515
commit 33c624e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/test/types/connection.test.ts b/test/types/connection.test.ts
@@ -20,6 +20,10 @@ conn.readyState === 99;
 
 expectError(conn.readyState = 0);
 
+expectType<Promise<Record<string, Error | mongodb.Collection<any>>>>(
+  conn.createCollections()
+);
+
 expectType<Connection>(new Connection());
 expectType<Promise<Connection>>(new Connection().asPromise());
 

diff --git a/types/connection.d.ts b/types/connection.d.ts
@@ -81,6 +81,11 @@ declare module 'mongoose' {
      */
     createCollection<T extends AnyObject = AnyObject>(name: string, options?: mongodb.CreateCollectionOptions): Promise<mongodb.Collection<T>>;
 
+    /**
+     * https://mongoosejs.com/docs/api/connection.html#Connection.prototype.createCollections()
+     */
+    createCollections(continueOnError?: boolean): Promise<Record<string, Error | mongodb.Collection<any>>>;
+
     /**
      * Removes the model named `name` from this connection, if it exists. You can
      * use this function to clean up any models you created in your tests to