[5.x] Make limit modifier work with query builders (#10818)

Co-authored-by: Jason Varga <[email protected]>

src/Modifiers/CoreModifiers.php

@@ -1452,6 +1452,10 @@ public function limit($value, $params)
     {
         $limit = Arr::get($params, 0, 0);
 
+        if (Compare::isQueryBuilder($value)) {
+            return $value->limit($limit);
+        }
+
         if ($value instanceof Collection) {
             return $value->take($limit);
         }

tests/Modifiers/LimitTest.php

@@ -5,6 +5,7 @@
 use Illuminate\Support\Collection;
 use PHPUnit\Framework\Attributes\Group;
 use PHPUnit\Framework\Attributes\Test;
+use Statamic\Contracts\Query\Builder;
 use Statamic\Modifiers\Modify;
 use Tests\TestCase;
 
@@ -34,8 +35,18 @@ public function it_limits_collections(): void
         $this->assertEquals(['one', 'two', 'three'], $limited->all());
     }
 
-    public function modify($arr, $limit)
+    #[Test]
+    public function it_limits_builders(): void
+    {
+        $query = \Mockery::mock(Builder::class);
+        $query->shouldReceive('limit')->with(2)->once()->andReturnSelf();
+
+        $limited = $this->modify($query, 2);
+        $this->assertSame($query, $limited);
+    }
+
+    public function modify($value, $limit)
     {
-        return Modify::value($arr)->limit($limit)->fetch();
+        return Modify::value($value)->limit($limit)->fetch();
     }
 }