add type assertion test for query builder for

Author: liamduckett

phpstan.neon.dist

@@ -8,6 +8,7 @@ parameters:
         - src/
         - config/
         - database/
+        - types/
 
     # Level 9 is the highest level
     level: 5

types/query-builder.php

@@ -0,0 +1,22 @@
+<?php
+
+use Illuminate\Database\Eloquent\Model;
+use Illuminate\Database\Eloquent\Relations\BelongsTo;
+use Spatie\QueryBuilder\QueryBuilder;
+use function PHPStan\Testing\assertType;
+
+class Book extends Model {
+    /**
+     * @return BelongsTo<Author, $this>
+     */
+    public function author(): BelongsTo
+    {
+        return $this->belongsTo(Author::class);
+    }
+}
+
+class Author extends Model {}
+
+assertType('Spatie\QueryBuilder\QueryBuilder<Book>', QueryBuilder::for(Book::class));
+assertType('Spatie\QueryBuilder\QueryBuilder<Book>', QueryBuilder::for(Book::query()));
+assertType('Spatie\QueryBuilder\QueryBuilder<Author>', QueryBuilder::for((new Book)->author()));