Populating nested mutations with other relationships for @update

src/Execution/MutationExecutor.php

@@ -147,6 +147,10 @@ protected static function saveModelWithBelongsTo(Model $model, Collection $args,
                     $belongsTo = $model->{$relationName}();
                     $belongsTo->associate($values);
                 }
+
+                if ($operationKey === 'delete') {
+                    $relation->getModel()::destroy($values);
+                }
             });
         });
 
@@ -212,6 +216,16 @@ public static function executeUpdate(Model $model, Collection $args, ?HasMany $p
 
         [$hasMany, $remaining] = self::partitionArgsByRelationType($reflection, $args, HasMany::class);
 
+        [$morphMany, $remaining] = self::partitionArgsByRelationType($reflection, $remaining, MorphMany::class);
+
+        [$hasOne, $remaining] = self::partitionArgsByRelationType($reflection, $remaining, HasOne::class);
+
+        [$belongsToMany, $remaining] = self::partitionArgsByRelationType($reflection, $remaining, BelongsToMany::class);
+
+        [$morphOne, $remaining] = self::partitionArgsByRelationType($reflection, $remaining, MorphOne::class);
+
+        [$morphToMany, $remaining] = self::partitionArgsByRelationType($reflection, $remaining, MorphToMany::class);
+
         $model = self::saveModelWithBelongsTo($model, $remaining, $parentRelation);
 
         $hasMany->each(function (array $nestedOperations, string $relationName) use ($model): void {
@@ -235,6 +249,113 @@ public static function executeUpdate(Model $model, Collection $args, ?HasMany $p
             });
         });
 
+        $hasOne->each(function (array $nestedOperations, string $relationName) use ($model): void {
+            /** @var \Illuminate\Database\Eloquent\Relations\HasOne $relation */
+            $relation = $model->{$relationName}();
+
+            collect($nestedOperations)->each(function ($values, string $operationKey) use ($relation): void {
+                if ($operationKey === 'create') {
+                    self::handleSingleRelationCreate(collect($values), $relation);
+                }
+
+                if ($operationKey === 'update') {
+                    collect($values)->each(function ($singleValues) use ($relation): void {
+                        self::executeUpdate($relation->getModel()->newInstance(), collect($singleValues), $relation);
+                    });
+                }
+
+                if ($operationKey === 'delete') {
+                    $relation->getModel()::destroy($values);
+                }
+            });
+        });
+
+        $morphMany->each(function (array $nestedOperations, string $relationName) use ($model): void {
+            /** @var \Illuminate\Database\Eloquent\Relations\MorphMany $relation */
+            $relation = $model->{$relationName}();
+
+            collect($nestedOperations)->each(function ($values, string $operationKey) use ($relation): void {
+                if ($operationKey === 'create') {
+                    self::handleMultiRelationCreate(collect($values), $relation);
+                }
+                if ($operationKey === 'update') {
+                    collect($values)->each(function ($singleValues) use ($relation): void {
+                        self::executeUpdate($relation->getModel()->newInstance(), collect($singleValues), $relation);
+                    });
+                }
+
+                if ($operationKey === 'delete') {
+                    $relation->getModel()::destroy($values);
+                }
+            });
+        });
+
+        $morphOne->each(function (array $nestedOperations, string $relationName) use ($model): void {
+            /** @var \Illuminate\Database\Eloquent\Relations\MorphOne $relation */
+            $relation = $model->{$relationName}();
+
+            collect($nestedOperations)->each(function ($values, string $operationKey) use ($relation): void {
+                if ($operationKey === 'create') {
+                    self::handleSingleRelationCreate(collect($values), $relation);
+                }
+
+                if ($operationKey === 'update') {
+                    collect($values)->each(function ($singleValues) use ($relation): void {
+                        self::executeUpdate($relation->getModel()->newInstance(), collect($singleValues), $relation);
+                    });
+                }
+
+                if ($operationKey === 'delete') {
+                    $relation->getModel()::destroy($values);
+                }
+            });
+        });
+
+        $belongsToMany->each(function (array $nestedOperations, string $relationName) use ($model): void {
+            /** @var \Illuminate\Database\Eloquent\Relations\BelongsToMany $relation */
+            $relation = $model->{$relationName}();
+
+            collect($nestedOperations)->each(function ($values, string $operationKey) use ($relation): void {
+                if ($operationKey === 'create') {
+                    self::handleMultiRelationCreate(collect($values), $relation);
+                }
+
+                if ($operationKey === 'update') {
+                    collect($values)->each(function ($singleValues) use ($relation): void {
+                        self::executeUpdate($relation->getModel()->newInstance(), collect($singleValues), $relation);
+                    });
+                }
+
+                if ($operationKey === 'delete') {
+                    $relation->getModel()::destroy($values);
+                }
+
+                if ($operationKey === 'connect') {
+                    $relation->attach($values);
+                }
+            });
+        });
+
+        $morphToMany->each(function (array $nestedOperations, string $relationName) use ($model): void {
+            /** @var \Illuminate\Database\Eloquent\Relations\HasMany $relation */
+            $relation = $model->{$relationName}();
+
+            collect($nestedOperations)->each(function ($values, string $operationKey) use ($relation): void {
+                if ($operationKey === 'create') {
+                    self::handleMultiRelationCreate(collect($values), $relation);
+                }
+                if ($operationKey === 'update') {
+                    collect($values)->each(function ($singleValues) use ($relation): void {
+                        self::executeUpdate($relation->getModel()->newInstance(), collect($singleValues), $relation);
+                    });
+                }
+
+                if ($operationKey === 'delete') {
+                    $relation->getModel()::destroy($values);
+                }
+            });
+        });
+
         return $model;
     }
 

