Fix #3

Author: Nayjest

src/Processor/PaginateProcessor.php

@@ -2,23 +2,23 @@
 
 namespace ViewComponents\Eloquent\Processor;
 
-use Illuminate\Database\Eloquent\Builder;
+use Illuminate\Database\Eloquent\Builder as EloquentBuilder;
+use Illuminate\Database\Query\Builder;
 use ViewComponents\ViewComponents\Data\Operation\OperationInterface;
 use ViewComponents\ViewComponents\Data\Operation\PaginateOperation;
 use ViewComponents\ViewComponents\Data\Processor\AbstractPaginateProcessor;
 
 class PaginateProcessor extends AbstractPaginateProcessor
 {
     /**
-     * @param Builder $src
+     * @param Builder|EloquentBuilder $src
      * @param OperationInterface|PaginateOperation $operation
      * @return mixed
      */
     public function process($src, OperationInterface $operation)
     {
 
-        $src->getQuery()
-            ->limit($operation->getPageSize())
+        $src->limit($operation->getPageSize())
             ->offset($this->getOffset($operation));
         return $src;
     }

tests/phpunit/Processor/PaginateProcessorTest.php

@@ -0,0 +1,32 @@
+<?php
+
+namespace ViewComponents\Eloquent\Test\Processor;
+
+use ViewComponents\Eloquent\EloquentDataProvider;
+use ViewComponents\Eloquent\Test\Mock\TestUser;
+use ViewComponents\ViewComponents\Data\Operation\PaginateOperation;
+use Illuminate\Database\Capsule\Manager as DB;
+
+class PaginateProcessorTest extends \PHPUnit_Framework_TestCase
+{
+    public function test()
+    {
+        $recordsQty = TestUser::all()->count();
+        if ($recordsQty <= 5) {
+            throw new \Exception("Can't test PaginateProcessor: not enough fixture data");
+        }
+        $provider =  new EloquentDataProvider(TestUser::class);
+        $provider->operations()->add(new PaginateOperation(1,3));
+        $data = iterator_to_array($provider);
+        $id1 = array_first($data)->id;
+        self::assertEquals(3, count($data));
+
+        $provider =  new EloquentDataProvider(DB::table((new TestUser)->getTable()));
+        $provider->operations()->add(new PaginateOperation(2,2));
+        $data = iterator_to_array($provider);
+        self::assertEquals(2, count($data));
+
+        $id2 = array_first($data)->id;
+        self::assertTrue($id2 > 0 && $id2 > 0 && $id1 !== $id2);
+    }
+}