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 = "0g1ffv7nhd3z6sych2qpqkk03ec6hzkmufyhp5hf8" ) var ( DenomMap = map[int]string{0: "btc", 1: "eth", 2: "bnb", 3: "xrp", 4: "dai"} TestUser1 = sdk.AccAddress(crypto.AddressHash([]byte("0gTestUser1"))) TestUser2 = sdk.AccAddress(crypto.AddressHash([]byte("0gTestUser2"))) ) 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) }