Skip to content

Commit f524a94

Browse files
alenakhineikaalenakhineika
authored
fix(NODE-5790): type error with $addToSet in bulkWrite (#3956)
1 parent d9f5d9f commit f524a94

File tree

5 files changed

+288
-11
lines changed

5 files changed

+288
-11
lines changed

‎src/bulk/common.ts

+12-4
Original file line numberDiff line numberDiff line change
@@ -85,8 +85,12 @@ export interface ReplaceOneModel<TSchema extends Document = Document> {
8585
export interface UpdateOneModel<TSchema extends Document = Document> {
8686
/** The filter to limit the updated documents. */
8787
filter: Filter<TSchema>;
88-
/** A document or pipeline containing update operators. */
89-
update: UpdateFilter<TSchema> | UpdateFilter<TSchema>[];
88+
/**
89+
* The modifications to apply. The value can be either:
90+
* UpdateFilter<TSchema> - A document that contains update operator expressions,
91+
* Document[] - an aggregation pipeline.
92+
*/
93+
update: UpdateFilter<TSchema> | Document[];
9094
/** A set of filters specifying to which array elements an update should apply. */
9195
arrayFilters?: Document[];
9296
/** Specifies a collation. */
@@ -101,8 +105,12 @@ export interface UpdateOneModel<TSchema extends Document = Document> {
101105
export interface UpdateManyModel<TSchema extends Document = Document> {
102106
/** The filter to limit the updated documents. */
103107
filter: Filter<TSchema>;
104-
/** A document or pipeline containing update operators. */
105-
update: UpdateFilter<TSchema> | UpdateFilter<TSchema>[];
108+
/**
109+
* The modifications to apply. The value can be either:
110+
* UpdateFilter<TSchema> - A document that contains update operator expressions,
111+
* Document[] - an aggregation pipeline.
112+
*/
113+
update: UpdateFilter<TSchema> | Document[];
106114
/** A set of filters specifying to which array elements an update should apply. */
107115
arrayFilters?: Document[];
108116
/** Specifies a collation. */

‎src/collection.ts

+13-5
Original file line numberDiff line numberDiff line change
@@ -344,13 +344,17 @@ export class Collection<TSchema extends Document = Document> {
344344
/**
345345
* Update a single document in a collection
346346
*
347+
* The value of `update` can be either:
348+
* - UpdateFilter<TSchema> - A document that contains update operator expressions,
349+
* - Document[] - an aggregation pipeline.
350+
*
347351
* @param filter - The filter used to select the document to update
348-
* @param update - The update operations to be applied to the document
352+
* @param update - The modifications to apply
349353
* @param options - Optional settings for the command
350354
*/
351355
async updateOne(
352356
filter: Filter<TSchema>,
353-
update: UpdateFilter<TSchema> | Partial<TSchema>,
357+
update: UpdateFilter<TSchema> | Document[],
354358
options?: UpdateOptions
355359
): Promise<UpdateResult<TSchema>> {
356360
return executeOperation(
@@ -390,13 +394,17 @@ export class Collection<TSchema extends Document = Document> {
390394
/**
391395
* Update multiple documents in a collection
392396
*
393-
* @param filter - The filter used to select the documents to update
394-
* @param update - The update operations to be applied to the documents
397+
* The value of `update` can be either:
398+
* - UpdateFilter<TSchema> - A document that contains update operator expressions,
399+
* - Document[] - an aggregation pipeline.
400+
*
401+
* @param filter - The filter used to select the document to update
402+
* @param update - The modifications to apply
395403
* @param options - Optional settings for the command
396404
*/
397405
async updateMany(
398406
filter: Filter<TSchema>,
399-
update: UpdateFilter<TSchema>,
407+
update: UpdateFilter<TSchema> | Document[],
400408
options?: UpdateOptions
401409
): Promise<UpdateResult<TSchema>> {
402410
return executeOperation(

‎src/mongo_types.ts

+5-1
Original file line numberDiff line numberDiff line change
@@ -237,7 +237,11 @@ export type SetFields<TSchema> = ({
237237
readonly [key in KeysOfAType<TSchema, ReadonlyArray<any> | undefined>]?:
238238
| OptionalId<Flatten<TSchema[key]>>
239239
| AddToSetOperators<Array<OptionalId<Flatten<TSchema[key]>>>>;
240-
} & NotAcceptedFields<TSchema, ReadonlyArray<any> | undefined>) & {
240+
} & IsAny<
241+
TSchema[keyof TSchema],
242+
object,
243+
NotAcceptedFields<TSchema, ReadonlyArray<any> | undefined>
244+
>) & {
241245
readonly [key: string]: AddToSetOperators<any> | any;
242246
};
243247

‎test/types/community/collection/bulkWrite.test-d.ts

+195-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import { expectError } from 'tsd';
22

3-
import { MongoClient, ObjectId } from '../../../mongodb';
3+
import { type Collection, type Document, MongoClient, ObjectId } from '../../../mongodb';
44

55
// TODO(NODE-3347): Improve these tests to use more expect assertions
66

@@ -297,3 +297,197 @@ collectionType.bulkWrite([
297297
}
298298
}
299299
]);
300+
301+
{
302+
// NODE-5647 - Type error with $addToSet in bulkWrite
303+
interface TestDocument {
304+
readonly myId: number;
305+
readonly mySet: number[];
306+
readonly myAny: any;
307+
}
308+
const collection = undefined as unknown as Collection<TestDocument>;
309+
expectError(
310+
collection.bulkWrite([
311+
{
312+
updateOne: {
313+
filter: { myId: 0 },
314+
update: {
315+
$addToSet: { mySet: 'value of other type' }
316+
}
317+
}
318+
}
319+
])
320+
);
321+
collection.bulkWrite([
322+
{
323+
updateOne: {
324+
filter: { myId: 0 },
325+
update: {
326+
$addToSet: { mySet: 0 }
327+
}
328+
}
329+
}
330+
]);
331+
collection.bulkWrite([
332+
{
333+
updateOne: {
334+
filter: { myId: 0 },
335+
update: {
336+
$addToSet: { myAny: 1 }
337+
}
338+
}
339+
}
340+
]);
341+
collection.bulkWrite([
342+
{
343+
updateOne: {
344+
filter: { myId: 0 },
345+
update: [
346+
{
347+
$addToSet: { myAny: 0 }
348+
}
349+
]
350+
}
351+
}
352+
]);
353+
expectError(
354+
collection.bulkWrite([
355+
{
356+
updateMany: {
357+
filter: { myId: 0 },
358+
update: {
359+
$addToSet: { mySet: 'value of other type' }
360+
}
361+
}
362+
}
363+
])
364+
);
365+
collection.bulkWrite([
366+
{
367+
updateMany: {
368+
filter: { myId: 0 },
369+
update: {
370+
$addToSet: { mySet: 0 }
371+
}
372+
}
373+
}
374+
]);
375+
collection.bulkWrite([
376+
{
377+
updateMany: {
378+
filter: { myId: 0 },
379+
update: {
380+
$addToSet: { myAny: 1 }
381+
}
382+
}
383+
}
384+
]);
385+
collection.bulkWrite([
386+
{
387+
updateMany: {
388+
filter: { myId: 0 },
389+
update: [
390+
{
391+
$addToSet: { mySet: 0 }
392+
}
393+
]
394+
}
395+
}
396+
]);
397+
398+
interface IndexSingatureTestDocument extends Document {
399+
readonly myId: number;
400+
readonly mySet: number[];
401+
}
402+
const indexSingatureCollection = undefined as unknown as Collection<IndexSingatureTestDocument>;
403+
indexSingatureCollection.bulkWrite([
404+
{
405+
updateOne: {
406+
filter: { myId: 0 },
407+
update: {
408+
$addToSet: { mySet: 0 }
409+
}
410+
}
411+
}
412+
]);
413+
indexSingatureCollection.bulkWrite([
414+
{
415+
updateOne: {
416+
filter: { myId: 0 },
417+
update: [
418+
{
419+
$addToSet: { mySet: 0 }
420+
}
421+
]
422+
}
423+
}
424+
]);
425+
indexSingatureCollection.bulkWrite([
426+
{
427+
updateMany: {
428+
filter: { myId: 0 },
429+
update: {
430+
$addToSet: { mySet: 0 }
431+
}
432+
}
433+
}
434+
]);
435+
indexSingatureCollection.bulkWrite([
436+
{
437+
updateMany: {
438+
filter: { myId: 0 },
439+
update: [
440+
{
441+
$addToSet: { mySet: 0 }
442+
}
443+
]
444+
}
445+
}
446+
]);
447+
448+
const collectionOfAnyType = undefined as unknown as Collection<any>;
449+
collectionOfAnyType.bulkWrite([
450+
{
451+
updateOne: {
452+
filter: { myId: 0 },
453+
update: {
454+
$addToSet: { mySet: 0 }
455+
}
456+
}
457+
}
458+
]);
459+
collectionOfAnyType.bulkWrite([
460+
{
461+
updateOne: {
462+
filter: { myId: 0 },
463+
update: [
464+
{
465+
$addToSet: { mySet: 0 }
466+
}
467+
]
468+
}
469+
}
470+
]);
471+
collectionOfAnyType.bulkWrite([
472+
{
473+
updateMany: {
474+
filter: { myId: 0 },
475+
update: {
476+
$addToSet: { mySet: 0 }
477+
}
478+
}
479+
}
480+
]);
481+
collectionOfAnyType.bulkWrite([
482+
{
483+
updateMany: {
484+
filter: { myId: 0 },
485+
update: [
486+
{
487+
$addToSet: { mySet: 0 }
488+
}
489+
]
490+
}
491+
}
492+
]);
493+
}

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

+63
Original file line numberDiff line numberDiff line change
@@ -439,3 +439,66 @@ export async function testPushWithId(): Promise<void> {
439439
collectionWithSchema.updateMany({}, {})
440440
);
441441
}
442+
443+
{
444+
// NODE-5647 - Type error with $addToSet in bulkWrite
445+
interface TestDocument {
446+
readonly myId: number;
447+
readonly mySet: number[];
448+
readonly myAny: any;
449+
}
450+
const collection = undefined as unknown as Collection<TestDocument>;
451+
expectError(collection.updateOne({ myId: 0 }, { $addToSet: { mySet: 'value of other type' } }));
452+
collection.updateOne({ myId: 0 }, { $addToSet: { mySet: 0 } });
453+
collection.updateOne({ myId: 0 }, { $addToSet: { myAny: 1 } });
454+
collection.updateOne({ myId: 0 }, [
455+
{
456+
$addToSet: { mySet: 0 }
457+
}
458+
]);
459+
expectError(collection.updateMany({ myId: 0 }, { $addToSet: { mySet: 'value of other type' } }));
460+
collection.updateMany({ myId: 0 }, { $addToSet: { mySet: 0 } });
461+
collection.updateMany({ myId: 0 }, { $addToSet: { myAny: 1 } });
462+
collection.updateMany({ myId: 0 }, [
463+
{
464+
$addToSet: { mySet: 0 }
465+
}
466+
]);
467+
468+
interface IndexSingatureTestDocument extends Document {
469+
readonly myId: number;
470+
readonly mySet: number[];
471+
readonly myAny: any;
472+
}
473+
const indexSingatureCollection = undefined as unknown as Collection<IndexSingatureTestDocument>;
474+
indexSingatureCollection.updateOne({ myId: 0 }, { $addToSet: { mySet: 0 } });
475+
indexSingatureCollection.updateOne({ myId: 0 }, { $addToSet: { myAny: 1 } });
476+
indexSingatureCollection.updateOne({ myId: 0 }, [
477+
{
478+
$addToSet: { mySet: 0 }
479+
}
480+
]);
481+
indexSingatureCollection.updateMany({ myId: 0 }, { $addToSet: { mySet: 0 } });
482+
indexSingatureCollection.updateMany({ myId: 0 }, { $addToSet: { myAny: 1 } });
483+
indexSingatureCollection.updateMany({ myId: 0 }, [
484+
{
485+
$addToSet: { mySet: 0 }
486+
}
487+
]);
488+
489+
const collectionOfAnyType = undefined as unknown as Collection<any>;
490+
collectionOfAnyType.updateOne({ myId: 0 }, { $addToSet: { mySet: 0 } });
491+
collectionOfAnyType.updateOne({ myId: 0 }, { $addToSet: { myAny: 1 } });
492+
collectionOfAnyType.updateOne({ myId: 0 }, [
493+
{
494+
$addToSet: { mySet: 0 }
495+
}
496+
]);
497+
collectionOfAnyType.updateMany({ myId: 0 }, { $addToSet: { mySet: 0 } });
498+
collectionOfAnyType.updateMany({ myId: 0 }, { $addToSet: { myAny: 1 } });
499+
collectionOfAnyType.updateMany({ myId: 0 }, [
500+
{
501+
$addToSet: { mySet: 0 }
502+
}
503+
]);
504+
}

0 commit comments

Comments
 (0)