106 lines
3.2 KiB
Go
106 lines
3.2 KiB
Go
|
package wallet
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"fmt"
|
||
|
|
"math/big"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/common"
|
||
|
|
"github.com/ethereum/go-ethereum/core"
|
||
|
|
"github.com/ethereum/go-ethereum/core/types"
|
||
|
|
"github.com/ethereum/go-ethereum/crypto"
|
||
|
|
"github.com/ethereum/go-ethereum/event"
|
||
|
|
"github.com/status-im/status-go/multiaccounts/accounts"
|
||
|
|
"github.com/status-im/status-go/t/devtests/testchain"
|
||
|
|
"github.com/status-im/status-go/t/utils"
|
||
|
|
"github.com/stretchr/testify/suite"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestReactorChanges(t *testing.T) {
|
||
|
|
suite.Run(t, new(ReactorChangesSuite))
|
||
|
|
}
|
||
|
|
|
||
|
|
type ReactorChangesSuite struct {
|
||
|
|
suite.Suite
|
||
|
|
backend *testchain.Backend
|
||
|
|
reactor *Reactor
|
||
|
|
db *Database
|
||
|
|
dbStop func()
|
||
|
|
feed *event.Feed
|
||
|
|
|
||
|
|
first, second common.Address
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ReactorChangesSuite) txToAddress(nonce uint64, address common.Address) *types.Transaction {
|
||
|
|
tx := types.NewTransaction(nonce, address, big.NewInt(1e17), 21000, big.NewInt(1), nil)
|
||
|
|
tx, err := types.SignTx(tx, s.backend.Signer, s.backend.Faucet)
|
||
|
|
s.Require().NoError(err)
|
||
|
|
return tx
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ReactorChangesSuite) SetupTest() {
|
||
|
|
var err error
|
||
|
|
db, stop := setupTestDB(s.Suite.T())
|
||
|
|
s.db = db
|
||
|
|
s.dbStop = stop
|
||
|
|
s.backend, err = testchain.NewBackend()
|
||
|
|
s.Require().NoError(err)
|
||
|
|
s.feed = &event.Feed{}
|
||
|
|
s.reactor = NewReactor(s.db, &event.Feed{}, s.backend.Client, big.NewInt(1337))
|
||
|
|
account, err := crypto.GenerateKey()
|
||
|
|
s.Require().NoError(err)
|
||
|
|
s.first = crypto.PubkeyToAddress(account.PublicKey)
|
||
|
|
account, err = crypto.GenerateKey()
|
||
|
|
s.Require().NoError(err)
|
||
|
|
s.second = crypto.PubkeyToAddress(account.PublicKey)
|
||
|
|
nonce := uint64(0)
|
||
|
|
blocks := s.backend.GenerateBlocks(1, 0, func(n int, gen *core.BlockGen) {
|
||
|
|
gen.AddTx(s.txToAddress(nonce, s.first))
|
||
|
|
nonce++
|
||
|
|
gen.AddTx(s.txToAddress(nonce, s.second))
|
||
|
|
nonce++
|
||
|
|
})
|
||
|
|
_, err = s.backend.Ethereum.BlockChain().InsertChain(blocks)
|
||
|
|
s.Require().NoError(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *ReactorChangesSuite) TestWatchNewAccounts() {
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
defer cancel()
|
||
|
|
group := NewGroup(ctx)
|
||
|
|
group.Add(func(ctx context.Context) error {
|
||
|
|
return WatchAccountsChanges(ctx, s.feed, []common.Address{s.first}, s.reactor)
|
||
|
|
})
|
||
|
|
s.Require().NoError(s.reactor.Start([]common.Address{s.first}))
|
||
|
|
s.Require().NoError(utils.Eventually(func() error {
|
||
|
|
transfers, err := s.db.GetTransfersByAddress(s.first, big.NewInt(0), nil)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if len(transfers) != 1 {
|
||
|
|
return fmt.Errorf("expect to get 1 transfer for first address %x, got %d", s.first, len(transfers))
|
||
|
|
}
|
||
|
|
transfers, err = s.db.GetTransfersByAddress(s.second, big.NewInt(0), nil)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if len(transfers) != 0 {
|
||
|
|
return fmt.Errorf("expect not to get any transfer for second address %x", s.second)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}, 5*time.Second, 500*time.Millisecond))
|
||
|
|
s.feed.Send([]accounts.Account{{Address: s.first}, {Address: s.second}})
|
||
|
|
s.Require().NoError(utils.Eventually(func() error {
|
||
|
|
transfers, err := s.db.GetTransfersByAddress(s.second, big.NewInt(0), nil)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if len(transfers) == 0 {
|
||
|
|
return fmt.Errorf("expect 1 transfer for second address %x, got %d", s.second, len(transfers))
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}, 5*time.Second, 500*time.Millisecond))
|
||
|
|
}
|