LSM staking migration

x/staking/legacy/v045/migrate.go

@@ -0,0 +1,39 @@
+package v11
+
+import (
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	stakingkeeper "github.com/cosmos/cosmos-sdk/x/staking/keeper"
+)
+
+// Set initial param values, based on https://github.com/iqlusioninc/liquidity-staking-module/blob/master/x/staking/spec/08_params.md
+func SetParamsStaking(ctx sdk.Context, k stakingkeeper.Keeper) {
+	params := k.GetParams(ctx)
+
+	params.ValidatorBondFactor = sdk.Dec(sdk.NewInt(250))
+	params.GlobalLiquidStakingCap = sdk.Dec(sdk.NewInt(25).Quo(sdk.NewInt(100)))
+	params.ValidatorLiquidStakingCap = sdk.Dec(sdk.NewInt(50).Quo(sdk.NewInt(100)))
+
+	k.SetParams(ctx, params)
+}
+
+// Set each validator's TotalValidatorBondShares and TotalLiquidShares to 0
+func SetAllValidatorBondAndLiquidSharesToZero(ctx sdk.Context, k stakingkeeper.Keeper) {
+
+	for _, Val := range k.GetAllValidators(ctx) {
+
+		Val.TotalValidatorBondShares = sdk.ZeroDec()
+		Val.TotalLiquidShares = sdk.ZeroDec()
+
+		k.SetValidator(ctx, Val)
+	}
+}
+
+// Set each validator's ValidatorBond to false
+func SetAllDelegationValidatorBondsFalse(ctx sdk.Context, k stakingkeeper.Keeper) {
+	for _, Del := range k.GetAllDelegations(ctx) {
+
+		Del.ValidatorBond = false
+
+		k.SetDelegation(ctx, Del)
+	}
+}

x/staking/legacy/v045/migrate_test.go

@@ -0,0 +1,97 @@
+package v11
+
+import (
+	"testing"
+
+	"github.com/cosmos/cosmos-sdk/simapp"
+	sdk "github.com/cosmos/cosmos-sdk/types"
+	stakingtypes "github.com/cosmos/cosmos-sdk/x/staking/types"
+	"github.com/stretchr/testify/require"
+	tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
+)
+
+// Test setting params in the staking module
+func TestSetParamsStaking(t *testing.T) {
+
+	app := simapp.Setup(false)
+	ctx := app.BaseApp.NewContext(false, tmproto.Header{})
+
+	// set the params in the staking store; exclude the new v11 params
+	params := app.StakingKeeper.GetParams(ctx)
+	preUpgradeParams := stakingtypes.Params{
+		UnbondingTime:     params.UnbondingTime,
+		MaxValidators:     params.MaxValidators,
+		MaxEntries:        params.MaxEntries,
+		HistoricalEntries: params.HistoricalEntries,
+		BondDenom:         params.BondDenom,
+	}
+	app.StakingKeeper.SetParams(ctx, preUpgradeParams)
+
+	SetParamsStaking(ctx, app.StakingKeeper)
+
+	// check that the params were set correctly
+	params = app.StakingKeeper.GetParams(ctx)
+	require.Equal(t, sdk.NewDec(250), params.ValidatorBondFactor)
+	require.Equal(t, sdk.NewDec(25).Quo(sdk.NewDec(100)), params.GlobalLiquidStakingCap)
+	require.Equal(t, sdk.NewDec(50).Quo(sdk.NewDec(100)), params.ValidatorLiquidStakingCap)
+}
+
+// Test setting each validator's TotalValidatorBondShares and TotalLiquidShares to 0
+func TestSetAllValidatorBondAndLiquidSharesToZero(t *testing.T) {
+
+	app := simapp.Setup(false)
+	ctx := app.BaseApp.NewContext(false, tmproto.Header{})
+
+	for _, Val := range app.StakingKeeper.GetAllValidators(ctx) {
+
+		// set the validator attributes on each val; exclude the new v11 attributes
+		preUpgradeValidator := stakingtypes.Validator{
+			OperatorAddress:   Val.OperatorAddress,
+			ConsensusPubkey:   Val.ConsensusPubkey,
+			Jailed:            Val.Jailed,
+			Status:            Val.Status,
+			Tokens:            Val.Tokens,
+			DelegatorShares:   Val.DelegatorShares,
+			Description:       Val.Description,
+			UnbondingHeight:   Val.UnbondingHeight,
+			MinSelfDelegation: Val.MinSelfDelegation,
+			Commission:        Val.Commission,
+			UnbondingTime:     Val.UnbondingTime,
+		}
+		app.StakingKeeper.SetValidator(ctx, preUpgradeValidator)
+	}
+
+	SetAllValidatorBondAndLiquidSharesToZero(ctx, app.StakingKeeper)
+
+	// check that the validator TotalValidatorBondShares and TotalLiquidShares were correctly set to 0
+	for _, Val := range app.StakingKeeper.GetAllValidators(ctx) {
+		require.Equal(t, sdk.ZeroDec(), Val.TotalValidatorBondShares)
+		require.Equal(t, sdk.ZeroDec(), Val.TotalLiquidShares)
+	}
+}
+
+// Test setting each validator's TotalDelegatorBondShares to 0
+func TestSetAllDelegatorBondSharesToZero(t *testing.T) {
+
+	app := simapp.Setup(false)
+	ctx := app.BaseApp.NewContext(false, tmproto.Header{})
+
+	for _, Del := range app.StakingKeeper.GetAllDelegations(ctx) {
+
+		// set the delegation attributes on each del; exclude the new v11 attributes
+		preUpgradeDelegation := stakingtypes.Delegation{
+			DelegatorAddress: Del.DelegatorAddress,
+			ValidatorAddress: Del.ValidatorAddress,
+			Shares:           Del.Shares,
+		}
+		app.StakingKeeper.SetDelegation(ctx, preUpgradeDelegation)
+	}
+
+	SetAllDelegationValidatorBondsFalse(ctx, app.StakingKeeper)
+
+	// check that the delegation ValidatorBond was correctly set to false
+	for _, Del := range app.StakingKeeper.GetAllDelegations(ctx) {
+		require.Equal(t, false, Del.ValidatorBond)
+	}
+
+}