package keeper_test

import (
	"time"

	"go.uber.org/mock/gomock"

	"cosmossdk.io/math"
	"cosmossdk.io/x/protocolpool/types"

	sdk "github.com/cosmos/cosmos-sdk/types"
)

func (suite *KeeperTestSuite) TestInitExportGenesis() {
	suite.bankKeeper.EXPECT().SendCoinsFromModuleToModule(gomock.Any(), types.ProtocolPoolDistrAccount, gomock.Any(), gomock.Any()).Return(nil).AnyTimes()
	suite.bankKeeper.EXPECT().SendCoinsFromModuleToAccount(gomock.Any(), types.StreamAccount, gomock.Any(), gomock.Any()).Return(nil).AnyTimes()

	hour := time.Hour
	gs := types.NewGenesisState(
		[]*types.ContinuousFund{
			{
				Recipient:  "cosmos1qy3529yj3v4xw2z3vz3vz3vz3vz3vz3v3k0vyf",
				Percentage: math.LegacyMustNewDecFromStr("0.1"),
				Expiry:     nil,
			},
		},
		[]*types.Budget{
			{
				RecipientAddress: "cosmos1qy3529yj3v4xw2z3vz3vz3vz3vz3vz3v3k0vyf",
				ClaimedAmount:    &sdk.Coin{},
				LastClaimedAt:    &time.Time{},
				TranchesLeft:     10,
				BudgetPerTranche: &sdk.Coin{Denom: "stake", Amount: math.NewInt(100)},
				Period:           &hour,
			},
		},
	)

	gs.Distributions = append(gs.Distributions, &types.Distribution{
		Amount: types.DistributionAmount{Amount: sdk.NewCoins(sdk.NewCoin("stake", math.NewInt(100)))},
		Time:   &time.Time{},
	})

	err := suite.poolKeeper.InitGenesis(suite.ctx, gs)
	suite.Require().ErrorContains(err, "total to be distributed is greater than the last balance")

	// Set last balance
	gs.LastBalance = types.DistributionAmount{Amount: sdk.NewCoins(sdk.NewCoin("stake", math.NewInt(101)))}
	err = suite.poolKeeper.InitGenesis(suite.ctx, gs)
	suite.Require().NoError(err)

	// Export
	exportedGenState, err := suite.poolKeeper.ExportGenesis(suite.ctx)
	suite.Require().NoError(err)
	suite.Require().Equal(gs.ContinuousFund, exportedGenState.ContinuousFund)
	suite.Require().Equal(gs.Budget, exportedGenState.Budget)
	suite.Require().Equal(math.ZeroInt(), exportedGenState.LastBalance.Amount.AmountOf("stake"))
}