Adding support for stdClass to be used as a map

ICanBoogie · Oct 23, 2018 · bbd39ea · bbd39ea
1 parent 5be831b
commit bbd39ea
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/lib/MainTransformation.php b/lib/MainTransformation.php
@@ -52,7 +52,7 @@ public function __invoke($data, callable $transformation = null)
 
         $transformation = $transformation ?: $this;
 
-        if (is_object($data)) {
+        if (is_object($data) && !$data instanceof \stdClass) {
             try {
                 return $this->resolveTransformation($data)($data, $transformation);
             } catch (TransformationNotFound $e) {
@@ -90,8 +90,14 @@ private function transformIterable($data, callable $transformation): array
     {
         $array = [];
 
-        foreach ($data as $item) {
-            $array[] = $this($item, $transformation);
+        if ($data instanceof \stdClass) {
+            foreach ($data as $key => $item) {
+                $array[$key] = $this($item, $transformation);
+            }
+        } else {
+            foreach ($data as $item) {
+                $array[] = $this($item, $transformation);
+            }
         }
 
         return $array;

diff --git a/tests/MainTransformationTest.php b/tests/MainTransformationTest.php
@@ -160,7 +160,9 @@ public function shouldCallTransformationWithSpecifiedTransformer($data, $expecte
     public function provideTransformData()
     {
         return [
-
+            [ ['abc' => 1], [1] ],
+            [ (object) [ 'abc' => 1 ], [ 'abc' => 1 ] ],
+            [ (object) [ 'abc' => new DataSample1 ], [ 'abc' => DataSample1::class ] ],
             [ new DataSample1, DataSample1::class ],
             [ [ new DataSample1, new DataSample2 ], [ DataSample1::class, DataSample2::class ] ],
             [ new \ArrayIterator([ new DataSample1, new DataSample2 ]), [ DataSample1::class, DataSample2::class ] ],