Added unsetRelation method to Eloquent models

laravel · Jun 6, 2018 · 8d8fc13 · 8d8fc13
1 parent 4b736f7
commit 8d8fc13
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/Illuminate/Database/Eloquent/Concerns/HasRelationships.php b/src/Illuminate/Database/Eloquent/Concerns/HasRelationships.php
@@ -706,6 +706,21 @@ public function setRelation($relation, $value)
         return $this;
     }
 
+    /**
+     * Unset an existing relation.
+     *
+     * @param  string  $relation
+     * @return $this
+     */
+    public function unsetRelation($relation)
+    {
+        if ($this->relationLoaded($relation)) {
+            unset($this->relations[$relation]);
+        }
+
+        return $this;
+    }
+
     /**
      * Set the entire relations array on the model.
      *

diff --git a/tests/Database/DatabaseEloquentRelationTest.php b/tests/Database/DatabaseEloquentRelationTest.php
@@ -25,6 +25,15 @@ public function testSetRelationFail()
         $this->assertArrayNotHasKey('foo', $parent->toArray());
     }
 
+    public function testUnsetExistingRelation() 
+    {
+        $parent = new EloquentRelationResetModelStub;
+        $relation = new EloquentRelationResetModelStub;
+        $parent->setRelation('foo', $relation);
+        $parent->unsetRelation('foo');
+        $this->assertFalse($parent->relationLoaded('foo'));
+    }
+
     public function testTouchMethodUpdatesRelatedTimestamps()
     {
         $builder = m::mock(Builder::class);