Merge pull request #37 from Weebly/fix-original

Properly unserialize `getOriginal` values
Weebly · Oct 16, 2024 · 04d6a1d · 04d6a1d
2 parents 7ceb9b6 + 3384db1
commit 04d6a1d
Show file tree

Hide file tree

Showing 2 changed files with 65 additions and 1 deletion.
diff --git a/src/Database/Model.php b/src/Database/Model.php
@@ -54,6 +54,30 @@ public function attributesToArray()
         return $attributes;
     }
 
+    /**
+     * {@inheritdoc}
+     */
+    public function getOriginal($key = null, $default = null)
+    {
+        $original = parent::getOriginal($key, $default);
+
+        if ($key) {
+            if ($this->hasMutator($key)) {
+                return $this->unserializeAttribute($key, $original);
+            }
+
+            return $original;
+        }
+
+        foreach ($original as $attribute => $value) {
+            if ($this->hasMutator($attribute)) {
+                $original[$attribute] = $this->unserializeAttribute($attribute, $value);
+            }
+        }
+
+        return $original;
+    }
+
     /**
      * Define a many-to-many relationship.
      *

diff --git a/tests/Unit/Database/Traits/HasMutatorsTest.php b/tests/Unit/Database/Traits/HasMutatorsTest.php
@@ -31,7 +31,6 @@ public function testSerializeAttribute()
 
     public function testUnserializeAttribute()
     {
-        $uuid = 'cf98906e-9074-11e7-9c8e-437b4bab8527';
         $mutator = M::mock(MutatorContract::class)
             ->shouldReceive('get')
             ->with('test_mutator')
@@ -52,6 +51,47 @@ public function testUnserializeAttribute()
         $this->assertEquals('serialized_attribute', $model->id);
     }
 
+    public function testGetOriginal()
+    {
+        $mutator = M::mock(MutatorContract::class)
+            ->shouldReceive('get')
+            ->with('test_mutator')
+            ->andReturnSelf()
+            ->once()
+            ->shouldReceive('unserializeAttribute')
+            ->with('unserialized_attribute')
+            ->andReturn('serialized_attribute')
+            ->once()
+            ->getMock();
+
+        app()['mutator'] = $mutator;
+
+        $model = new SampleModel();
+        $original = $model->getOriginal();
+
+        $this->assertIsArray($original);
+        $this->assertEquals('serialized_attribute', $original['id']);
+    }
+
+    public function testGetOriginalProperty()
+    {
+        $mutator = M::mock(MutatorContract::class)
+            ->shouldReceive('get')
+            ->with('test_mutator')
+            ->andReturnSelf()
+            ->once()
+            ->shouldReceive('unserializeAttribute')
+            ->with('unserialized_attribute')
+            ->andReturn('serialized_attribute')
+            ->once()
+            ->getMock();
+
+        app()['mutator'] = $mutator;
+
+        $model = new SampleModel();
+        $this->assertEquals('serialized_attribute', $model->getOriginal('id'));
+    }
+
     public function testGetMutators()
     {
         $this->assertEquals(['id' => 'test_mutator'], (new SampleModel())->getMutators());