From cd603e8c78f24422dcad4c65e9ca22eed94aa55e Mon Sep 17 00:00:00 2001
From: Eric Adum <eric.adum@mongodb.com>
Date: Fri, 3 Sep 2021 10:32:57 -0400
Subject: [PATCH] fix(NODE-3467): allow object type for aggregate out helper
 (#2971)

---
 src/cursor/aggregation_cursor.ts |  2 +-
 test/types/mongodb.test-d.ts     | 10 ++++++++--
 2 files changed, 9 insertions(+), 3 deletions(-)

diff --git a/src/cursor/aggregation_cursor.ts b/src/cursor/aggregation_cursor.ts
index 6efae2035c..ffe658e34e 100644
--- a/src/cursor/aggregation_cursor.ts
+++ b/src/cursor/aggregation_cursor.ts
@@ -121,7 +121,7 @@ export class AggregationCursor<TSchema = Document> extends AbstractCursor<TSchem
   }
 
   /** Add an out stage to the aggregation pipeline */
-  out($out: string): this {
+  out($out: { db: string; coll: string } | string): this {
     assertUninitialized(this);
     this[kPipeline].push({ $out });
     return this;
diff --git a/test/types/mongodb.test-d.ts b/test/types/mongodb.test-d.ts
index c84f100b4f..bd16ce6795 100644
--- a/test/types/mongodb.test-d.ts
+++ b/test/types/mongodb.test-d.ts
@@ -37,5 +37,11 @@ expectType<string | null>(await composedMap.next());
 expectType<string[]>(await composedMap.toArray());
 
 const builtCursor = coll.aggregate();
-expectType<AggregationCursor<Document>>(builtCursor.out('string')); // should allow string values for the out helper
-expectError(builtCursor.out(1)); // should error on non-string values
+// should allow string values for the out helper
+expectType<AggregationCursor<Document>>(builtCursor.out('collection'));
+// should also allow an object specifying db/coll (as of MongoDB 4.4)
+expectType<AggregationCursor<Document>>(builtCursor.out({ db: 'db', coll: 'collection' }));
+// should error on other object shapes
+expectError(builtCursor.out({ other: 'shape' }));
+// should error on non-object, non-string values
+expectError(builtCursor.out(1));