diff --git a/src/Criteria/SuggestCriteria.php b/src/Criteria/SuggestCriteria.php index 50eb6fe..fe7c387 100644 --- a/src/Criteria/SuggestCriteria.php +++ b/src/Criteria/SuggestCriteria.php @@ -19,6 +19,12 @@ final class SuggestCriteria implements SerializableInterface */ private $track; + /** + * @var int + * @Serializer\Type("integer") + */ + private $size = 10; + /** * @param string $query * @return SuggestCriteria @@ -38,4 +44,14 @@ public function setTrack(bool $track): SuggestCriteria $this->track = $track; return $this; } + + /** + * @param int $size + * @return SuggestCriteria + */ + public function setSize(int $size) + { + $this->size = $size; + return $this; + } } diff --git a/tests/Builder/SuggestCriteriaBuilderTest.php b/tests/Builder/SuggestCriteriaBuilderTest.php index b8278db..94f16c6 100644 --- a/tests/Builder/SuggestCriteriaBuilderTest.php +++ b/tests/Builder/SuggestCriteriaBuilderTest.php @@ -30,12 +30,15 @@ public function criteriaProvider() return [ [ new SuggestCriteria(), - [] + [ + 'size' => 10 + ] ], [ (new SuggestCriteria())->setTrack(false), [ 'track' => false, + 'size' => 10 ] ], [ @@ -44,6 +47,17 @@ public function criteriaProvider() [ 'q' => 'test query', 'track' => false, + 'size' => 10 + ] + ], + [ + (new SuggestCriteria())->setQuery('test query') + ->setTrack(false) + ->setSize(5), + [ + 'q' => 'test query', + 'track' => false, + 'size' => 5 ] ], ];