Prepare for Collection::load() fix

staudenmeir · Sep 10, 2024 · 0d237ac · 0d237ac
1 parent 041e1d0
commit 0d237ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tests/Concatenation/LaravelAdjacencyList/AncestorsTest.php b/tests/Concatenation/LaravelAdjacencyList/AncestorsTest.php
@@ -2,6 +2,7 @@
 
 namespace Tests\Concatenation\LaravelAdjacencyList;
 
+use Illuminate\Database\Eloquent\Relations\Relation;
 use Staudenmeir\EloquentHasManyDeep\HasManyDeep;
 use Tests\Concatenation\LaravelAdjacencyList\Models\User;
 
@@ -65,7 +66,7 @@ public function testEagerLoadingWithHasOneDeep()
     public function testLazyEagerLoading()
     {
         $users = User::all()->load([
-            'ancestorPosts' => fn (HasManyDeep $query) => $query->orderBy('id'),
+            'ancestorPosts' => fn (Relation $query) => $query->getQuery()->orderBy('id'),
         ]);
 
         $this->assertEquals([], $users[0]->ancestorPosts->pluck('id')->all());

diff --git a/tests/Concatenation/LaravelAdjacencyList/BloodlineTest.php b/tests/Concatenation/LaravelAdjacencyList/BloodlineTest.php
@@ -2,6 +2,7 @@
 
 namespace Tests\Concatenation\LaravelAdjacencyList;
 
+use Illuminate\Database\Eloquent\Relations\Relation;
 use Staudenmeir\EloquentHasManyDeep\HasManyDeep;
 use Tests\Concatenation\LaravelAdjacencyList\Models\User;
 
@@ -35,7 +36,7 @@ public function testEagerLoading()
     public function testLazyEagerLoading()
     {
         $users = User::all()->load([
-            'bloodlinePosts' => fn (HasManyDeep $query) => $query->orderBy('id'),
+            'bloodlinePosts' => fn (Relation $query) => $query->getQuery()->orderBy('id'),
         ]);
 
         $this->assertEquals([10, 20, 30, 40, 50, 60, 70, 80], $users[0]->bloodlinePosts->pluck('id')->all());