Add factory() method for Factory.php

zero-to-prod · Oct 15, 2024 · e4841f2 · e4841f2
1 parent c0d6d69
commit e4841f2
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 13 deletions.
diff --git a/README.md b/README.md
@@ -32,12 +32,8 @@ class User
 
     public $first_name;
     public $last_name;
-
-    public static function factory(array $states): UserFactory
-    {
-        return new UserFactory($states);
-    }
 }
+
 class UserFactory
 {
     use \Zerotoprod\DataModelFactory\Factory;
@@ -70,7 +66,7 @@ class UserFactory
     }
 }
 
-$User = User::factory([User::last_name => 'Doe'])
+$User = UserFactory::factory([User::last_name => 'Doe'])
             ->setFirstName('Jane')
             ->make();
 

diff --git a/src/Factory.php b/src/Factory.php
@@ -2,6 +2,8 @@
 
 namespace Zerotoprod\DataModelFactory;
 
+use Tests\Unit\Override\UserFactory;
+
 /**
  * Factory for Instantiating a Class.
  *
@@ -43,7 +45,7 @@
  * ```
  *
  * @link https://github.com/zero-to-prod/data-model-factory
- * @see https://github.com/zero-to-prod/data-model
+ * @see  https://github.com/zero-to-prod/data-model
  */
 trait Factory
 {
@@ -77,6 +79,19 @@ private function definition(): array
         return [];
     }
 
+    /**
+     * Get a new factory instance for the model
+     *
+     * @return self
+     *
+     * @link https://github.com/zero-to-prod/data-model-factory
+     * @see  https://github.com/zero-to-prod/data-model
+     */
+    public static function factory(array $context = []): self
+    {
+        return new static($context);
+    }
+
     /**
      * Instantiates the class using `$this->context`.
      *

diff --git a/tests/Unit/Factory/FactoryTest.php b/tests/Unit/Factory/FactoryTest.php
@@ -13,7 +13,7 @@ class FactoryTest extends TestCase
      */
     public function from(): void
     {
-        $User = User::factory()->make();
+        $User = UserFactory::factory()->make();
 
         self::assertEquals('John', $User->first_name);
         self::assertEquals('N/A', $User->last_name);

diff --git a/tests/Unit/Factory/User.php b/tests/Unit/Factory/User.php
@@ -12,9 +12,4 @@ class User
     public const last_name = 'last_name';
     public $first_name;
     public $last_name;
-
-    public static function factory(array $states = []): UserFactory
-    {
-        return new UserFactory($states);
-    }
 }