add forceRefund method

Author: rez1dent3

src/Traits/CanBePaid.php

@@ -60,10 +60,11 @@ public function paid(Product $product): ?Transfer
 
     /**
      * @param Product $product
+     * @param bool $force
      * @return bool
      * @throws
      */
-    public function refund(Product $product): bool
+    public function refund(Product $product, bool $force = false): bool
     {
         $transfer = $this->paid($product);
 
@@ -72,23 +73,38 @@ public function refund(Product $product): bool
                 ->setModel($this->transfers()->getMorphClass());
         }
 
-        return DB::transaction(function() use ($product, $transfer) {
-            $product->transfer($this, $product->getAmountProduct(), $product->getMetaProduct());
+        return DB::transaction(function() use ($product, $transfer, $force) {
+            if ($force) {
+                $product->forceTransfer($this, $product->getAmountProduct(), $product->getMetaProduct());
+            } else {
+                $product->transfer($this, $product->getAmountProduct(), $product->getMetaProduct());
+            }
+
             return $transfer->update(['refund' => 1]);
         });
     }
 
     /**
      * @param Product $product
+     * @param bool $force
      * @return bool
      */
-    public function safeRefund(Product $product): bool
+    public function safeRefund(Product $product, bool $force = false): bool
     {
         try {
-            return $this->refund($product);
+            return $this->refund($product, $force);
         } catch (\Throwable $throwable) {
             return false;
         }
     }
 
+    /**
+     * @param Product $product
+     * @return bool
+     */
+    public function forceRefund(Product $product): bool
+    {
+        return $this->refund($product, true);
+    }
+
 }

tests/ProductTest.php

@@ -54,4 +54,45 @@ public function testRefund(): void
         $this->assertEquals($buyer->balance, 0);
     }
 
+    /**
+     * @return void
+     */
+    public function testForceRefund(): void
+    {
+        /**
+         * @var Buyer $buyer
+         * @var Item $product
+         */
+        $buyer = factory(Buyer::class)->create();
+        $product = factory(Item::class)->create([
+            'quantity' => 1,
+        ]);
+
+        $this->assertNotEquals($product->balance, 0);
+        $product->withdraw($product->balance);
+
+        $this->assertEquals($buyer->balance, 0);
+        $buyer->deposit($product->price);
+
+        $this->assertEquals($buyer->balance, $product->price);
+
+        $buyer->pay($product);
+        $this->assertEquals($buyer->balance, 0);
+        $this->assertEquals($product->balance, $product->price);
+
+        $product->withdraw($product->balance);
+        $this->assertEquals($product->balance, 0);
+
+        $this->assertFalse($buyer->safeRefund($product));
+        $this->assertTrue($buyer->forceRefund($product));
+
+        $this->assertEquals($product->balance, -$product->price);
+        $this->assertEquals($buyer->balance, $product->price);
+        $product->deposit(-$product->balance);
+        $buyer->withdraw($buyer->balance);
+
+        $this->assertEquals($product->balance, 0);
+        $this->assertEquals($buyer->balance, 0);
+    }
+
 }