Add upsert tests (#3864)

Closes #3856.
FerretDB · Dec 18, 2023 · 82143ff · 82143ff
1 parent 39aee11
commit 82143ff
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/integration/findandmodify_compat_test.go b/integration/findandmodify_compat_test.go
@@ -690,6 +690,26 @@ func TestFindAndModifyCompatUpsertSet(t *testing.T) {
 				{"update", bson.D{{"$set", bson.D{{"_id", "int32"}, {"v", int32(2)}}}}},
 			},
 		},
+		"UpsertQueryOperatorEq": {
+			command: bson.D{
+				{"query", bson.D{{"_id", bson.D{{"$eq", "non-existent"}}}}},
+				{"upsert", true},
+				{"update", bson.D{{"$set", bson.D{{"new", "val"}}}}},
+			},
+			skip: "https://github.com/FerretDB/FerretDB/issues/3856",
+		},
+		"UpsertQueryOperatorMixed": {
+			command: bson.D{
+				{"query", bson.D{
+					{"_id", bson.D{{"$eq", "non-existent"}}},
+					{"v", bson.D{{"$lt", 43}}},
+					{"non_existent", int32(0)},
+				}},
+				{"upsert", true},
+				{"update", bson.D{{"$set", bson.D{{"new", "val"}}}}},
+			},
+			skip: "https://github.com/FerretDB/FerretDB/issues/3856",
+		},
 	}
 
 	testFindAndModifyCompat(t, testCases)

diff --git a/integration/update_field_compat_test.go b/integration/update_field_compat_test.go
@@ -1011,6 +1011,22 @@ func TestUpdateFieldCompatMixed(t *testing.T) {
 			update:     bson.D{{"$foo", bson.D{{"foo", int32(1)}}}},
 			resultType: emptyResult,
 		},
+		"UpsertQueryOperatorEq": {
+			filter:     bson.D{{"_id", bson.D{{"$eq", "non-existent"}}}},
+			update:     bson.D{{"$set", bson.D{{"new", "val"}}}},
+			updateOpts: options.Update().SetUpsert(true),
+			skip:       "https://github.com/FerretDB/FerretDB/issues/3856",
+		},
+		"UpsertQueryOperatorMixed": {
+			filter: bson.D{
+				{"_id", bson.D{{"$eq", "non-existent"}}},
+				{"v", bson.D{{"$lt", 43}}},
+				{"non_existent", int32(0)},
+			},
+			update:     bson.D{{"$set", bson.D{{"new", "val"}}}},
+			updateOpts: options.Update().SetUpsert(true),
+			skip:       "https://github.com/FerretDB/FerretDB/issues/3856",
+		},
 	}
 
 	testUpdateCompat(t, testCases)