import hiltes

src/Entity/Stock.php

@@ -69,12 +69,12 @@ class Stock
         return $this;
     }
 
-    public function getWarehouseId(): ?Warehouse
+    public function getWarehouseId(): ?int
     {
         return $this->warehouse_id;
     }
 
-    public function setWarehouseId(?Warehouse $warehouse_id): self
+    public function setWarehouseId(int $warehouse_id): self
     {
         $this->warehouse_id = $warehouse_id;
 
@@ -92,4 +92,10 @@ class Stock
 
         return $this;
     }
+
+    public function setWarehouse(Warehouse $warehouse): self
+    {
+        $this->warehouse_id = $warehouse->getId();
+        return $this;
+    }
 }

src/Entity/Warehouse.php

@@ -31,6 +31,20 @@ class Warehouse
         return $this->id;
     }
 
+    public function setID(int $id): self
+    {
+        $this->id = $id;
+
+        return $this;
+    }
+
+    public function setWarehouseId(int $warehouse_id): self
+    {
+        $this->warehouse_id = $warehouse_id;
+
+        return $this;
+    }
+
     /**
      * @return Collection<int, Stock>
      */