New common criteria “Take”

src/Criteria/Common/Take.php

@@ -0,0 +1,33 @@
+<?php
+namespace Czim\Repository\Criteria\Common;
+
+use Czim\Repository\Criteria\AbstractCriteria;
+use Illuminate\Database\Query\Builder;
+
+class Take extends AbstractCriteria
+{
+ /**
+ * @var int
+ */
+ protected $quantity;
+
+
+ /**
+ * The number of records returned
+ *
+ * @param int $quantity
+ */
+ public function __construct($quantity)
+ {
+ $this->quantity = $quantity;
+ }
+
+ /**
+ * @param Builder $model
+ * @return mixed
+ */
+ public function applyToQuery($model)
+ {
+ return $model->take($this->quantity);
+ }
+}

tests/CommonCriteriaTest.php

@@ -9,6 +9,7 @@
 use Czim\Repository\Criteria\Common\OrderBy;
 use Czim\Repository\Criteria\Common\Scope;
 use Czim\Repository\Criteria\Common\Scopes;
+use Czim\Repository\Criteria\Common\Take;
 use Czim\Repository\Criteria\Common\WhereHas;
 use Czim\Repository\Criteria\Common\WithRelations;
 use Czim\Repository\Enums\CriteriaKey;
@@ -177,4 +178,14 @@ function with_relations_criteria_works()
  );
  }
 
+ /**
+ * @test
+ */
+ function take_criteria_works()
+ {
+ $this->repository->pushCriteria(new Take(2));
+
+ $this->assertCount(2, $this->repository->all(), "Take Criteria doesn't work");
+ }
+
 }