added endpoint builder getters tests

src/SearchEndpoint/HighlightEndpoint.php

@@ -29,6 +29,11 @@ class HighlightEndpoint extends AbstractSearchEndpoint
      */
     private $highlight;
 
+    /**
+     * @var string Key for highlight storing.
+     */
+    private $key;
+
     /**
      * {@inheritdoc}
      */
@@ -50,6 +55,15 @@ class HighlightEndpoint extends AbstractSearchEndpoint
             throw new \OverflowException('Only one highlight can be set');
         }
 
+        $this->key = $key;
         $this->highlight = $builder;
     }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function getAll($boolType = null)
+    {
+        return [$this->key => $this->highlight];
+    }
 }

src/SearchEndpoint/QueryEndpoint.php

@@ -104,4 +104,12 @@ class QueryEndpoint extends AbstractSearchEndpoint implements OrderedNormalizerI
     {
         return new BoolQuery();
     }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function getAll($boolType = null)
+    {
+        return $this->bool->getQueries($boolType);
+    }
 }

tests/SearchEndpoint/AggregationsEndpointTest.php

@@ -11,11 +11,8 @@
 
 namespace ONGR\ElasticsearchDSL\Tests\Unit\SearchEndpoint;
 
-use ONGR\ElasticsearchDSL\BuilderInterface;
-use ONGR\ElasticsearchDSL\NamedBuilderInterface;
+use ONGR\ElasticsearchDSL\Aggregation\MissingAggregation;
 use ONGR\ElasticsearchDSL\SearchEndpoint\AggregationsEndpoint;