tests/Integration/Schema/Directives/Fields/UpdateDirectiveTest.php → tests/Integration/Schema/Directives/Fields/UpdateDirectiveTests/CoreTest.php

@@ -1,14 +1,14 @@
 <?php
 
-namespace Tests\Integration\Schema\Directives\Fields;
+namespace Tests\Integration\Schema\Directives\Fields\UpdateDirectiveTests;
 
 use Tests\DBTestCase;
 use Tests\Utils\Models\Task;
 use Tests\Utils\Models\User;
 use Tests\Utils\Models\Company;
 use Tests\Utils\Models\Category;
 
-class UpdateDirectiveTest extends DBTestCase
+class CoreTest extends DBTestCase
 {
     /**
      * @test
@@ -98,70 +98,6 @@ public function itCanUpdateFromInputObject(): void
         $this->assertSame('bar', Company::first()->name);
     }
 
-    /**
-     * @test
-     */
-    public function itCanUpdateWithBelongsTo(): void
-    {
-        factory(User::class, 2)->create();
-        factory(Task::class)->create([
-            'name' => 'bar',
-            'user_id' => 1,
-        ]);
-
-        $this->schema = '
-        type Task {
-            id: ID!
-            name: String!
-            user: User @belongsTo
-        }
-
-        type User {
-            id: ID
-        }
-
-        type Mutation {
-            updateTask(input: UpdateTaskInput!): Task @update(flatten: true)
-        }
-
-        input UpdateTaskInput {
-            id: ID!
-            name: String
-            user_id: ID
-        }
-        '.$this->placeholderQuery();
-
-        $this->query('
-        mutation {
-            updateTask(input: {
-                id: 1
-                name: "foo"
-                user_id: 2
-            }) {
-                id
-                name
-                user {
-                    id
-                }
-            }
-        }
-        ')->assertJson([
-            'data' => [
-                'updateTask' => [
-                    'id' => '1',
-                    'name' => 'foo',
-                    'user' => [
-                        'id' => '2',
-                    ],
-                ],
-            ],
-        ]);
-
-        $task = Task::first();
-        $this->assertSame('2', $task->user_id);
-        $this->assertSame('foo', $task->name);
-    }
-
     /**
      * @test
      */

tests/Integration/Schema/Directives/Fields/UpdateDirectiveTests/RelationshipTests/BelongsToManyTest.php

@@ -0,0 +1,12 @@
+<?php
+
+namespace Tests\Integration\Schema\Directives\Fields\UpdateDirectiveTests\RelationshipTests;
+
+use Tests\DBTestCase;
+use Illuminate\Support\Arr;
+use Tests\Utils\Models\User;
+
+class BelongsToManyTest extends DBTestCase
+{
+
+}

