fixed simapp and rewards tests

bze-alphateam · Aug 19, 2024 · a621649 · a621649
1 parent 2a3db56
commit a621649
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 5 deletions.
diff --git a/testutil/simapp/app.go b/testutil/simapp/app.go
@@ -708,7 +708,7 @@ func New(
 	return app
 }
 
-func (app *App) setupUpgradeHandlers(cfg module.Configurator) {
+func (app *SimApp) setupUpgradeHandlers(cfg module.Configurator) {
 	app.UpgradeKeeper.SetUpgradeHandler(
 		v710.UpgradeName,
 		v710.CreateUpgradeHandler(cfg, app.mm),

diff --git a/x/rewards/keeper/msg_server_staking_reward_test.go b/x/rewards/keeper/msg_server_staking_reward_test.go
@@ -303,7 +303,7 @@ func (suite *IntegrationTestSuite) TestCreateStakingReward_Success() {
 	initialUserBalances := suite.app.BankKeeper.GetAllBalances(suite.ctx, addr1)
 	suite.Require().True(initialUserBalances.IsZero())
 
-	balances := sdk.NewCoins(sdk.NewInt64Coin("ubze", 20000000000))
+	balances := sdk.NewCoins(sdk.NewInt64Coin("ubze", 40_000_000_000))
 	suite.Require().NoError(simapp.FundAccount(suite.app.BankKeeper, suite.ctx, addr1, balances))
 
 	msg := types.MsgCreateStakingReward{
@@ -330,9 +330,9 @@ func (suite *IntegrationTestSuite) TestCreateStakingReward_Success() {
 	suite.Require().EqualValues(1, storeStakingReward.Lock)
 	suite.Require().EqualValues(0, storeStakingReward.Payouts)
 
-	expectedRemainingBalance := sdk.NewCoins(sdk.NewInt64Coin("ubze", 9999999000))
+	expectedRemainingBalance := sdk.NewCoins(sdk.NewInt64Coin("ubze", 14999999000))
 	actualRemainingBalance := suite.app.BankKeeper.GetAllBalances(suite.ctx, addr1)
-	suite.Require().True(actualRemainingBalance.IsEqual(expectedRemainingBalance))
+	suite.Require().EqualValues(expectedRemainingBalance, actualRemainingBalance)
 }
 
 func (suite *IntegrationTestSuite) TestUpdateStakingReward_Success() {

diff --git a/x/rewards/keeper/msg_server_trading_reward_test.go b/x/rewards/keeper/msg_server_trading_reward_test.go
@@ -227,7 +227,7 @@ func (suite *IntegrationTestSuite) TestCreateTradingReward_Success() {
 	initialUserBalances := suite.app.BankKeeper.GetAllBalances(suite.ctx, addr1)
 	suite.Require().True(initialUserBalances.IsZero())
 
-	balances := sdk.NewCoins(sdk.NewInt64Coin("ubze", 10000002000))
+	balances := sdk.NewCoins(sdk.NewInt64Coin("ubze", 50000002000))
 	suite.Require().NoError(simapp.FundAccount(suite.app.BankKeeper, suite.ctx, addr1, balances))
 	suite.app.TradebinKeeper.SetMarket(suite.ctx, types2.Market{
 		Base:    "stake",