107 lines
2.8 KiB
Go
107 lines
2.8 KiB
Go
|
package destructive
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/status-im/status-go/geth/api"
|
||
|
|
"github.com/status-im/status-go/geth/log"
|
||
|
|
"github.com/status-im/status-go/t/e2e"
|
||
|
|
. "github.com/status-im/status-go/t/utils"
|
||
|
|
|
||
|
|
"github.com/ethereum/go-ethereum/p2p"
|
||
|
|
"github.com/ethereum/go-ethereum/p2p/discover"
|
||
|
|
"github.com/stretchr/testify/suite"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
defaultTimeout = 40 * time.Second
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestPeersSuiteLinkUpDown(t *testing.T) {
|
||
|
|
suite.Run(t, &PeersTestSuite{controller: new(NetworkConnectionController)})
|
||
|
|
}
|
||
|
|
|
||
|
|
type PeersTestSuite struct {
|
||
|
|
suite.Suite
|
||
|
|
|
||
|
|
backend *api.StatusBackend
|
||
|
|
controller *NetworkConnectionController
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *PeersTestSuite) SetupTest() {
|
||
|
|
s.backend = api.NewStatusBackend()
|
||
|
|
config, err := e2e.MakeTestNodeConfig(GetNetworkID())
|
||
|
|
s.Require().NoError(err)
|
||
|
|
// we need to enable atleast 1 protocol, otherwise peers won't connect
|
||
|
|
config.LightEthConfig.Enabled = false
|
||
|
|
config.WhisperConfig.Enabled = true
|
||
|
|
s.Require().NoError(s.backend.StartNode(config))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *PeersTestSuite) TearDownTest() {
|
||
|
|
s.Require().NoError(s.backend.StopNode())
|
||
|
|
}
|
||
|
|
|
||
|
|
func consumeUntil(events <-chan *p2p.PeerEvent, f func(ev *p2p.PeerEvent) bool, timeout time.Duration) error {
|
||
|
|
timer := time.After(timeout)
|
||
|
|
for {
|
||
|
|
select {
|
||
|
|
case ev := <-events:
|
||
|
|
if f(ev) {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
case <-timer:
|
||
|
|
return errors.New("timeout")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func (s *PeersTestSuite) TestStaticPeersReconnect() {
|
||
|
|
// both on rinkeby and ropsten we can expect atleast 2 peers connected
|
||
|
|
expectedPeersCount := 2
|
||
|
|
events := make(chan *p2p.PeerEvent, 10)
|
||
|
|
node, err := s.backend.NodeManager().Node()
|
||
|
|
s.Require().NoError(err)
|
||
|
|
|
||
|
|
subscription := node.Server().SubscribeEvents(events)
|
||
|
|
defer subscription.Unsubscribe()
|
||
|
|
peers := map[discover.NodeID]struct{}{}
|
||
|
|
before := time.Now()
|
||
|
|
s.Require().NoError(consumeUntil(events, func(ev *p2p.PeerEvent) bool {
|
||
|
|
log.Info("tests", "event", ev)
|
||
|
|
if ev.Type == p2p.PeerEventTypeAdd {
|
||
|
|
peers[ev.Peer] = struct{}{}
|
||
|
|
}
|
||
|
|
return len(peers) == expectedPeersCount
|
||
|
|
}, defaultTimeout))
|
||
|
|
s.WithinDuration(time.Now(), before, 5*time.Second)
|
||
|
|
|
||
|
|
s.Require().NoError(s.controller.Enable())
|
||
|
|
before = time.Now()
|
||
|
|
|
||
|
|
s.Require().NoError(consumeUntil(events, func(ev *p2p.PeerEvent) bool {
|
||
|
|
log.Info("tests", "event", ev)
|
||
|
|
if ev.Type == p2p.PeerEventTypeDrop {
|
||
|
|
delete(peers, ev.Peer)
|
||
|
|
}
|
||
|
|
return len(peers) == 0
|
||
|
|
}, defaultTimeout))
|
||
|
|
s.WithinDuration(time.Now(), before, 31*time.Second)
|
||
|
|
|
||
|
|
s.Require().NoError(s.controller.Disable())
|
||
|
|
before = time.Now()
|
||
|
|
go func() {
|
||
|
|
s.NoError(s.backend.NodeManager().ReconnectStaticPeers())
|
||
|
|
}()
|
||
|
|
s.Require().NoError(consumeUntil(events, func(ev *p2p.PeerEvent) bool {
|
||
|
|
log.Info("tests", "event", ev)
|
||
|
|
if ev.Type == p2p.PeerEventTypeAdd {
|
||
|
|
peers[ev.Peer] = struct{}{}
|
||
|
|
}
|
||
|
|
return len(peers) == expectedPeersCount
|
||
|
|
}, defaultTimeout))
|
||
|
|
s.WithinDuration(time.Now(), before, 31*time.Second)
|
||
|
|
}
|