fix bug

Author: Babichev Maxim

composer.json

@@ -24,6 +24,7 @@
   ],
   "require": {
     "php": "^7.1|^8.0",
+    "ext-pdo": "*",
     "illuminate/database": "^5.5|^6.0",
     "nesbot/carbon": "^1.20|^2.0",
     "doctrine/dbal": "^2.8",

src/Services/CommonService.php

@@ -225,7 +225,14 @@ public function addBalance(Wallet $wallet, int $amount): bool
             $balance = app(Storable::class)
                 ->incBalance($wallet, $amount);
 
-            $result = $wallet->update(compact('balance'));
+            try {
+                $result = $wallet->update(compact('balance'));
+            } catch (\Throwable $throwable) {
+                app(Storable::class)
+                    ->setBalance($wallet, $wallet->getAvailableBalance());
+
+                throw $throwable;
+            }
 
             if (!$result) {
                 app(Storable::class)

tests/BalanceTest.php

@@ -4,13 +4,14 @@
 
 use Bavix\Wallet\Interfaces\Storable;
 use Bavix\Wallet\Models\Wallet;
-use Bavix\Wallet\Objects\Cart;
 use Bavix\Wallet\Services\CommonService;
 use Bavix\Wallet\Services\ProxyService;
 use Bavix\Wallet\Services\WalletService;
 use Bavix\Wallet\Test\Models\Buyer;
 use Bavix\Wallet\Test\Models\UserMulti;
 use Illuminate\Support\Facades\DB;
+use PHPUnit\Framework\MockObject\MockObject;
+use PDOException;
 use function app;
 
 class BalanceTest extends TestCase
@@ -148,7 +149,7 @@ public function testFailUpdate(): void
         $this->assertTrue($wallet->exists);
 
         /**
-         * @var Wallet $mockWallet
+         * @var Wallet|MockObject $mockWallet
          */
         $mockWallet = $this->createMock(\get_class($wallet));
         $mockWallet->method('update')->willReturn(false);
@@ -161,6 +162,36 @@ public function testFailUpdate(): void
         $this->assertEquals(0, app(Storable::class)->getBalance($wallet));
     }
 
+    /**
+     * @return void
+     * @throws
+     */
+    public function testThrowUpdate(): void
+    {
+        $this->expectException(PDOException::class);
+
+        /**
+         * @var Buyer $buyer
+         */
+        $buyer = factory(Buyer::class)->create();
+        $this->assertFalse($buyer->relationLoaded('wallet'));
+        $wallet = $buyer->wallet;
+
+        $this->assertFalse($wallet->exists);
+        $this->assertEquals($wallet->balance, 0);
+        $this->assertTrue($wallet->exists);
+
+        /**
+         * @var Wallet|MockObject $mockWallet
+         */
+        $mockWallet = $this->createMock(\get_class($wallet));
+        $mockWallet->method('update')->willThrowException(new PDOException());
+        $mockWallet->method('getKey')->willReturn($wallet->getKey());
+
+        app(CommonService::class)
+            ->addBalance($mockWallet, 100);
+    }
+
     /**
      * @throws
      */