From de8607e6d6477ece3741257485d16702cd54195a Mon Sep 17 00:00:00 2001
From: antonhlazkov <38909861+antonhlazkov@users.noreply.github.com>
Date: Tue, 31 Jul 2018 11:37:40 +0300
Subject: [PATCH] Size property for SuggestCriteria (#37)

* SEARCH-201 Size property for SuggestCriteria;

* SEARCH-201 Size property for SuggestCriteria; Fix Unit Tests;
---
 src/Criteria/SuggestCriteria.php             | 16 ++++++++++++++++
 tests/Builder/SuggestCriteriaBuilderTest.php | 16 +++++++++++++++-
 2 files changed, 31 insertions(+), 1 deletion(-)

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
                 ]
             ],
         ];