Merge pull request #12001 from hasezoey/extendTypesTestFor11960

Extend types test for #11960
Automattic · Jul 1, 2022 · 39f651c · 39f651c
2 parents fec4d6a + b6b0973
commit 39f651c
Showing 1 changed file with 26 additions and 9 deletions.
diff --git a/test/types/document.test.ts b/test/types/document.test.ts
@@ -227,14 +227,31 @@ async function gh11960() {
 
   const ParentModel = model<DocumentType<Parent>>('Parent', ParentSchema);
 
-  const doc = new ParentModel({
-    username: 'user1',
-    map: { key1: 'value1', key2: 'value2' },
-    nested: { dummy: 'hello' },
-    nestedArray: [{ dummy: 'hello again' }]
-  });
-
-  expectType<Map<string, string> | undefined>(doc.map);
-  doc.nested!.parent();
+  {
+    const doc = new ParentModel({
+      username: 'user1',
+      map: { key1: 'value1', key2: 'value2' },
+      nested: { dummy: 'hello' },
+      nestedArray: [{ dummy: 'hello again' }]
+    });
+
+    expectType<Document<any, any, any> & Parent & { _id: Types.ObjectId }>(doc);
+    expectType<Map<string, string> | undefined>(doc.map);
+    doc.nested!.parent();
+    doc.nestedArray?.[0].parentArray();
+  }
 
+  {
+    const doc = await ParentModel.create({
+      username: 'user1',
+      map: { key1: 'value1', key2: 'value2' },
+      nested: { dummy: 'hello' },
+      nestedArray: [{ dummy: 'hello again' }]
+    });
+
+    expectType<Document<any, any, any> & Parent & { _id: Types.ObjectId }>(doc);
+    expectType<Map<string, string> | undefined>(doc.map);
+    doc.nested!.parent();
+    doc.nestedArray?.[0].parentArray();
+  }
 }