package community_test

import (
	"testing"
	"time"

	sdkmath "cosmossdk.io/math"
	"github.com/0glabs/0g-chain/app"
	"github.com/0glabs/0g-chain/x/community"
	"github.com/0glabs/0g-chain/x/community/types"
	tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
	sdk "github.com/cosmos/cosmos-sdk/types"
	authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
	"github.com/stretchr/testify/require"
)

func TestABCIStakingRewardsArePaidOutOnDisableInflationBlock(t *testing.T) {
	app.SetSDKConfig()
	tApp := app.NewTestApp()
	tApp.InitializeFromGenesisStates()
	keeper := tApp.GetCommunityKeeper()
	accountKeeper := tApp.GetAccountKeeper()
	bankKeeper := tApp.GetBankKeeper()

	// a block that runs after addition of the disable inflation code on chain
	// but before the disable inflation time
	initialBlockTime := time.Now()
	ctx := tApp.NewContext(true, tmproto.Header{Height: 1, Time: initialBlockTime})

	poolAcc := accountKeeper.GetModuleAccount(ctx, types.ModuleName)
	feeCollectorAcc := accountKeeper.GetModuleAccount(ctx, authtypes.FeeCollectorName)

	disableTime := initialBlockTime.Add(9 * time.Second)

	// set state
	params, _ := keeper.GetParams(ctx)
	params.UpgradeTimeDisableInflation = disableTime
	params.UpgradeTimeSetStakingRewardsPerSecond = sdkmath.LegacyNewDec(1000000) // 1 KAVA
	params.StakingRewardsPerSecond = sdkmath.LegacyZeroDec()
	keeper.SetParams(ctx, params)

	// fund community pool account
	tApp.FundAccount(ctx, poolAcc.GetAddress(), sdk.NewCoins(sdk.NewCoin("ukava", sdkmath.NewInt(10000000)))) // 10 KAVA
	initialFeeCollectorBalance := bankKeeper.GetBalance(ctx, feeCollectorAcc.GetAddress(), "ukava").Amount

	// run one block
	community.BeginBlocker(ctx, keeper)

	// assert that staking rewards in parameters are still set to zero
	params, found := keeper.GetParams(ctx)
	require.True(t, found)
	require.Equal(t, sdkmath.LegacyZeroDec(), params.StakingRewardsPerSecond)

	// assert no rewards are given yet
	rewards := bankKeeper.GetBalance(ctx, feeCollectorAcc.GetAddress(), "ukava").Amount.Sub(initialFeeCollectorBalance)
	require.Equal(t, sdkmath.ZeroInt(), rewards)

	// new block when disable inflation runs, 10 seconds from initial block for easy math
	blockTime := disableTime.Add(1 * time.Second)
	ctx = tApp.NewContext(true, tmproto.Header{Height: ctx.BlockHeight() + 1, Time: blockTime})

	// run the next block
	community.BeginBlocker(ctx, keeper)

	// assert that staking rewards have been set and disable inflation time is zero
	params, found = keeper.GetParams(ctx)
	require.True(t, found)
	require.True(t, params.UpgradeTimeDisableInflation.IsZero())
	require.Equal(t, sdkmath.LegacyNewDec(1000000), params.StakingRewardsPerSecond)

	// assert that 10 KAVA has been distributed in rewards
	rewards = bankKeeper.GetBalance(ctx, feeCollectorAcc.GetAddress(), "ukava").Amount.Sub(initialFeeCollectorBalance)
	require.Equal(t, sdkmath.NewInt(10000000).String(), rewards.String())
}