-use PHPUnit_Framework_MockObject_MockObject as MockObject;
-use Symfony\Component\Serializer\Normalizer\NormalizerInterface;
 
 /**
  * Class AggregationsEndpointTest.
@@ -32,4 +29,16 @@ class AggregationsEndpointTest extends \PHPUnit_Framework_TestCase
             new AggregationsEndpoint()
         );
     }
+
+    public function testEndpointGetter()
+    {
+        $aggName = 'acme_agg';
+        $agg = new MissingAggregation('acme');
+        $endpoint = new AggregationsEndpoint();
+        $endpoint->add($agg, $aggName);
+        $builders = $endpoint->getAll();
+
+        $this->assertCount(1, $builders);
+        $this->assertSame($agg, $builders[$aggName]);
+    }
 }

tests/SearchEndpoint/FilterEndpointTest.php

@@ -11,7 +11,6 @@
 
 namespace ONGR\ElasticsearchDSL\Tests\Unit\SearchEndpoint;
 
-use ONGR\ElasticsearchDSL\BuilderInterface;
 use ONGR\ElasticsearchDSL\Filter\MatchAllFilter;
 use ONGR\ElasticsearchDSL\Query\FilteredQuery;
 use ONGR\ElasticsearchDSL\SearchEndpoint\FilterEndpoint;
@@ -67,4 +66,16 @@ class FilterEndpointTest extends \PHPUnit_Framework_TestCase
         $this->assertInstanceOf('ONGR\ElasticsearchDSL\Query\FilteredQuery', $reference);
         $this->assertSame($matchAllFilter, $reference->getFilter());
     }
+
+    public function testEndpointGetter()
+    {
+        $filterName = 'acme_filter';
+        $filter = new MatchAllFilter();
+        $endpoint = new FilterEndpoint();
+        $endpoint->add($filter, $filterName);
+        $builders = $endpoint->getAll();
+
+        $this->assertCount(1, $builders);
+        $this->assertSame($filter, $builders[$filterName]);
+    }
 }

tests/SearchEndpoint/HighlightEndpointTest.php

@@ -11,7 +11,6 @@
 
 namespace ONGR\ElasticsearchDSL\Tests\Unit\SearchEndpoint;
 
-use ONGR\ElasticsearchDSL\BuilderInterface;
 use ONGR\ElasticsearchDSL\Highlight\Highlight;
 use ONGR\ElasticsearchDSL\SearchEndpoint\HighlightEndpoint;
 use Symfony\Component\Serializer\Normalizer\NormalizerInterface;
@@ -52,4 +51,18 @@ class HighlightEndpointTest extends \PHPUnit_Framework_TestCase
             json_encode($instance->normalize($normalizerInterface))
         );
     }
+
+    public function testEndpointGetter()
+    {
+        $highlightName = 'acme_highlight';
+        $highlight = new Highlight();
+        $highlight->addField('acme');
+
+        $endpoint = new HighlightEndpoint();
+        $endpoint->add($highlight, $highlightName);
+        $builders = $endpoint->getAll();
+
+        $this->assertCount(1, $builders);
+        $this->assertSame($highlight, $builders[$highlightName]);
+    }
 }

tests/SearchEndpoint/PostFilterEndpointTest.php

@@ -11,7 +11,6 @@
 
 namespace ONGR\ElasticsearchDSL\Tests\Unit\SearchEndpoint;
 
-use ONGR\ElasticsearchDSL\BuilderInterface;
 use ONGR\ElasticsearchDSL\Filter\MatchAllFilter;
 use ONGR\ElasticsearchDSL\SearchEndpoint\PostFilterEndpoint;
 use Symfony\Component\Serializer\Normalizer\NormalizerInterface;
@@ -59,4 +58,17 @@ class PostFilterEndpointTest extends \PHPUnit_Framework_TestCase
             json_encode($instance->normalize($normalizerInterface))
         );
     }
+
+    public function testEndpointGetter()
+    {
+        $filterName = 'acme_post_filter';
+        $filter = new MatchAllFilter();
+
+        $endpoint = new PostFilterEndpoint();
+        $endpoint->add($filter, $filterName);
+        $builders = $endpoint->getAll();
+
+        $this->assertCount(1, $builders);
+        $this->assertSame($filter, $builders[$filterName]);
+    }
 }

tests/SearchEndpoint/QueryEndpointTest.php

@@ -11,9 +11,6 @@
 
 namespace ONGR\ElasticsearchDSL\Tests\Unit\SearchEndpoint;
 
-use ONGR\ElasticsearchDSL\BuilderInterface;
-use ONGR\ElasticsearchDSL\Query\BoolQuery;
-use ONGR\ElasticsearchDSL\Query\FilteredQuery;
 use ONGR\ElasticsearchDSL\Query\MatchAllQuery;
 use ONGR\ElasticsearchDSL\SearchEndpoint\QueryEndpoint;
 use PHPUnit_Framework_MockObject_MockObject as MockObject;
@@ -62,4 +59,16 @@ class QueryEndpointTest extends \PHPUnit_Framework_TestCase
             $instance->normalize($normalizerInterface)
         );
     }
+
+    public function testEndpointGetter()
+    {
+        $queryName = 'acme_query';
+        $query = new MatchAllQuery();
+        $endpoint = new QueryEndpoint();
+        $endpoint->add($query, $queryName);
+        $builders = $endpoint->getAll();
+
+        $this->assertCount(1, $builders);
+        $this->assertSame($query, $builders[$queryName]);
+    }
 }

tests/SearchEndpoint/SortEndpointTest.php

@@ -11,7 +11,6 @@
 
 namespace ONGR\ElasticsearchDSL\Tests\Unit\SearchEndpoint;
 
-use ONGR\ElasticsearchDSL\BuilderInterface;
 use ONGR\ElasticsearchDSL\SearchEndpoint\SortEndpoint;
 use ONGR\ElasticsearchDSL\Sort\FieldSort;
 use PHPUnit_Framework_MockObject_MockObject as MockObject;
@@ -50,4 +49,16 @@ class SortEndpointTest extends \PHPUnit_Framework_TestCase
             $instance->normalize($normalizerInterface)
         );
     }
+
+    public function testEndpointGetter()
+    {
+        $sortName = 'acme_sort';
+        $sort = new FieldSort('acme');
+        $endpoint = new SortEndpoint();
+        $endpoint->add($sort, $sortName);
+        $builders = $endpoint->getAll();
+
+        $this->assertCount(1, $builders);
+        $this->assertSame($sort, $builders[$sortName]);
+    }
 }