Add transformer test

Author: yajra

src/Commands/stubs/transformer.inc.stub

@@ -14,7 +14,7 @@ class DummyClass extends TransformerAbstract
      * @param \App\Dummy $dummy
      * @return array
      */
-    public function transform(Dummy $dummy)
+    public function transform(Dummy $dummy): array
     {
         return [
             'id' => (int) $dummy->id,

src/Commands/stubs/transformer.stub

@@ -11,7 +11,7 @@ class DummyClass extends TransformerAbstract
      * @param \App\Dummy $dummy
      * @return array
      */
-    public function transform(Dummy $dummy)
+    public function transform(Dummy $dummy): array
     {
         return [
             'id' => (int) $dummy->id,

src/Transformers/FractalTransformer.php

@@ -2,6 +2,7 @@
 
 namespace Yajra\DataTables\Transformers;
 
+use Illuminate\Support\Collection as LaravelCollection;
 use League\Fractal\Manager;
 use League\Fractal\Resource\Collection;
 use League\Fractal\Serializer\SerializerAbstract;
@@ -17,7 +18,7 @@ class FractalTransformer
     /**
      * FractalTransformer constructor.
      *
-     * @param \League\Fractal\Manager $fractal
+     * @param  \League\Fractal\Manager  $fractal
      */
     public function __construct(Manager $fractal)
     {
@@ -27,24 +28,27 @@ public function __construct(Manager $fractal)
     /**
      * Transform output using the given transformer and serializer.
      *
-     * @param  array  $output
+     * @param  array|\Illuminate\Support\Collection  $output
      * @param  iterable  $transformer
      * @param  SerializerAbstract|null  $serializer
      * @return array
      */
-    public function transform(array $output, iterable $transformer, SerializerAbstract $serializer = null): array
-    {
+    public function transform(
+        array|LaravelCollection $output,
+        iterable $transformer,
+        SerializerAbstract $serializer = null
+    ): array {
         if ($serializer !== null) {
             $this->fractal->setSerializer($this->createSerializer($serializer));
         }
 
         $collector = [];
         foreach ($transformer as $transform) {
             if ($transform != null) {
-                $resource       = new Collection($output, $this->createTransformer($transform));
-                $collection     = $this->fractal->createData($resource)->toArray();
-                $transformed    = $collection['data'] ?? $collection;
-                $collector      = array_map(
+                $resource = new Collection($output, $this->createTransformer($transform));
+                $collection = $this->fractal->createData($resource)->toArray();
+                $transformed = $collection['data'] ?? $collection;
+                $collector = array_map(
                     function ($item_collector, $item_transformed) {
                         if (! is_array($item_collector)) {
                             $item_collector = [];

tests/FractalTest.php

@@ -5,6 +5,7 @@
 
 use Illuminate\Foundation\Testing\DatabaseTransactions;
 use Yajra\DataTables\Fractal\Tests\Models\User;
+use Yajra\DataTables\Fractal\Tests\Transformers\UserTransformer;
 
 class FractalTest extends TestCase
 {
@@ -13,21 +14,26 @@ class FractalTest extends TestCase
     /** @test */
     public function it_can_transform_response()
     {
-        $response = $this->getAjax('/users');
+        $json = $this->getAjax('/users');
 
-        $response->assertJson([
+        $json->assertJson([
             'draw' => 0,
             'recordsTotal' => 20,
             'recordsFiltered' => 20,
         ]);
+
+        $this->assertIsInt($json['data'][0]['id']);
+        $this->assertIsString($json['data'][0]['name']);
     }
 
     protected function setUp(): void
     {
         parent::setUp();
 
         $this->app['router']->get('/users', function () {
-            return datatables()->eloquent(User::query())->toJson();
+            return datatables()->eloquent(User::query())
+                               ->setTransformer(UserTransformer::class)
+                               ->toJson();
         });
     }
 }

tests/Transformers/UserTransformer.php

@@ -0,0 +1,21 @@
+<?php
+
+namespace Yajra\DataTables\Fractal\Tests\Transformers;
+
+use League\Fractal\TransformerAbstract;
+use Yajra\DataTables\Fractal\Tests\Models\User;
+
+class UserTransformer extends TransformerAbstract
+{
+    /**
+     * @param  \Yajra\DataTables\Fractal\Tests\Models\User  $user
+     * @return array
+     */
+    public function transform(User $user): array
+    {
+        return [
+            'id' => (int) $user->id,
+            'name' => $user->name,
+        ];
+    }
+}