From 24ad74f136392531b3402650603f8f96f770c471 Mon Sep 17 00:00:00 2001
From: snipe <snipe@snipe.net>
Date: Fri, 5 Jul 2024 08:21:18 +0100
Subject: [PATCH] Updated factories

Signed-off-by: snipe <snipe@snipe.net>
---
 database/factories/AssetFactory.php    | 6 +++---
 database/factories/LocationFactory.php | 9 +++++++++
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/database/factories/AssetFactory.php b/database/factories/AssetFactory.php
index 43845c307..8f910b903 100644
--- a/database/factories/AssetFactory.php
+++ b/database/factories/AssetFactory.php
@@ -300,11 +300,11 @@ class AssetFactory extends Factory
         });
     }
 
-    public function assignedToLocation()
+    public function assignedToLocation(Location $location = null)
     {
-        return $this->state(function () {
+        return $this->state(function () use ($location) {
             return [
-                'assigned_to' => Location::factory(),
+                'assigned_to' => $location->id ?? Location::factory(),
                 'assigned_type' => Location::class,
             ];
         });
diff --git a/database/factories/LocationFactory.php b/database/factories/LocationFactory.php
index 6db268e8c..abd773010 100644
--- a/database/factories/LocationFactory.php
+++ b/database/factories/LocationFactory.php
@@ -25,4 +25,13 @@ class LocationFactory extends Factory
             'image' => rand(1, 9).'.jpg',
         ];
     }
+
+    public function deletedLocation()
+    {
+        return $this->state(function () {
+            return [
+                'deleted_at' => $this->faker->dateTime(),
+            ];
+        });
+    }
 }