add withOwner on shortUrlFactory

YorCreative · Jul 18, 2024 · b0c5d23 · b0c5d23
1 parent 25719cb
commit b0c5d23
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/Utility/Factories/ShortUrlFactory.php b/src/Utility/Factories/ShortUrlFactory.php
@@ -3,8 +3,10 @@
 namespace YorCreative\UrlShortener\Utility\Factories;
 
 use Illuminate\Database\Eloquent\Factories\Factory;
+use Illuminate\Database\Eloquent\Model;
 use Illuminate\Support\Carbon;
 use YorCreative\UrlShortener\Models\ShortUrl;
+use YorCreative\UrlShortener\Models\ShortUrlOwnership;
 use YorCreative\UrlShortener\Services\UtilityService;
 use YorCreative\UrlShortener\Traits\ShortUrlHelper;
 
@@ -85,4 +87,11 @@ public function withLink(string $link): self
             ];
         });
     }
+
+    public function withOwner(Model $model): self
+    {
+        ShortUrlOwnership::factory()->withOwner($model)->create();
+
+        return $this;
+    }
 }