122 lines
2.8 KiB
Go
122 lines
2.8 KiB
Go
// nolint
|
|
package types
|
|
|
|
import (
|
|
"crypto/ecdsa"
|
|
"math/big"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/wire"
|
|
"github.com/cosmos/cosmos-sdk/x/auth"
|
|
ethcmn "github.com/ethereum/go-ethereum/common"
|
|
ethtypes "github.com/ethereum/go-ethereum/core/types"
|
|
ethcrypto "github.com/ethereum/go-ethereum/crypto"
|
|
)
|
|
|
|
var (
|
|
TestSDKAddr = GenerateEthAddress()
|
|
TestChainID = big.NewInt(3)
|
|
|
|
TestPrivKey1, _ = ethcrypto.GenerateKey()
|
|
TestPrivKey2, _ = ethcrypto.GenerateKey()
|
|
|
|
TestAddr1 = PrivKeyToEthAddress(TestPrivKey1)
|
|
TestAddr2 = PrivKeyToEthAddress(TestPrivKey2)
|
|
)
|
|
|
|
func NewTestCodec() *wire.Codec {
|
|
codec := wire.NewCodec()
|
|
|
|
RegisterWire(codec)
|
|
auth.RegisterWire(codec)
|
|
wire.RegisterCrypto(codec)
|
|
codec.RegisterConcrete(&sdk.TestMsg{}, "test/TestMsg", nil)
|
|
|
|
return codec
|
|
}
|
|
|
|
func NewTestStdFee() auth.StdFee {
|
|
return auth.NewStdFee(5000, sdk.NewCoin("photon", sdk.NewInt(150)))
|
|
}
|
|
|
|
func NewTestStdTx(
|
|
chainID *big.Int, msgs []sdk.Msg, accNums []int64, seqs []int64, pKeys []*ecdsa.PrivateKey, fee auth.StdFee,
|
|
) sdk.Tx {
|
|
|
|
sigs := make([]auth.StdSignature, len(pKeys))
|
|
|
|
for i, priv := range pKeys {
|
|
signBytes := GetStdTxSignBytes(chainID.String(), accNums[i], seqs[i], NewTestStdFee(), msgs, "")
|
|
|
|
sig, err := ethcrypto.Sign(signBytes, priv)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
sigs[i] = auth.StdSignature{Signature: sig, AccountNumber: accNums[i], Sequence: seqs[i]}
|
|
}
|
|
|
|
return auth.NewStdTx(msgs, fee, sigs, "")
|
|
}
|
|
|
|
func NewTestGethTxs(
|
|
chainID *big.Int, seqs []int64, addrs []ethcmn.Address, pKeys []*ecdsa.PrivateKey,
|
|
) []*ethtypes.Transaction {
|
|
|
|
txs := make([]*ethtypes.Transaction, len(pKeys))
|
|
|
|
for i, privKey := range pKeys {
|
|
ethTx := ethtypes.NewTransaction(
|
|
uint64(seqs[i]), addrs[i], big.NewInt(10), 1000, big.NewInt(100), []byte{},
|
|
)
|
|
|
|
signer := ethtypes.NewEIP155Signer(chainID)
|
|
|
|
ethTx, err := ethtypes.SignTx(ethTx, signer, privKey)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
txs[i] = ethTx
|
|
}
|
|
|
|
return txs
|
|
}
|
|
|
|
func NewTestEthTxs(
|
|
chainID *big.Int, seqs []int64, addrs []ethcmn.Address, pKeys []*ecdsa.PrivateKey,
|
|
) []*Transaction {
|
|
|
|
txs := make([]*Transaction, len(pKeys))
|
|
|
|
for i, privKey := range pKeys {
|
|
ethTx := NewTransaction(
|
|
uint64(seqs[i]), addrs[i], big.NewInt(10), 1000, big.NewInt(100), []byte{},
|
|
)
|
|
|
|
ethTx.Sign(chainID, privKey)
|
|
txs[i] = ethTx
|
|
}
|
|
|
|
return txs
|
|
}
|
|
|
|
func NewTestSDKTxs(
|
|
codec *wire.Codec, chainID *big.Int, to ethcmn.Address, msgs []sdk.Msg,
|
|
accNums []int64, seqs []int64, pKeys []*ecdsa.PrivateKey, fee auth.StdFee,
|
|
) []*Transaction {
|
|
|
|
txs := make([]*Transaction, len(pKeys))
|
|
stdTx := NewTestStdTx(chainID, msgs, accNums, seqs, pKeys, fee)
|
|
payload := codec.MustMarshalBinary(stdTx)
|
|
|
|
for i, privKey := range pKeys {
|
|
ethTx := NewTransaction(uint64(seqs[i]), to, big.NewInt(10), 1000, big.NewInt(100), payload)
|
|
|
|
ethTx.Sign(chainID, privKey)
|
|
txs[i] = ethTx
|
|
}
|
|
|
|
return txs
|
|
}
|