certusone
/
cosmos-sdk
mirror of https://github.com/certusone/cosmos-sdk.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

refactor TestGetRedelegationsFromSrcValidator to use simapp

This commit is contained in:
Jonathan Gimeno 2020-02-24 12:50:02 +01:00
parent 62dbd8bcee
commit 9a5aae9f0e
2 changed files with 27 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
x/staking/keeper/delegation_test.go
View File

@ -622,30 +622,33 @@ func TestUnbondingAllDelegationFromValidator(t *testing.T) {
require.False(t, found)
}
//// Make sure that that the retrieving the delegations doesn't affect the state
//func TestGetRedelegationsFromSrcValidator(t *testing.T) {
// ctx, _, _, keeper, _ := CreateTestInput(t, false, 0)
//
// rd := types.NewRedelegation(addrDels[0], addrVals[0], addrVals[1], 0,
// time.Unix(0, 0), sdk.NewInt(5),
// sdk.NewDec(5))
//
// // set and retrieve a record
// keeper.SetRedelegation(ctx, rd)
// resBond, found := keeper.GetRedelegation(ctx, addrDels[0], addrVals[0], addrVals[1])
// require.True(t, found)
//
// // get the redelegations one time
// redelegations := keeper.GetRedelegationsFromSrcValidator(ctx, addrVals[0])
// require.Equal(t, 1, len(redelegations))
// require.True(t, redelegations[0].Equal(resBond))
//
// // get the redelegations a second time, should be exactly the same
// redelegations = keeper.GetRedelegationsFromSrcValidator(ctx, addrVals[0])
// require.Equal(t, 1, len(redelegations))
// require.True(t, redelegations[0].Equal(resBond))
//}
//
// Make sure that that the retrieving the delegations doesn't affect the state
func TestGetRedelegationsFromSrcValidator(t *testing.T) {
_, app, ctx := getBaseSimappWithCustomKeeper()
addrDels := simapp.AddTestAddrsIncremental(app, ctx, 2, sdk.NewInt(0))
addrVals := simapp.ConvertAddrsToValAddrs(addrDels)
rd := types.NewRedelegation(addrDels[0], addrVals[0], addrVals[1], 0,
time.Unix(0, 0), sdk.NewInt(5),
sdk.NewDec(5))
// set and retrieve a record
app.StakingKeeper.SetRedelegation(ctx, rd)
resBond, found := app.StakingKeeper.GetRedelegation(ctx, addrDels[0], addrVals[0], addrVals[1])
require.True(t, found)
// get the redelegations one time
redelegations := app.StakingKeeper.GetRedelegationsFromSrcValidator(ctx, addrVals[0])
require.Equal(t, 1, len(redelegations))
require.True(t, redelegations[0].Equal(resBond))
// get the redelegations a second time, should be exactly the same
redelegations = app.StakingKeeper.GetRedelegationsFromSrcValidator(ctx, addrVals[0])
require.Equal(t, 1, len(redelegations))
require.True(t, redelegations[0].Equal(resBond))
}
//// tests Get/Set/Remove/Has UnbondingDelegation
//func TestRedelegation(t *testing.T) {
// ctx, _, _, keeper, _ := CreateTestInput(t, false, 0)

24
x/staking/keeper/old_delegation_test.go
View File

@ -11,30 +11,6 @@ import (
"github.com/cosmos/cosmos-sdk/x/staking/types"
)
// Make sure that that the retrieving the delegations doesn't affect the state
func TestGetRedelegationsFromSrcValidator(t *testing.T) {
ctx, _, _, keeper, _ := CreateTestInput(t, false, 0)
rd := types.NewRedelegation(addrDels[0], addrVals[0], addrVals[1], 0,
time.Unix(0, 0), sdk.NewInt(5),
sdk.NewDec(5))
// set and retrieve a record
keeper.SetRedelegation(ctx, rd)
resBond, found := keeper.GetRedelegation(ctx, addrDels[0], addrVals[0], addrVals[1])
require.True(t, found)
// get the redelegations one time
redelegations := keeper.GetRedelegationsFromSrcValidator(ctx, addrVals[0])
require.Equal(t, 1, len(redelegations))
require.True(t, redelegations[0].Equal(resBond))
// get the redelegations a second time, should be exactly the same
redelegations = keeper.GetRedelegationsFromSrcValidator(ctx, addrVals[0])
require.Equal(t, 1, len(redelegations))
require.True(t, redelegations[0].Equal(resBond))
}
// tests Get/Set/Remove/Has UnbondingDelegation
func TestRedelegation(t *testing.T) {
ctx, _, _, keeper, _ := CreateTestInput(t, false, 0)