Merge branch 'master' of github.com:signifly/laravel-pivot-events

.gitignore

@@ -1,2 +1,3 @@
 vendor
 composer.lock
+.phpunit.result.cache

src/BelongsToMany.php

@@ -28,6 +28,8 @@ public function attach($id, array $attributes = [], $touch = true)
 
         $this->parent->firePivotAttachedEvent();
 
+        $this->parent->resetPivotChanges();
+
         return $result;
     }
 
@@ -60,6 +62,8 @@ public function detach($ids = null, $touch = true)
 
         $this->parent->firePivotDetachedEvent();
 
+        $this->parent->resetPivotChanges();
+
         return $result;
     }
 
@@ -85,6 +89,8 @@ public function updateExistingPivot($id, array $attributes, $touch = true)
 
         $this->parent->firePivotUpdatedEvent();
 
+        $this->parent->resetPivotChanges();
+
         return $result;
     }
 }

src/HasPivotEvents.php

@@ -2,21 +2,22 @@
 
 namespace Signifly\PivotEvents;
 
+use Illuminate\Support\Collection;
 use Illuminate\Database\Eloquent\Model;
 use Illuminate\Database\Eloquent\Builder;
 
 trait HasPivotEvents
 {
     protected $pivotChanges = [];
 
-    public function setPivotChanges(string $type, string $relation, array $ids = [])
+    public function setPivotChanges(string $type, string $relation, array $ids = []): void
     {
         collect($ids)->each(function ($attributes, $id) use ($type, $relation) {
             data_set($this->pivotChanges, "{$type}.{$relation}.{$id}", $attributes);
         });
     }
 
-    public function getPivotChanges($type = null)
+    public function getPivotChanges($type = null): Collection
     {
         if ($type) {
             return collect(data_get($this->pivotChanges, $type));
@@ -25,11 +26,16 @@ public function getPivotChanges($type = null)
         return collect($this->pivotChanges);
     }
 
-    public function getPivotChangeIds($type, $relation)
+    public function getPivotChangeIds($type, $relation): Collection
     {
         return collect($this->getPivotChanges("{$type}.{$relation}"))->keys();
     }
 
+    public function resetPivotChanges(): void
+    {
+        $this->pivotChanges = [];
+    }
+
     public static function pivotAttaching($callback)
     {
         static::registerModelEvent('pivotAttaching', $callback);
@@ -98,15 +104,12 @@ public function firePivotUpdatedEvent($halt = false)
     public function getObservableEvents()
     {
         return array_merge(
+            parent::getObservableEvents(),
             [
-                'retrieved', 'creating', 'created', 'updating', 'updated',
-                'saving', 'saved', 'restoring', 'restored',
-                'deleting', 'deleted', 'forceDeleted',
                 'pivotAttaching', 'pivotAttached',
                 'pivotDetaching', 'pivotDetached',
                 'pivotUpdating', 'pivotUpdated',
-            ],
-            $this->observables
+            ]
         );
     }
 
@@ -123,9 +126,16 @@ public function getObservableEvents()
      * @param  string  $relationName
      * @return \Illuminate\Database\Eloquent\Relations\BelongsToMany
      */
-    protected function newBelongsToMany(Builder $query, Model $parent, $table, $foreignPivotKey, $relatedPivotKey,
-                                        $parentKey, $relatedKey, $relationName = null)
-    {
+    protected function newBelongsToMany(
+        Builder $query,
+        Model $parent,
+        $table,
+        $foreignPivotKey,
+        $relatedPivotKey,
+        $parentKey,
+        $relatedKey,
+        $relationName = null
+    ) {
         return new BelongsToMany($query, $parent, $table, $foreignPivotKey, $relatedPivotKey, $parentKey, $relatedKey, $relationName);
     }
 }

tests/PivotEventsTest.php

@@ -99,6 +99,31 @@ public function it_dispatches_pivot_update_events()
         Event::assertDispatched('eloquent.pivotUpdated: '.User::class);
     }
 
+    /** @test */
+    public function it_resets_pivot_changes_after_an_event_has_been_dispatched()
+    {
+        $user = $this->createUser();
+        $roleA = $this->createRole(['name' => 'role-a']);
+        $roleB = $this->createRole(['name' => 'role-b']);
+
+        $calledTimes = 0;
+
+        User::pivotAttached(function ($model) use (&$calledTimes) {
+            $this->assertCount(1, $model->getPivotChanges());
+            $calledTimes++;
+        });
+
+        $user->roles()->attach($roleA->id);
+        $this->assertCount(0, $user->getPivotChanges());
+
+        $user->roles()->attach($roleB->id);
+        $this->assertCount(0, $user->getPivotChanges());
+
+        $this->assertCount(2, $user->roles);
+
+        $this->assertEquals(2, $calledTimes);
+    }
+
     protected function createRole(array $overwrites = [])
     {
         return Role::create(array_merge([