Merge branch 'kronthto/feature/search-callback' (ErickTamayo#56) into…

… extended
dieMAYREI · May 16, 2017 · ad5ffaa · ad5ffaa
2 parents 1cb09c3 + 9dbde2d
commit ad5ffaa
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/src/ElasticsearchEngine.php b/src/ElasticsearchEngine.php
@@ -154,6 +154,15 @@ protected function performSearch(Builder $builder, array $options = [])
                 $options['numericFilters']);
         }
 
+        if ($builder->callback) {
+            return call_user_func(
+                $builder->callback,
+                $this->elastic,
+                $builder->query,
+                $params
+            );
+        }
+
         return $this->elastic->search($params);
     }
 

diff --git a/tests/ElasticsearchEngineTest.php b/tests/ElasticsearchEngineTest.php
@@ -82,6 +82,28 @@ public function test_search_sends_correct_parameters_to_elasticsearch()
         $engine->search($builder);
     }
 
+    public function test_builder_callback_can_manipulate_search_parameters_to_elasticsearch()
+    {
+        /** @var \Elasticsearch\Client|\Mockery\MockInterface $client */
+        $client = Mockery::mock(\Elasticsearch\Client::class);
+        $client->shouldReceive('search')->with('modified_by_callback');
+
+        $engine = new ElasticsearchEngine($client, 'scout');
+        $builder = new Laravel\Scout\Builder(
+            new ElasticsearchEngineTestModel(),
+            'huayra',
+            function (\Elasticsearch\Client $client, $query, $params) {
+                $this->assertNotEmpty($params);
+                $this->assertEquals('huayra', $query);
+                $params = 'modified_by_callback';
+
+                return $client->search($params);
+            }
+        );
+
+        $engine->search($builder);
+    }
+
     public function test_map_correctly_maps_results_to_models()
     {
         $client = Mockery::mock('Elasticsearch\Client');