Size property for SuggestCriteria (#37)

* SEARCH-201 Size property for SuggestCriteria; * SEARCH-201 Size property for SuggestCriteria; Fix Unit Tests;
printdeal · Jul 31, 2018 · de8607e · de8607e
1 parent 7708da1
commit de8607e
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 1 deletion.
diff --git 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
@@ -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
                 ]
             ],
         ];