mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-15 04:25:27 +00:00
120 lines
4.0 KiB
Go
120 lines
4.0 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"time"
|
|
|
|
sdkmath "cosmossdk.io/math"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
authtypes "github.com/cosmos/cosmos-sdk/x/auth/types"
|
|
|
|
"github.com/cometbft/cometbft/crypto"
|
|
tmtime "github.com/cometbft/cometbft/types/time"
|
|
|
|
"github.com/0glabs/0g-chain/app"
|
|
"github.com/0glabs/0g-chain/x/bep3/types"
|
|
)
|
|
|
|
const (
|
|
TestSenderOtherChain = "bnb1uky3me9ggqypmrsvxk7ur6hqkzq7zmv4ed4ng7"
|
|
TestRecipientOtherChain = "bnb1urfermcg92dwq36572cx4xg84wpk3lfpksr5g7"
|
|
TestDeputy = "kava1xy7hrjy9r0algz9w3gzm8u6mrpq97kwta747gj"
|
|
)
|
|
|
|
var (
|
|
DenomMap = map[int]string{0: "btc", 1: "eth", 2: "bnb", 3: "xrp", 4: "dai"}
|
|
TestUser1 = sdk.AccAddress(crypto.AddressHash([]byte("KavaTestUser1")))
|
|
TestUser2 = sdk.AccAddress(crypto.AddressHash([]byte("KavaTestUser2")))
|
|
)
|
|
|
|
func c(denom string, amount int64) sdk.Coin { return sdk.NewInt64Coin(denom, amount) }
|
|
func cs(coins ...sdk.Coin) sdk.Coins { return sdk.NewCoins(coins...) }
|
|
func ts(minOffset int) int64 { return tmtime.Now().Add(time.Duration(minOffset) * time.Minute).Unix() }
|
|
|
|
func NewAuthGenStateFromAccs(cdc codec.JSONCodec, accounts ...authtypes.GenesisAccount) app.GenesisState {
|
|
authGenesis := authtypes.NewGenesisState(authtypes.DefaultParams(), accounts)
|
|
return app.GenesisState{authtypes.ModuleName: cdc.MustMarshalJSON(authGenesis)}
|
|
}
|
|
|
|
func NewBep3GenStateMulti(cdc codec.JSONCodec, deputyAddress sdk.AccAddress) app.GenesisState {
|
|
bep3Genesis := types.GenesisState{
|
|
Params: types.Params{
|
|
AssetParams: types.AssetParams{
|
|
{
|
|
Denom: "bnb",
|
|
CoinID: 714,
|
|
SupplyLimit: types.SupplyLimit{
|
|
Limit: sdkmath.NewInt(350000000000000),
|
|
TimeLimited: false,
|
|
TimeBasedLimit: sdk.ZeroInt(),
|
|
TimePeriod: time.Hour,
|
|
},
|
|
Active: true,
|
|
DeputyAddress: deputyAddress,
|
|
FixedFee: sdkmath.NewInt(1000),
|
|
MinSwapAmount: sdk.OneInt(),
|
|
MaxSwapAmount: sdkmath.NewInt(1000000000000),
|
|
MinBlockLock: types.DefaultMinBlockLock,
|
|
MaxBlockLock: types.DefaultMaxBlockLock,
|
|
},
|
|
{
|
|
Denom: "inc",
|
|
CoinID: 9999,
|
|
SupplyLimit: types.SupplyLimit{
|
|
Limit: sdkmath.NewInt(100000000000000),
|
|
TimeLimited: true,
|
|
TimeBasedLimit: sdkmath.NewInt(50000000000),
|
|
TimePeriod: time.Hour,
|
|
},
|
|
Active: false,
|
|
DeputyAddress: deputyAddress,
|
|
FixedFee: sdkmath.NewInt(1000),
|
|
MinSwapAmount: sdk.OneInt(),
|
|
MaxSwapAmount: sdkmath.NewInt(100000000000),
|
|
MinBlockLock: types.DefaultMinBlockLock,
|
|
MaxBlockLock: types.DefaultMaxBlockLock,
|
|
},
|
|
},
|
|
},
|
|
Supplies: types.AssetSupplies{
|
|
types.NewAssetSupply(
|
|
sdk.NewCoin("bnb", sdk.ZeroInt()),
|
|
sdk.NewCoin("bnb", sdk.ZeroInt()),
|
|
sdk.NewCoin("bnb", sdk.ZeroInt()),
|
|
sdk.NewCoin("bnb", sdk.ZeroInt()),
|
|
time.Duration(0),
|
|
),
|
|
types.NewAssetSupply(
|
|
sdk.NewCoin("inc", sdk.ZeroInt()),
|
|
sdk.NewCoin("inc", sdk.ZeroInt()),
|
|
sdk.NewCoin("inc", sdk.ZeroInt()),
|
|
sdk.NewCoin("inc", sdk.ZeroInt()),
|
|
time.Duration(0),
|
|
),
|
|
},
|
|
PreviousBlockTime: types.DefaultPreviousBlockTime,
|
|
}
|
|
return app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&bep3Genesis)}
|
|
}
|
|
|
|
func atomicSwaps(ctx sdk.Context, count int) types.AtomicSwaps {
|
|
var swaps types.AtomicSwaps
|
|
for i := 0; i < count; i++ {
|
|
swap := atomicSwap(ctx, i)
|
|
swaps = append(swaps, swap)
|
|
}
|
|
return swaps
|
|
}
|
|
|
|
func atomicSwap(ctx sdk.Context, index int) types.AtomicSwap {
|
|
expireOffset := uint64(200) // Default expire height + offet to match timestamp
|
|
timestamp := ts(index) // One minute apart
|
|
randomNumber, _ := types.GenerateSecureRandomNumber()
|
|
randomNumberHash := types.CalculateRandomHash(randomNumber[:], timestamp)
|
|
|
|
return types.NewAtomicSwap(cs(c("bnb", 50000)), randomNumberHash,
|
|
uint64(ctx.BlockHeight())+expireOffset, timestamp, TestUser1, TestUser2,
|
|
TestSenderOtherChain, TestRecipientOtherChain, 0, types.SWAP_STATUS_OPEN, true,
|
|
types.SWAP_DIRECTION_INCOMING)
|
|
}
|