Correct types on factory state() method

Author: mdpoulter

src/Illuminate/Database/Eloquent/Factories/Factory.php

@@ -469,7 +469,7 @@ protected function expandAttributes(array $definition)
     /**
      * Add a new state transformation to the model definition.
      *
-     * @param  (callable(array<string, mixed>): array<string, mixed>)|array<string, mixed>  $state
+     * @param  (callable(array<string, mixed>, \Illuminate\Database\Eloquent\Model|null=): array<string, mixed>)|array<string, mixed>  $state
      * @return static
      */
     public function state($state)

types/Database/Eloquent/Factories/Factory.php

@@ -141,7 +141,15 @@ public function definition()
 }));
 
 assertType('UserFactory', $factory->state(['string' => 'string']));
-assertType('UserFactory', $factory->state(function () {
+assertType('UserFactory', $factory->state(function ($attributes) {
+    assertType('array<string, mixed>', $attributes);
+
+    return ['string' => 'string'];
+}));
+assertType('UserFactory', $factory->state(function ($attributes, $model) {
+    assertType('array<string, mixed>', $attributes);
+    assertType('Illuminate\Database\Eloquent\Model|null', $model);
+
     return ['string' => 'string'];
 }));