mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-29 17:55:19 +00:00
3375484f79
* Use cosmossdk.io/errors for deprecated error methods * Update error registration with cosmossdk.io/errors * Use cosmossdk.io/math for deprecated sdk.Int alias * Fix modified proto file * Update sdk.Int usage in swap hooks * Update e2e test deprecated method usage
148 lines
3.7 KiB
Go
148 lines
3.7 KiB
Go
package v0_16
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
sdkmath "cosmossdk.io/math"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
app "github.com/kava-labs/kava/app"
|
|
v015swap "github.com/kava-labs/kava/x/swap/legacy/v0_15"
|
|
v016swap "github.com/kava-labs/kava/x/swap/types"
|
|
)
|
|
|
|
type migrateTestSuite struct {
|
|
suite.Suite
|
|
|
|
addresses []sdk.AccAddress
|
|
v15genstate v015swap.GenesisState
|
|
cdc codec.Codec
|
|
legacyCdc *codec.LegacyAmino
|
|
}
|
|
|
|
func (s *migrateTestSuite) SetupTest() {
|
|
app.SetSDKConfig()
|
|
|
|
s.v15genstate = v015swap.GenesisState{
|
|
Params: v015swap.Params{},
|
|
PoolRecords: v015swap.PoolRecords{},
|
|
ShareRecords: v015swap.ShareRecords{},
|
|
}
|
|
|
|
config := app.MakeEncodingConfig()
|
|
s.cdc = config.Marshaler
|
|
|
|
legacyCodec := codec.NewLegacyAmino()
|
|
s.legacyCdc = legacyCodec
|
|
|
|
_, accAddresses := app.GeneratePrivKeyAddressPairs(10)
|
|
s.addresses = accAddresses
|
|
}
|
|
|
|
func (s *migrateTestSuite) TestMigrate_JSON() {
|
|
// Migrate v15 swap to v16
|
|
file := filepath.Join("testdata", "v15-swap.json")
|
|
data, err := ioutil.ReadFile(file)
|
|
s.Require().NoError(err)
|
|
err = s.legacyCdc.UnmarshalJSON(data, &s.v15genstate)
|
|
s.Require().NoError(err)
|
|
genstate := Migrate(s.v15genstate)
|
|
|
|
// Compare expect v16 swap json with migrated json
|
|
actual := s.cdc.MustMarshalJSON(genstate)
|
|
file = filepath.Join("testdata", "v16-swap.json")
|
|
expected, err := ioutil.ReadFile(file)
|
|
s.Require().NoError(err)
|
|
s.Require().JSONEq(string(expected), string(actual))
|
|
}
|
|
|
|
func (s *migrateTestSuite) TestMigrate_Params() {
|
|
params := v015swap.Params{
|
|
SwapFee: sdk.MustNewDecFromStr("0.33"),
|
|
AllowedPools: v015swap.AllowedPools{
|
|
{TokenA: "A", TokenB: "B"},
|
|
{TokenA: "C", TokenB: "D"},
|
|
},
|
|
}
|
|
expectedParams := v016swap.Params{
|
|
SwapFee: sdk.MustNewDecFromStr("0.33"),
|
|
AllowedPools: v016swap.AllowedPools{
|
|
{TokenA: "A", TokenB: "B"},
|
|
{TokenA: "C", TokenB: "D"},
|
|
},
|
|
}
|
|
s.v15genstate.Params = params
|
|
genState := Migrate(s.v15genstate)
|
|
s.Require().Equal(expectedParams, genState.Params)
|
|
}
|
|
|
|
func (s *migrateTestSuite) TestMigrate_PoolRecords() {
|
|
s.v15genstate.PoolRecords = v015swap.PoolRecords{
|
|
{
|
|
PoolID: "pool-1",
|
|
ReservesA: sdk.NewCoin("usdx", sdkmath.NewInt(100)),
|
|
ReservesB: sdk.NewCoin("xrpb", sdkmath.NewInt(200)),
|
|
TotalShares: sdkmath.NewInt(300),
|
|
},
|
|
{
|
|
PoolID: "pool-2",
|
|
ReservesA: sdk.NewCoin("usdx", sdkmath.NewInt(500)),
|
|
ReservesB: sdk.NewCoin("ukava", sdkmath.NewInt(500)),
|
|
TotalShares: sdkmath.NewInt(1000),
|
|
},
|
|
}
|
|
expected := v016swap.PoolRecords{
|
|
{
|
|
PoolID: "pool-1",
|
|
ReservesA: sdk.NewCoin("usdx", sdkmath.NewInt(100)),
|
|
ReservesB: sdk.NewCoin("xrpb", sdkmath.NewInt(200)),
|
|
TotalShares: sdkmath.NewInt(300),
|
|
},
|
|
{
|
|
PoolID: "pool-2",
|
|
ReservesA: sdk.NewCoin("usdx", sdkmath.NewInt(500)),
|
|
ReservesB: sdk.NewCoin("ukava", sdkmath.NewInt(500)),
|
|
TotalShares: sdkmath.NewInt(1000),
|
|
},
|
|
}
|
|
genState := Migrate(s.v15genstate)
|
|
s.Require().Equal(expected, genState.PoolRecords)
|
|
}
|
|
|
|
func (s *migrateTestSuite) TestMigrate_ShareRecords() {
|
|
s.v15genstate.ShareRecords = v015swap.ShareRecords{
|
|
{
|
|
PoolID: "pool-1",
|
|
Depositor: s.addresses[0],
|
|
SharesOwned: sdkmath.NewInt(100),
|
|
},
|
|
{
|
|
PoolID: "pool-2",
|
|
Depositor: s.addresses[1],
|
|
SharesOwned: sdkmath.NewInt(410),
|
|
},
|
|
}
|
|
expected := v016swap.ShareRecords{
|
|
{
|
|
PoolID: "pool-1",
|
|
Depositor: s.addresses[0],
|
|
SharesOwned: sdkmath.NewInt(100),
|
|
},
|
|
{
|
|
PoolID: "pool-2",
|
|
Depositor: s.addresses[1],
|
|
SharesOwned: sdkmath.NewInt(410),
|
|
},
|
|
}
|
|
genState := Migrate(s.v15genstate)
|
|
s.Require().Equal(expected, genState.ShareRecords)
|
|
}
|
|
|
|
func TestMigrateTestSuite(t *testing.T) {
|
|
suite.Run(t, new(migrateTestSuite))
|
|
}
|