Allow testing framework factory to return models

src/mantle/testing/factory/class-attachment-factory.php

@@ -99,6 +99,8 @@ public function create_upload_object( $file, $parent = 0 ) {
 	 * @return \WP_Post|null
 	 */
 	public function get_object_by_id( int $object_id ): ?\WP_Post {
-		return get_post_object( $object_id );
+		return $this->as_models
+			? Post::find( $object_id )
+			: get_post_object( $object_id );
 	}
 }

src/mantle/testing/factory/class-factory.php

@@ -7,12 +7,22 @@
 
 namespace Mantle\Testing\Factory;
 
+use Mantle\Contracts\Database\Core_Object;
+
 use function Mantle\Support\Helpers\collect;
+use function Mantle\Support\Helpers\tap;
 
 /**
  * Base Factory
  */
 abstract class Factory {
+	/**
+	 * Flag to return the factory as a model.
+	 *
+	 * @var bool
+	 */
+	protected bool $as_models = false;
+
 	/**
 	 * Creates an object.
 	 *
@@ -29,6 +39,30 @@ abstract public function create( array $args = [] );
 	 */
 	abstract public function get_object_by_id( int $object_id );
 
+	/**
+	 * Generate models from the factory.
+	 *
+	 * @return static
+	 */
+	public function as_models() {
+		return tap(
+			clone $this,
+			fn ( $factory ) => $factory->as_models = true,
+		);
+	}
+
+	/**
+	 * Generate core WordPress objects from the factory.
+	 *
+	 * @return static
+	 */
+	public function as_objects() {
+		return tap(
+			clone $this,
+			fn ( $factory ) => $factory->as_models = false,
+		);
+	}
+
 	/**
 	 * Creates multiple objects.
 	 *
@@ -52,7 +86,7 @@ function() use ( $args ) {
 	 * Creates an object and returns its object.
 	 *
 	 * @param array $args Optional. The arguments for the object to create. Default is empty array.
-	 * @return mixed The created object.
+	 * @return mixed|Core_Object The created object.
 	 */
 	public function create_and_get( $args = [] ) {
 		$object_id = $this->create( $args );

src/mantle/testing/factory/class-post-factory.php

@@ -136,6 +136,8 @@ public function create_ordered_set(
 	 * @return \WP_Post|null
 	 */
 	public function get_object_by_id( int $object_id ) {
-		return get_post_object( $object_id );
+		return $this->as_models
+			? Post::find( $object_id )
+			: get_post_object( $object_id );
 	}
 }

src/mantle/testing/factory/class-term-factory.php

@@ -67,6 +67,12 @@ public function create( array $args = [] ) {
 	 * @return \WP_Term|null
 	 */
 	public function get_object_by_id( int $object_id ) {
-		return get_term_object( $object_id );
+		$term = get_term_object( $object_id );
+
+		if ( $term && $this->as_models ) {
+			return Term::new_from_existing( (array) $term );
+		}
+
+		return $term;
 	}
 }

tests/testing/test-factory.php

@@ -2,6 +2,8 @@
 namespace Mantle\Testing;
 
 use Carbon\Carbon;
+use Mantle\Database\Model\Post;
+use Mantle\Database\Model\Term;
 
 use function Mantle\Support\Helpers\collect;
 
@@ -99,6 +101,14 @@ public function test_comment_factory() {
 		$this->shim_test( \WP_Comment::class, 'comment' );
 	}
 
+	public function test_as_models() {
+		$post = static::factory()->post->as_models()->create_and_get();
+		$term = static::factory()->term->as_models()->create_and_get();
+
+		$this->assertInstanceOf( Post::class, $post );
+		$this->assertInstanceOf( Term::class, $term );
+	}
+
 	protected function shim_test( string $class_name, string $property ) {
 		$this->assertInstanceOf( $class_name, static::factory()->$property->create_and_get() );