feat: implement updateManyAndReturn (#5098)

* feat: implement updateManyAndReturn * fix: correct update_one_without_selection * feat: add the mutation * test: update snapshot * test: update snapshot * test: add a test suite * test: add UpdateReturning capability * test: correct test suite attributes * fix: add missing case in From<&str> for QueryTag * test: test relational reads * chore: trigger integration * [integration]
prisma · Jan 2, 2025 · a510d27 · a510d27
1 parent aef02d5
commit a510d27
Show file tree

Hide file tree

Showing 20 changed files with 971 additions and 36 deletions.
diff --git a/prisma-fmt/src/get_dmmf.rs b/prisma-fmt/src/get_dmmf.rs
@@ -5533,6 +5533,50 @@ mod tests {
                             "isList": false
                           }
                         },
+                        {
+                          "name": "updateManyAAndReturn",
+                          "args": [
+                            {
+                              "name": "data",
+                              "isRequired": true,
+                              "isNullable": false,
+                              "inputTypes": [
+                                {
+                                  "type": "AUpdateManyMutationInput",
+                                  "namespace": "prisma",
+                                  "location": "inputObjectTypes",
+                                  "isList": false
+                                },
+                                {
+                                  "type": "AUncheckedUpdateManyInput",
+                                  "namespace": "prisma",
+                                  "location": "inputObjectTypes",
+                                  "isList": false
+                                }
+                              ]
+                            },
+                            {
+                              "name": "where",
+                              "isRequired": false,
+                              "isNullable": false,
+                              "inputTypes": [
+                                {
+                                  "type": "AWhereInput",
+                                  "namespace": "prisma",
+                                  "location": "inputObjectTypes",
+                                  "isList": false
+                                }
+                              ]
+                            }
+                          ],
+                          "isNullable": false,
+                          "outputType": {
+                            "type": "UpdateManyAAndReturnOutputType",
+                            "namespace": "model",
+                            "location": "outputObjectTypes",
+                            "isList": true
+                          }
+                        },
                         {
                           "name": "deleteManyA",
                           "args": [
@@ -5851,6 +5895,50 @@ mod tests {
                             "isList": false
                           }
                         },
+                        {
+                          "name": "updateManyBAndReturn",
+                          "args": [
+                            {
+                              "name": "data",
+                              "isRequired": true,
+                              "isNullable": false,
+                              "inputTypes": [
+                                {
+                                  "type": "BUpdateManyMutationInput",
+                                  "namespace": "prisma",
+                                  "location": "inputObjectTypes",
+                                  "isList": false
+                                },
+                                {
+                                  "type": "BUncheckedUpdateManyInput",
+                                  "namespace": "prisma",
+                                  "location": "inputObjectTypes",
+                                  "isList": false
+                                }
+                              ]
+                            },
+                            {
+                              "name": "where",
+                              "isRequired": false,
+                              "isNullable": false,
+                              "inputTypes": [
+                                {
+                                  "type": "BWhereInput",
+                                  "namespace": "prisma",
+                                  "location": "inputObjectTypes",
+                                  "isList": false
+                                }
+                              ]
+                            }
+                          ],
+                          "isNullable": false,
+                          "outputType": {
+                            "type": "UpdateManyBAndReturnOutputType",
+                            "namespace": "model",
+                            "location": "outputObjectTypes",
+                            "isList": true
+                          }
+                        },
                         {
                           "name": "deleteManyB",
                           "args": [
@@ -6399,6 +6487,42 @@ mod tests {
                         }
                       ]
                     },
+                    {
+                      "name": "UpdateManyAAndReturnOutputType",
+                      "fields": [
+                        {
+                          "name": "id",
+                          "args": [],
+                          "isNullable": false,
+                          "outputType": {
+                            "type": "String",
+                            "location": "scalar",
+                            "isList": false
+                          }
+                        },
+                        {
+                          "name": "b_id",
+                          "args": [],
+                          "isNullable": false,
+                          "outputType": {
+                            "type": "String",
+                            "location": "scalar",
+                            "isList": false
+                          }
+                        },
+                        {
+                          "name": "b",
+                          "args": [],
+                          "isNullable": false,
+                          "outputType": {
+                            "type": "B",
+                            "namespace": "model",
+                            "location": "outputObjectTypes",
+                            "isList": false
+                          }
+                        }
+                      ]
+                    },
                     {
                       "name": "CreateManyBAndReturnOutputType",
                       "fields": [
@@ -6413,6 +6537,21 @@ mod tests {
                           }
                         }
                       ]
+                    },
+                    {
+                      "name": "UpdateManyBAndReturnOutputType",
+                      "fields": [
+                        {
+                          "name": "id",
+                          "args": [],
+                          "isNullable": false,
+                          "outputType": {
+                            "type": "String",
+                            "location": "scalar",
+                            "isList": false
+                          }
+                        }
+                      ]
                     }
                   ]
                 },
@@ -6622,6 +6761,7 @@ mod tests {
                     "findUniqueOrThrow": "findUniqueAOrThrow",
                     "groupBy": "groupByA",
                     "updateMany": "updateManyA",
+                    "updateManyAndReturn": "updateManyAAndReturn",
                     "updateOne": "updateOneA",
                     "upsertOne": "upsertOneA"
                   },
@@ -6640,6 +6780,7 @@ mod tests {
                     "findUniqueOrThrow": "findUniqueBOrThrow",
                     "groupBy": "groupByB",
                     "updateMany": "updateManyB",
+                    "updateManyAndReturn": "updateManyBAndReturn",
                     "updateOne": "updateOneB",
                     "upsertOne": "upsertOneB"
                   }

diff --git a/...y-engine/connector-test-kit-rs/query-engine-tests/tests/writes/top_level_mutations/mod.rs b/...y-engine/connector-test-kit-rs/query-engine-tests/tests/writes/top_level_mutations/mod.rs
@@ -11,4 +11,5 @@ mod insert_null_in_required_field;
 mod non_embedded_upsert;
 mod update;
 mod update_many;
+mod update_many_and_return;
 mod upsert;