Merge pull request #859 from bavix/check-bind-units

add a check for passing the correct model
bavix · Jan 23, 2024 · 5f30995 · 5f30995
2 parents fff328f + 8dafccc
commit 5f30995
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/Units/Domain/DiscountTest.php b/tests/Units/Domain/DiscountTest.php
@@ -14,6 +14,8 @@
 use Bavix\Wallet\Test\Infra\Models\Buyer;
 use Bavix\Wallet\Test\Infra\Models\ItemDiscount;
 use Bavix\Wallet\Test\Infra\TestCase;
+use Bavix\Wallet\Test\Infra\PackageModels\Transaction as InfraTransaction;
+use Bavix\Wallet\Test\Infra\PackageModels\Wallet as InfraWallet;
 
 /**
  * @internal
@@ -239,7 +241,11 @@ public function testPayFree(): void
 
         $transfer = $buyer->payFree($product);
         self::assertSame(Transaction::TYPE_DEPOSIT, $transfer->deposit->type);
+        self::assertInstanceOf(InfraTransaction::class, $transfer->deposit);
+        self::assertInstanceOf(InfraWallet::class, $transfer->from);
         self::assertSame(Transaction::TYPE_WITHDRAW, $transfer->withdraw->type);
+        self::assertInstanceOf(InfraTransaction::class, $transfer->withdraw);
+        self::assertInstanceOf(InfraWallet::class, $transfer->to);
 
         self::assertSame(0, $buyer->balanceInt);
         self::assertSame(0, $product->balanceInt);