type cleanup and support for creating provisional drafts

src/factories/Asset.php

@@ -24,6 +24,8 @@
  *
  * > **Note**
  * > Any assets created during a test will be cleaned up and deleted after the test.
+ *
+ * @extends Factory<\craft\elements\Asset>
  */
 class Asset extends Element
 {

src/factories/Category.php

@@ -4,6 +4,7 @@
 
 /**
  * @method self title(string $title)
+ * @extends Factory<\craft\elements\Category>
  */
 class Category extends Element
 {

src/factories/Entry.php

@@ -24,6 +24,7 @@
  * @phpstan-ignore-next-line ignored because the file is generated
  *
  * @mixin FactoryFields
+ * @extends Factory<\craft\elements\Entry>
  */
 class Entry extends Element
 {
@@ -104,6 +105,13 @@ public function setDateField($key, $value)
         $this->attributes[$key] = $value;
     }
 
+    public function isProvisionalDraft($provisionalDraft=true): self
+    {
+        $this->attributes['isProvisionalDraft'] = $provisionalDraft;
+
+        return $this;
+    }
+
     /**
      * Set the author of the entry. You may pass a full user object, a user ID,
      * a username, email, or a user ID.

src/factories/Factory.php

@@ -166,7 +166,7 @@ public function __isset($key)
      * Create a new factory by calling `::factory()` on the type of element to be
      * created, such as `Entry::factory()` or `Asset::factory()`.
      */
-    public static function factory()
+    public static function factory(): static
     {
         return new static;
     }

src/factories/User.php

@@ -4,6 +4,7 @@
 
 /**
  * @method self admin(bool $isAdmin)
+ * @extends Factory<\craft\elements\User>
  */
 class User extends Element
 {

src/factories/Volume.php

@@ -11,6 +11,9 @@
 use function markhuot\craftpest\helpers\craft\volumeDefinition;
 use function markhuot\craftpest\helpers\craft\volumeDeleteRootDirectory;
 
+/**
+ * @extends Factory<\craft\models\Volume>
+ */
 class Volume extends Factory
 {
     /**

tests/ProvisionalDraftTest.php

@@ -0,0 +1,11 @@
+<?php
+
+use markhuot\craftpest\factories\Entry;
+
+it('creates provisional drafts', function () {
+    $entry = Entry::factory()
+        ->isProvisionalDraft()
+        ->create();
+
+    expect($entry)->isProvisionalDraft->toBeTrue();
+});