From a37a0d3d7f723131a3356af7aea576ec5beee0f9 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C4=90=C3=B4ng=20Li=E1=BB=81u?= <duongduydongtq@gmail.com>
Date: Tue, 21 Jan 2025 13:02:54 +0700
Subject: [PATCH] tests

---
 x/vaults/keeper/msg_server_test.go | 67 ++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 x/vaults/keeper/msg_server_test.go

diff --git a/x/vaults/keeper/msg_server_test.go b/x/vaults/keeper/msg_server_test.go
new file mode 100644
index 0000000..5412009
--- /dev/null
+++ b/x/vaults/keeper/msg_server_test.go
@@ -0,0 +1,67 @@
+package keeper_test
+
+import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+
+	"cosmossdk.io/math"
+
+	"github.com/onomyprotocol/reserve/x/vaults/types"
+)
+
+func (s *KeeperTestSuite) TestBurnShortfallByMintDenom() {
+	testcases := []struct {
+		name                        string
+		mintDenom                   string
+		setup                       func() types.MsgBurnShortfall
+		expShortfallAmountAfterBurn math.Int
+		expReserveBalcesAfterBurn   math.Int
+		expPass                     bool
+	}{
+		{
+			name:      "success, shortfallAmount is less than reserve balances",
+			mintDenom: "fxUSD",
+			setup: func() types.MsgBurnShortfall {
+				// make sure reserve has money
+				mintCoin := sdk.NewCoins(sdk.NewCoin("fxUSD", math.NewInt(10_000_000)))
+				s.FundAccount(s.TestAccs[0], types.ModuleName, mintCoin)
+				s.k.BankKeeper.SendCoinsFromAccountToModule(s.Ctx, s.TestAccs[0], types.ReserveModuleName, mintCoin)
+
+				// make sure Guaranteed Shortfall Amount
+				s.k.ShortfallAmount.Set(s.Ctx, "fxUSD", math.NewInt(1_000_000))
+				return types.MsgBurnShortfall{
+					Authority: "onomy10d07y265gmmuvt4z0w9aw880jnsr700jqr8n8k",
+					MintDenom: "fxUSD",
+				}
+			},
+			expShortfallAmountAfterBurn: math.ZeroInt(),
+			expReserveBalcesAfterBurn:   math.NewInt(9_000_000),
+			expPass:                     true,
+		},
+	}
+
+	for _, t := range testcases {
+		s.Run(t.name, func() {
+			s.SetupTest()
+			msg := t.setup()
+
+			// burn
+			_, err := s.msgServer.BurnShortfall(s.Ctx, &msg)
+			if t.expPass {
+				s.Require().NoError(err)
+
+				// check reserve balances after burn
+				reserveBalces := s.k.BankKeeper.GetAllBalances(s.Ctx, s.App.AccountKeeper.GetModuleAddress(types.ReserveModuleName))
+				s.Require().True(reserveBalces.AmountOf(t.mintDenom).Equal(t.expReserveBalcesAfterBurn))
+
+				// check ShortfallAmount after burn
+
+				shortfallAmountAfterBurn, err := s.k.ShortfallAmount.Get(s.Ctx, t.mintDenom)
+				s.Require().NoError(err)
+
+				s.Require().True(shortfallAmountAfterBurn.Equal(t.expShortfallAmountAfterBurn))
+			} else {
+				s.Require().Error(err)
+			}
+		})
+	}
+}