mirror of
				https://github.com/0glabs/0g-chain.git
				synced 2025-11-04 03:07:28 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			120 lines
		
	
	
		
			3.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			120 lines
		
	
	
		
			3.9 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              = "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)
 | 
						|
}
 |