package stake

import (
	"testing"

	"github.com/stretchr/testify/require"

	sdk "github.com/cosmos/cosmos-sdk/types"
	keep "github.com/cosmos/cosmos-sdk/x/stake/keeper"
	"github.com/cosmos/cosmos-sdk/x/stake/types"
)

func TestInitGenesis(t *testing.T) {
	ctx, _, keeper := keep.CreateTestInput(t, false, 1000)

	pool := keeper.GetPool(ctx)
	pool.LooseTokens = sdk.OneRat()

	params := keeper.GetParams(ctx)
	var delegations []Delegation

	validators := []Validator{
		NewValidator(keep.Addrs[0], keep.PKs[0], Description{Moniker: "hoop"}),
	}

	genesisState := types.NewGenesisState(pool, params, validators, delegations)
	err := InitGenesis(ctx, keeper, genesisState)
	require.Error(t, err)

	validators[0].Tokens = sdk.OneRat()
	validators[0].DelegatorShares = sdk.OneRat()

	genesisState = types.NewGenesisState(pool, params, validators, delegations)
	err = InitGenesis(ctx, keeper, genesisState)
	require.NoError(t, err)
}