psalm-plugin-laravel/tests/acceptance/EloquentBuilderTypes.feature

Feature: Eloquent Builder types
  Illuminate\Database\Eloquent\Builder has type support

  Background:
    Given I have the following config
      """
      <?xml version="1.0"?>
      <psalm totallyTyped="true">
        <projectFiles>
          <directory name="."/>
          <ignoreFiles> <directory name="../../vendor"/> </ignoreFiles>
        </projectFiles>
        <plugins>
          <pluginClass class="Psalm\LaravelPlugin\Plugin"/>
        </plugins>
      </psalm>
      """

  Scenario: Models can call eloquent query builder instance methods
    Given I have the following code
    """
    <?php declare(strict_types=1);

    use Tests\Psalm\LaravelPlugin\Models\User;

    final class UserRepository
    {

        /**
        * @return \Illuminate\Database\Eloquent\Builder<User>
        */
        public function getNewQuery(): \Illuminate\Database\Eloquent\Builder
        {
          return User::query();
        }

        /**
        * @return \Illuminate\Database\Eloquent\Builder<User>
        */
        public function getNewModelQuery(): \Illuminate\Database\Eloquent\Builder
        {
          return (new User())->newModelQuery();
        }

        /**
        * @param \Illuminate\Database\Eloquent\Builder<User> $builder
        */
        public function firstOrFailFromBuilderInstance(\Illuminate\Database\Eloquent\Builder $builder): User {
          return $builder->firstOrFail();
        }

        /**
        * @param \Illuminate\Database\Eloquent\Builder<User> $builder
        */
        public function findOrFailFromBuilderInstance(\Illuminate\Database\Eloquent\Builder $builder): User {
          return $builder->findOrFail(1);
        }

        /**
        * @param \Illuminate\Database\Eloquent\Builder<User> $builder
        * @return \Illuminate\Database\Eloquent\Collection<User>
        */
        public function findMultipleOrFailFromBuilderInstance(\Illuminate\Database\Eloquent\Builder $builder): \Illuminate\Database\Eloquent\Collection {
          return $builder->findOrFail([1, 2]);
        }

        /**
        * @param \Illuminate\Database\Eloquent\Builder<User> $builder
        */
        public function findOne(\Illuminate\Database\Eloquent\Builder $builder): ?User {
          return $builder->find(1);
        }

        /**
        * @param \Illuminate\Database\Eloquent\Builder<User> $builder
        */
        public function findViaArray(\Illuminate\Database\Eloquent\Builder $builder): \Illuminate\Database\Eloquent\Collection {
          return $builder->find([1]);
        }

        /**
        * @return \Illuminate\Database\Eloquent\Builder<User>
        */
        public function getWhereBuilderViaInstance(array $attributes): \Illuminate\Database\Eloquent\Builder {
          return (new User())->where($attributes);
        }
    }
    """
    When I run Psalm
    Then I see no errors

  Scenario: can call static methods on model
    Given I have the following code
    """
    <?php declare(strict_types=1);

    final class User extends \Illuminate\Database\Eloquent\Model {
      protected $table = 'users';
    };

    final class UserRepository
    {

        /**
        * @return \Illuminate\Database\Eloquent\Builder<User>
        */
        public function getWhereBuilderViaStatic(array $attributes): \Illuminate\Database\Eloquent\Builder
        {
          return User::where($attributes);
        }

        /**
        * @psalm-return \Illuminate\Database\Eloquent\Collection<User>
        */
        public function getWhereViaStatic(array $attributes): \Illuminate\Database\Eloquent\Collection
        {
          return User::where($attributes)->get();
        }
    }
    """
    When I run Psalm
    Then I see no errors

  Scenario:
    Given I have the following code
    """
    <?php declare(strict_types=1);

    final class User extends \Illuminate\Database\Eloquent\Model {
      protected $table = 'users';
    };

    final class UserRepository
    {
        /**
        * @return \Illuminate\Database\Eloquent\Builder<User>
        */
        public function test_failure(): \Illuminate\Database\Eloquent\Builder
        {
          return User::fakeQueryMethodThatDoesntExist();
        }
    }
    """
    When I run Psalm
    Then I see these errors
      | Type  | Message |
      | MixedInferredReturnType | Could not verify return type 'Illuminate\Database\Eloquent\Builder<User>' for UserRepository::test_failure |
      | MixedReturnStatement    | Could not infer a return type                                                                              |

  Scenario: can call methods on underlying query builder
    Given I have the following code
    """
    <?php declare(strict_types=1);

    use Tests\Psalm\LaravelPlugin\Models\User;
    use \Illuminate\Database\Eloquent\Builder;

    /**
    * @psalm-param Builder<User> $builder
    * @psalm-return Builder<User>
    */
    function test(Builder $builder): Builder {
      return $builder->orderBy('id', 'ASC');
    }
    """
    When I run Psalm
    Then I see no errors