update unit type tests

Author: aditi-khare-mongoDB

src/mongo_types.ts

@@ -219,7 +219,7 @@ export type NotAcceptedFields<TSchema, FieldType> = {
 /** @public */
 export type OnlyFieldsOfType<TSchema, FieldType = any, AssignableType = FieldType> = IsAny<
   TSchema[keyof TSchema],
-  Record<string, FieldType>,
+  AssignableType extends FieldType ? Record<string, FieldType> : Record<string, AssignableType>,
   AcceptedFields<TSchema, FieldType, AssignableType> &
     NotAcceptedFields<TSchema, FieldType> &
     Record<string, AssignableType>
@@ -283,7 +283,7 @@ export type PullAllOperator<TSchema> = ({
 export type UpdateFilter<TSchema> = {
   $currentDate?: OnlyFieldsOfType<
     TSchema,
-    true | Date | Timestamp,
+    Date | Timestamp,
     true | { $type: 'date' | 'timestamp' }
   >;
   $inc?: OnlyFieldsOfType<TSchema, NumericType | undefined>;

test/integration/crud/find_and_modify.test.ts

@@ -330,35 +330,6 @@ describe('Collection (#findOneAnd...)', function () {
         });
       });
     });
-
-    describe('update filter', function () {
-      context('when $currentDate is provided', function () {
-        let client;
-        let db: Db;
-        let collection: Collection;
-
-        beforeEach(async function () {
-          client = this.configuration.newClient({ w: 1 });
-          await client.connect();
-          db = client.db(this.configuration.db);
-          collection = db.collection('test_coll');
-          await collection.insertOne({ a: 'c' });
-        });
-
-        afterEach(async function () {
-          await collection.drop();
-          await client.close();
-        });
-
-        it(`should support fields with value 'true'`, async function () {
-          await collection.findOneAndUpdate(
-            {},
-            { $set: { a: 1 }, $currentDate: { lastModified: true } },
-            { writeConcern: { w: 1 } }
-          );
-        });
-      });
-    });
   });
 
   describe('#findOneAndReplace', function () {

test/types/community/collection/updateX.test-d.ts

@@ -60,6 +60,12 @@ expectAssignable<UpdateFilter<Document>>({ $inc: { anyKeyWhatsoever: 2 } });
 // But this no longer asserts anything about what the original keys map to
 expectNotType<UpdateFilter<Document>>({ $inc: { numberField: '2' } });
 
+
+expectAssignable<UpdateFilter<Document>>({ $currentDate: { anyKeyWhatsoever:  { $type: 'timestamp' }} });
+expectAssignable<UpdateFilter<Document>>({ $currentDate: { anyKeyWhatsoever:  { $type: 'date' }} });
+expectAssignable<UpdateFilter<Document>>({ $currentDate: { anyKeyWhatsoever:  true } });
+expectNotType<UpdateFilter<Document>>({ $currentDate: { anyKeyWhatsoever:  'true' } });
+expectNotType<UpdateFilter<Document>>({ $currentDate: { anyKeyWhatsoever: { key2:  true }} });
 // collection.updateX tests
 const client = new MongoClient('');
 const db = client.db('test');

test/types/helper_types.test-d.ts

@@ -68,6 +68,7 @@ expectType<NotAcceptedFields<{ a: number; b: string; c: string }, number>>(notAc
 expectAssignable<OnlyFieldsOfType<{ a: number; b: string }, number>>({ a: 2 });
 expectAssignable<OnlyFieldsOfType<{ a: number; b: string }, string>>({ b: 'hello' });
 expectAssignable<OnlyFieldsOfType<{ a: number; b: string }, string, boolean>>({ b: true });
+expectAssignable<OnlyFieldsOfType<{ a: number; b: string }, string, boolean>>({ b: true });
 
 // test generic schema, essentially we expect nearly no safety here
 expectAssignable<OnlyFieldsOfType<Document, NumericType | undefined>>({ someKey: 2 });