Merge pull request laravel#192 from morloderex/add-macroable

Add macroable to scout builder
jorenvanhee · May 31, 2017 · 18992a2 · 18992a2
2 parents c822eaf + 353cb48
commit 18992a2
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/Builder.php b/src/Builder.php
@@ -3,11 +3,14 @@
 namespace Laravel\Scout;
 
 use Illuminate\Pagination\Paginator;
+use Illuminate\Support\Traits\Macroable;
 use Illuminate\Database\Eloquent\Collection;
 use Illuminate\Pagination\LengthAwarePaginator;
 
 class Builder
 {
+    use Macroable;
+
     /**
      * The model instance.
      *

diff --git a/tests/BuilderTest.php b/tests/BuilderTest.php
@@ -29,4 +29,16 @@ public function test_pagination_correctly_handles_paginated_results()
 
         $builder->paginate();
     }
+
+    public function testMacroable()
+    {
+        Builder::macro('foo', function () {
+            return 'bar';
+        });
+
+        $builder = new Builder($model = Mockery::mock(), 'zonda');
+        $this->assertEquals(
+            'bar', $builder->foo()
+        );
+    }
 }