status-go/services/wallet/downloader_test.go

263 lines
8.7 KiB
Go

package wallet
import (
"context"
"crypto/ecdsa"
"math/big"
"testing"
"time"
"github.com/stretchr/testify/suite"
"github.com/ethereum/go-ethereum/accounts/abi/bind"
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/core/types"
"github.com/ethereum/go-ethereum/crypto"
"github.com/ethereum/go-ethereum/ethclient"
"github.com/status-im/status-go/services/wallet/erc20"
"github.com/status-im/status-go/t/devtests/miner"
)
func TestETHTransfers(t *testing.T) {
suite.Run(t, new(ETHTransferSuite))
}
type ETHTransferSuite struct {
suite.Suite
ethclient *ethclient.Client
identity, secondary *ecdsa.PrivateKey
faucet *ecdsa.PrivateKey
signer types.Signer
downloader *ETHTransferDownloader
}
func (s *ETHTransferSuite) SetupTest() {
var err error
s.identity, err = crypto.GenerateKey()
s.Require().NoError(err)
s.faucet, err = crypto.GenerateKey()
s.Require().NoError(err)
s.secondary, err = crypto.GenerateKey()
s.Require().NoError(err)
node, err := miner.NewDevNode(crypto.PubkeyToAddress(s.faucet.PublicKey))
s.Require().NoError(err)
s.Require().NoError(miner.StartWithMiner(node))
client, err := node.Attach()
s.Require().NoError(err)
s.ethclient = ethclient.NewClient(client)
s.signer = types.NewEIP155Signer(big.NewInt(1337))
s.downloader = &ETHTransferDownloader{
signer: s.signer,
client: s.ethclient,
accounts: []common.Address{
crypto.PubkeyToAddress(s.identity.PublicKey),
crypto.PubkeyToAddress(s.secondary.PublicKey)},
}
}
// signAndMineTx signs transaction with provided key and waits for it to be mined.
// uses configured faucet key if pkey is nil.
func (s *ETHTransferSuite) signAndMineTx(tx *types.Transaction, pkey *ecdsa.PrivateKey) {
if pkey == nil {
pkey = s.faucet
}
tx, err := types.SignTx(tx, s.signer, pkey)
s.Require().NoError(err)
s.Require().NoError(s.ethclient.SendTransaction(context.Background(), tx))
timeout, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
_, err = bind.WaitMined(timeout, s.ethclient, tx)
s.Require().NoError(err)
}
func (s *ETHTransferSuite) TestNoBalance() {
ctx := context.TODO()
tx := types.NewTransaction(0, common.Address{1}, big.NewInt(1e18), 1e6, big.NewInt(10), nil)
s.signAndMineTx(tx, nil)
header, err := s.ethclient.HeaderByNumber(ctx, nil)
s.Require().NoError(err)
transfers, err := s.downloader.GetTransfers(ctx, toDBHeader(header))
s.Require().NoError(err)
s.Require().Empty(transfers)
}
func (s *ETHTransferSuite) TestBalanceUpdatedOnInbound() {
ctx := context.TODO()
tx := types.NewTransaction(0, crypto.PubkeyToAddress(s.identity.PublicKey), big.NewInt(1e18), 1e6, big.NewInt(10), nil)
tx, err := types.SignTx(tx, s.signer, s.faucet)
s.Require().NoError(err)
s.Require().NoError(s.ethclient.SendTransaction(ctx, tx))
timeout, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
_, err = bind.WaitMined(timeout, s.ethclient, tx)
s.Require().NoError(err)
header, err := s.ethclient.HeaderByNumber(ctx, nil)
s.Require().NoError(err)
s.Require().Equal(big.NewInt(1), header.Number)
transfers, err := s.downloader.GetTransfers(ctx, toDBHeader(header))
s.Require().NoError(err)
s.Require().Len(transfers, 1)
}
func (s *ETHTransferSuite) TestBalanceUpdatedOnOutbound() {
ctx := context.TODO()
tx := types.NewTransaction(0, crypto.PubkeyToAddress(s.identity.PublicKey), big.NewInt(1e18), 1e6, big.NewInt(10), nil)
tx, err := types.SignTx(tx, s.signer, s.faucet)
s.Require().NoError(err)
s.Require().NoError(s.ethclient.SendTransaction(ctx, tx))
timeout, cancel := context.WithTimeout(context.Background(), 5*time.Second)
_, err = bind.WaitMined(timeout, s.ethclient, tx)
cancel()
s.Require().NoError(err)
tx = types.NewTransaction(0, common.Address{1}, big.NewInt(5e17), 1e6, big.NewInt(10), nil)
tx, err = types.SignTx(tx, s.signer, s.identity)
s.Require().NoError(err)
s.Require().NoError(s.ethclient.SendTransaction(ctx, tx))
timeout, cancel = context.WithTimeout(context.Background(), 5*time.Second)
_, err = bind.WaitMined(timeout, s.ethclient, tx)
cancel()
s.Require().NoError(err)
header, err := s.ethclient.HeaderByNumber(ctx, nil)
s.Require().NoError(err)
s.Require().Equal(big.NewInt(2), header.Number)
transfers, err := s.downloader.GetTransfers(ctx, toDBHeader(header))
s.Require().NoError(err)
s.Require().Len(transfers, 1)
}
func (s *ETHTransferSuite) TestMultipleReferences() {
tx := types.NewTransaction(0, crypto.PubkeyToAddress(s.identity.PublicKey), big.NewInt(1e18), 1e6, big.NewInt(10), nil)
s.signAndMineTx(tx, nil)
tx = types.NewTransaction(0, crypto.PubkeyToAddress(s.secondary.PublicKey), big.NewInt(1e17), 1e6, big.NewInt(10), nil)
s.signAndMineTx(tx, s.identity)
header, err := s.ethclient.HeaderByNumber(context.Background(), nil)
s.Require().NoError(err)
s.Require().Equal(big.NewInt(2), header.Number)
transfers, err := s.downloader.GetTransfers(context.Background(), toDBHeader(header))
s.Require().NoError(err)
s.Require().Len(transfers, 2)
}
func TestERC20Transfers(t *testing.T) {
suite.Run(t, new(ERC20TransferSuite))
}
type ERC20TransferSuite struct {
suite.Suite
ethclient *ethclient.Client
identity *ecdsa.PrivateKey
faucet *ecdsa.PrivateKey
signer types.Signer
downloader *ERC20TransfersDownloader
contract *erc20.ERC20Transfer
}
func (s *ERC20TransferSuite) SetupTest() {
var err error
s.signer = types.NewEIP155Signer(big.NewInt(1337))
s.identity, err = crypto.GenerateKey()
s.Require().NoError(err)
s.faucet, err = crypto.GenerateKey()
s.Require().NoError(err)
node, err := miner.NewDevNode(crypto.PubkeyToAddress(s.faucet.PublicKey))
s.Require().NoError(err)
s.Require().NoError(miner.StartWithMiner(node))
client, err := node.Attach()
s.Require().NoError(err)
s.ethclient = ethclient.NewClient(client)
s.downloader = NewERC20TransfersDownloader(s.ethclient, []common.Address{crypto.PubkeyToAddress(s.identity.PublicKey)}, s.signer)
_, tx, contract, err := erc20.DeployERC20Transfer(bind.NewKeyedTransactor(s.faucet), s.ethclient)
s.Require().NoError(err)
timeout, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
_, err = bind.WaitMined(timeout, s.ethclient, tx)
s.Require().NoError(err)
s.contract = contract
}
func (s *ERC20TransferSuite) TestNoEvents() {
header, err := s.ethclient.HeaderByNumber(context.TODO(), nil)
s.Require().NoError(err)
transfers, err := s.downloader.GetTransfers(context.TODO(), toDBHeader(header))
s.Require().NoError(err)
s.Require().Empty(transfers)
}
func (s *ERC20TransferSuite) TestInboundEvent() {
tx, err := s.contract.Transfer(bind.NewKeyedTransactor(s.faucet), crypto.PubkeyToAddress(s.identity.PublicKey),
big.NewInt(100))
s.Require().NoError(err)
timeout, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
_, err = bind.WaitMined(timeout, s.ethclient, tx)
s.Require().NoError(err)
header, err := s.ethclient.HeaderByNumber(context.TODO(), nil)
s.Require().NoError(err)
transfers, err := s.downloader.GetTransfers(context.TODO(), toDBHeader(header))
s.Require().NoError(err)
s.Require().Len(transfers, 1)
}
func (s *ERC20TransferSuite) TestOutboundEvent() {
// give some eth to pay for gas
ctx := context.TODO()
// nonce is 1 - contact with nonce 0 was deployed in setup
// FIXME request nonce
tx := types.NewTransaction(1, crypto.PubkeyToAddress(s.identity.PublicKey), big.NewInt(1e18), 1e6, big.NewInt(10), nil)
tx, err := types.SignTx(tx, s.signer, s.faucet)
s.Require().NoError(err)
s.Require().NoError(s.ethclient.SendTransaction(ctx, tx))
timeout, cancel := context.WithTimeout(context.Background(), 5*time.Second)
_, err = bind.WaitMined(timeout, s.ethclient, tx)
cancel()
s.Require().NoError(err)
tx, err = s.contract.Transfer(bind.NewKeyedTransactor(s.identity), common.Address{1}, big.NewInt(100))
s.Require().NoError(err)
timeout, cancel = context.WithTimeout(context.Background(), 5*time.Second)
_, err = bind.WaitMined(timeout, s.ethclient, tx)
cancel()
s.Require().NoError(err)
header, err := s.ethclient.HeaderByNumber(context.TODO(), nil)
s.Require().NoError(err)
transfers, err := s.downloader.GetTransfers(context.TODO(), toDBHeader(header))
s.Require().NoError(err)
s.Require().Len(transfers, 1)
}
func (s *ERC20TransferSuite) TestInRange() {
for i := 0; i < 5; i++ {
tx, err := s.contract.Transfer(bind.NewKeyedTransactor(s.faucet), crypto.PubkeyToAddress(s.identity.PublicKey),
big.NewInt(100))
s.Require().NoError(err)
timeout, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
_, err = bind.WaitMined(timeout, s.ethclient, tx)
s.Require().NoError(err)
}
transfers, err := s.downloader.GetTransfersInRange(context.TODO(), big.NewInt(1), nil)
s.Require().NoError(err)
s.Require().Len(transfers, 5)
}