tests/Integration/Schema/Directives/Fields/UpdateDirectiveTests/RelationshipTests/BelongsToTest.php

@@ -0,0 +1,74 @@
+<?php
+
+namespace Tests\Integration\Schema\Directives\Fields\UpdateDirectiveTests\RelationshipTests;
+
+use Tests\DBTestCase;
+use Tests\Utils\Models\Task;
+use Tests\Utils\Models\User;
+
+class BelongsToTest extends DBTestCase
+{
+    /**
+     * @test
+     */
+    public function itCanUpdateWithBelongsTo()
+    {
+        factory(User::class, 2)->create();
+        factory(Task::class)->create([
+            'name' => 'bar',
+            'user_id' => 1,
+        ]);
+
+        $this->schema = '
+        type Task {
+            id: ID!
+            name: String!
+            user: User @belongsTo
+        }
+
+        type User {
+            id: ID
+        }
+
+        type Mutation {
+            updateTask(input: UpdateTaskInput!): Task @update(flatten: true)
+        }
+
+        input UpdateTaskInput {
+            id: ID!
+            name: String
+            user_id: ID
+        }
+        '.$this->placeholderQuery();
+
+        $this->query('
+        mutation {
+            updateTask(input: {
+                id: 1
+                name: "foo"
+                user_id: 2
+            }) {
+                id
+                name
+                user {
+                    id
+                }
+            }
+        }
+        ')->assertJson([
+            'data' => [
+                'updateTask' => [
+                    'id' => '1',
+                    'name' => 'foo',
+                    'user' => [
+                        'id' => '2'
+                    ]
+                ]
+            ]
+        ]);
+
+        $task = Task::first();
+        $this->assertSame('2', $task->user_id);
+        $this->assertSame('foo', $task->name);
+    }
+}

tests/Integration/Schema/Directives/Fields/UpdateDirectiveTests/RelationshipTests/HasManyTest.php

@@ -0,0 +1,12 @@
+<?php
+
+namespace Tests\Integration\Schema\Directives\Fields\UpdateDirectiveTests\RelationshipTests;
+
+use Tests\DBTestCase;
+use Illuminate\Support\Arr;
+use Tests\Utils\Models\Task;
+
+class HasManyTest extends DBTestCase
+{
+
+}

tests/Integration/Schema/Directives/Fields/UpdateDirectiveTests/RelationshipTests/HasOneTest.php

@@ -0,0 +1,13 @@
+<?php
+
+namespace Tests\Integration\Schema\Directives\Fields\UpdateDirectiveTests\RelationshipTests;
+
+use Tests\DBTestCase;
+use Illuminate\Support\Arr;
+use Tests\Utils\Models\Post;
+use Tests\Utils\Models\User;
+
+class HasOneTest extends DBTestCase
+{
+
+}

tests/Integration/Schema/Directives/Fields/UpdateDirectiveTests/RelationshipTests/MorphManyTest.php

@@ -0,0 +1,12 @@
+<?php
+
+namespace Tests\Integration\Schema\Directives\Fields\UpdateDirectiveTests\RelationshipTests;
+
+use Tests\DBTestCase;
+use Illuminate\Support\Arr;
+use Tests\Utils\Models\User;
+
+class MorphManyTest extends DBTestCase
+{
+
+}

tests/Integration/Schema/Directives/Fields/UpdateDirectiveTests/RelationshipTests/MorphOneTest.php

@@ -0,0 +1,12 @@
+<?php
+
+namespace Tests\Integration\Schema\Directives\Fields\UpdateDirectiveTests\RelationshipTests;
+
+use Tests\DBTestCase;
+use Illuminate\Support\Arr;
+use Tests\Utils\Models\User;
+
+class MorphOneTest extends DBTestCase
+{
+
+}

tests/Integration/Schema/Directives/Fields/UpdateDirectiveTests/RelationshipTests/MorphToTest.php

@@ -0,0 +1,12 @@
+<?php
+
+namespace Tests\Integration\Schema\Directives\Fields\UpdateDirectiveTests\RelationshipTests;
+
+use Tests\DBTestCase;
+use Illuminate\Support\Arr;
+use Tests\Utils\Models\Task;
+
+class MorphToTest extends DBTestCase
+{
+
+}