The engine mapIds method must return Collection object

Author: babenkoivan

src/ElasticEngine.php

@@ -262,7 +262,7 @@ public function searchRaw(Model $model, $query)
      */
     public function mapIds($results)
     {
-        return array_pluck($results['hits']['hits'], '_id');
+        return collect($results['hits']['hits'])->pluck('_id');
     }
 
     /**
@@ -284,7 +284,7 @@ public function map($results, $model)
             $columns[] = $primaryKey;
         }
 
-        $ids = $this->mapIds($results);
+        $ids = $this->mapIds($results)->all();
 
         $builder = $model->usesSoftDelete() ? $model->withTrashed() : $model->newQuery();
 

tests/ElasticEngineTest.php

@@ -300,7 +300,7 @@ public function testMapIds()
 
         $this->assertEquals(
             [1, 2],
-            $this->engine->mapIds($results)
+            $this->engine->mapIds($results)->all()
         );
     }