Merge pull request #35 from dvicklund/dev

add ability to get location by id
dvicklund · Jun 10, 2022 · 654e3ca · 654e3ca
2 parents 1020067 + bd944a5
commit 654e3ca
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/src/Traits/LocationTrait.php b/src/Traits/LocationTrait.php
@@ -4,6 +4,7 @@
 
 use Stevebauman\Inventory\Exceptions\InvalidLocationException;
 use Illuminate\Support\Facades\Lang;
+use Stevebauman\Inventory\Models\Location;
 
 /**
  * Trait LocationTrait.
@@ -24,6 +25,18 @@ public function getLocation($location)
     {
         if ($this->isLocation($location)) {
             return $location;
+        } else if (is_numeric($location)) {
+            try {
+                $result = Location::where('id', '=', $location)->first();
+
+                return $result;
+            } catch (\Exception $e) {
+                $message = Lang::get('inventory::exceptions.InvalidLocationException', [
+                    'location' => $location,
+                ]);
+
+                throw new InvalidLocationException($message);
+            }            
         } else {
             $message = Lang::get('inventory::exceptions.InvalidLocationException', [
                 'location' => $location,

diff --git a/tests/InventoryStockTest.php b/tests/InventoryStockTest.php
@@ -285,4 +285,16 @@ public function testRollbackStockMovement() {
 
         $this->assertEquals($initialQuantity, $stock->quantity);
     }
+
+    public function testNumericLocation() {
+        $location = $this->newLocation();
+
+        $item = $this->newInventory();
+
+        $item->createStockOnLocation(42, $location->id, "New stuff");
+
+        $newQuantity = $item->getTotalStock();
+
+        $this->assertEquals(42, $newQuantity);
+    }
 }