Fix decorating (#16)

Decorating wasn't setting an array correctly...
rickselby · May 14, 2018 · 056bb1d · 056bb1d
1 parent cef315c
commit 056bb1d
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/src/AutoPresenterMapper.php b/src/AutoPresenterMapper.php
@@ -38,7 +38,13 @@ public function map($class, $presenter = null)
      */
     public function decorate($class)
     {
-        $this->map($class, null);
+        if (is_array($class)) {
+            foreach ($class as $model) {
+                $this->mappings->put($model, null);
+            }
+        } else {
+            $this->mappings->put($class, null);
+        }
     }
 
     /**

diff --git a/tests/AutoPresenterMapperTest.php b/tests/AutoPresenterMapperTest.php
@@ -61,6 +61,12 @@ public function testDecoratingReturnsNoClass()
         $this->assertNull($this->autoPresenterMapper->getPresenter(new MappedStub()));
     }
 
+    public function testDecoratingAsArray()
+    {
+        $this->setDecoratingAsArray();
+        $this->assertTrue($this->autoPresenterMapper->hasPresenter(new MappedStub()));
+    }
+
     protected function getMappedStub()
     {
         return new MappedStub();
@@ -85,4 +91,9 @@ protected function setDecorating()
     {
         $this->autoPresenterMapper->decorate(MappedStub::class);
     }
+
+    protected function setDecoratingAsArray()
+    {
+        $this->autoPresenterMapper->decorate([MappedStub::class]);
+    }
 }