361 lines
10 KiB
Go
361 lines
10 KiB
Go
package node_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/accounts"
|
|
"github.com/ethereum/go-ethereum/les"
|
|
gethnode "github.com/ethereum/go-ethereum/node"
|
|
"github.com/ethereum/go-ethereum/rpc"
|
|
|
|
"github.com/status-im/status-go/node"
|
|
"github.com/status-im/status-go/params"
|
|
"github.com/status-im/status-go/whisper/v6"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
"github.com/status-im/status-go/t/e2e"
|
|
. "github.com/status-im/status-go/t/utils"
|
|
)
|
|
|
|
func TestManagerTestSuite(t *testing.T) {
|
|
e2e.Init()
|
|
suite.Run(t, new(ManagerTestSuite))
|
|
}
|
|
|
|
type ManagerTestSuite struct {
|
|
e2e.StatusNodeTestSuite
|
|
}
|
|
|
|
func (s *ManagerTestSuite) SetupTest() {
|
|
s.StatusNode = node.New()
|
|
}
|
|
|
|
func (s *ManagerTestSuite) TestReferencesWithoutStartedNode() {
|
|
var testCases = []struct {
|
|
name string
|
|
initFn func() (interface{}, error)
|
|
expectedErr error
|
|
}{
|
|
{
|
|
"non-null manager, no running node, AddPeer()",
|
|
func() (interface{}, error) {
|
|
return nil, s.StatusNode.AddPeer("enode://da3bf389a031f33fb55c9f5f54fde8473912402d27fffaa50efd74c0d0515f3a61daf6d52151f2876b19c15828e6f670352bff432b5ec457652e74755e8c864f@51.15.62.116:30303")
|
|
},
|
|
node.ErrNoRunningNode,
|
|
},
|
|
{
|
|
"non-null manager, no running node, get NodeConfig",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.Config(), nil
|
|
},
|
|
nil,
|
|
},
|
|
{
|
|
"non-null manager, no running node, get Node",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.GethNode(), nil
|
|
},
|
|
nil,
|
|
},
|
|
{
|
|
"non-null manager, no running node, get LES",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.LightEthereumService()
|
|
},
|
|
node.ErrNoRunningNode,
|
|
},
|
|
{
|
|
"non-null manager, no running node, get Whisper",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.WhisperService()
|
|
},
|
|
node.ErrNoRunningNode,
|
|
},
|
|
{
|
|
"non-null manager, no running node, get RPC Client",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.RPCClient(), nil
|
|
},
|
|
nil,
|
|
},
|
|
}
|
|
for _, tc := range testCases {
|
|
s.T().Log(tc.name)
|
|
obj, err := tc.initFn()
|
|
s.Nil(obj)
|
|
s.Equal(tc.expectedErr, err)
|
|
}
|
|
}
|
|
|
|
func (s *ManagerTestSuite) TestReferencesWithStartedNode() {
|
|
s.StartTestNode()
|
|
defer func() {
|
|
time.Sleep(100 * time.Millisecond)
|
|
s.StopTestNode()
|
|
}()
|
|
|
|
var testCases = []struct {
|
|
name string
|
|
initFn func() (interface{}, error)
|
|
expectedType interface{}
|
|
}{
|
|
{
|
|
"node is running, get NodeConfig",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.Config(), nil
|
|
},
|
|
¶ms.NodeConfig{},
|
|
},
|
|
{
|
|
"node is running, get Node",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.GethNode(), nil
|
|
},
|
|
&gethnode.Node{},
|
|
},
|
|
{
|
|
"node is running, get LES",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.LightEthereumService()
|
|
},
|
|
&les.LightEthereum{},
|
|
},
|
|
{
|
|
"node is running, get Whisper",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.WhisperService()
|
|
},
|
|
&whisper.Whisper{},
|
|
},
|
|
{
|
|
"node is running, get AccountManager",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.AccountManager()
|
|
},
|
|
&accounts.Manager{},
|
|
},
|
|
{
|
|
"node is running, get RPC Client",
|
|
func() (interface{}, error) {
|
|
return s.StatusNode.RPCClient(), nil
|
|
},
|
|
&rpc.Client{},
|
|
},
|
|
}
|
|
for _, tc := range testCases {
|
|
s.T().Log(tc.name)
|
|
obj, err := tc.initFn()
|
|
s.NoError(err)
|
|
s.NotNil(obj)
|
|
s.IsType(tc.expectedType, obj)
|
|
}
|
|
}
|
|
|
|
func (s *ManagerTestSuite) TestNodeStartStop() {
|
|
nodeConfig, err := MakeTestNodeConfig(GetNetworkID())
|
|
s.NoError(err)
|
|
|
|
// try stopping non-started node
|
|
s.False(s.StatusNode.IsRunning())
|
|
time.Sleep(100 * time.Millisecond) //https://github.com/status-im/status-go/issues/429#issuecomment-339663163
|
|
s.Equal(node.ErrNoRunningNode, s.StatusNode.Stop())
|
|
|
|
// start node
|
|
s.False(s.StatusNode.IsRunning())
|
|
s.NoError(s.StatusNode.Start(nodeConfig, nil))
|
|
// wait till node is started
|
|
s.True(s.StatusNode.IsRunning())
|
|
|
|
// try starting another node (w/o stopping the previously started node)
|
|
s.Equal(node.ErrNodeRunning, s.StatusNode.Start(nodeConfig, nil))
|
|
|
|
// now stop node
|
|
time.Sleep(100 * time.Millisecond) //https://github.com/status-im/status-go/issues/429#issuecomment-339663163
|
|
s.NoError(s.StatusNode.Stop())
|
|
s.False(s.StatusNode.IsRunning())
|
|
|
|
// start new node with exactly the same config
|
|
s.NoError(s.StatusNode.Start(nodeConfig, nil))
|
|
s.True(s.StatusNode.IsRunning())
|
|
|
|
// finally stop the node
|
|
time.Sleep(100 * time.Millisecond) //https://github.com/status-im/status-go/issues/429#issuecomment-339663163
|
|
s.NoError(s.StatusNode.Stop())
|
|
}
|
|
|
|
func (s *ManagerTestSuite) TestNetworkSwitching() {
|
|
// get Ropsten config
|
|
nodeConfig, err := MakeTestNodeConfig(GetNetworkID())
|
|
s.NoError(err)
|
|
s.False(s.StatusNode.IsRunning())
|
|
s.NoError(s.StatusNode.Start(nodeConfig, nil))
|
|
// wait till node is started
|
|
s.Require().True(s.StatusNode.IsRunning())
|
|
|
|
firstHash, err := e2e.FirstBlockHash(s.StatusNode)
|
|
s.NoError(err)
|
|
s.Equal(GetHeadHash(), firstHash)
|
|
|
|
// now stop node, and make sure that a new node, on different network can be started
|
|
time.Sleep(100 * time.Millisecond) //https://github.com/status-im/status-go/issues/429#issuecomment-339663163
|
|
s.NoError(s.StatusNode.Stop())
|
|
s.False(s.StatusNode.IsRunning())
|
|
|
|
// start new node with completely different config
|
|
nodeConfig, err = MakeTestNodeConfig(params.RinkebyNetworkID)
|
|
s.NoError(err)
|
|
s.NoError(s.StatusNode.Start(nodeConfig, nil))
|
|
s.True(s.StatusNode.IsRunning())
|
|
|
|
// make sure we are on another network indeed
|
|
firstHash, err = e2e.FirstBlockHash(s.StatusNode)
|
|
s.NoError(err)
|
|
s.Equal(GetHeadHashFromNetworkID(params.RinkebyNetworkID), firstHash)
|
|
|
|
time.Sleep(100 * time.Millisecond) //https://github.com/status-im/status-go/issues/429#issuecomment-339663163
|
|
s.NoError(s.StatusNode.Stop())
|
|
}
|
|
|
|
func (s *ManagerTestSuite) TestStartWithUpstreamEnabled() {
|
|
if GetNetworkID() == params.StatusChainNetworkID {
|
|
s.T().Skip()
|
|
}
|
|
|
|
nodeConfig, err := MakeTestNodeConfig(GetNetworkID())
|
|
s.NoError(err)
|
|
|
|
networkURL, err := GetRemoteURL()
|
|
s.NoError(err)
|
|
|
|
nodeConfig.UpstreamConfig.Enabled = true
|
|
nodeConfig.UpstreamConfig.URL = networkURL
|
|
|
|
s.NoError(s.StatusNode.Start(nodeConfig, nil))
|
|
s.True(s.StatusNode.IsRunning())
|
|
|
|
time.Sleep(100 * time.Millisecond) //https://github.com/status-im/status-go/issues/429#issuecomment-339663163
|
|
s.NoError(s.StatusNode.Stop())
|
|
}
|
|
|
|
// TODO(adam): race conditions should be tested with -race flag and unit tests, if possible.
|
|
// TODO(boris): going via https://github.com/status-im/status-go/pull/433#issuecomment-342232645 . Testing should be with -race flag
|
|
// Research if it's possible to do the same with unit tests.
|
|
//func (s *ManagerTestSuite) TestRaceConditions() {
|
|
// cnt := 25
|
|
// progress := make(chan struct{}, cnt)
|
|
// rnd := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
//
|
|
// nodeConfig1, e := MakeTestNodeConfig(GetNetworkID())
|
|
// s.NoError(e)
|
|
//
|
|
// nodeConfig2, e := MakeTestNodeConfig(GetNetworkID())
|
|
// s.NoError(e)
|
|
//
|
|
// nodeConfigs := []*params.NodeConfig{nodeConfig1, nodeConfig2}
|
|
//
|
|
// var funcsToTest = []func(*params.NodeConfig){
|
|
// func(config *params.NodeConfig) {
|
|
// log.Info("Start()")
|
|
// _, err := s.StatusNode.Start(config)
|
|
// s.T().Logf("Start() for network: %d, error: %v", config.NetworkID, err)
|
|
// progress <- struct{}{}
|
|
// },
|
|
// func(config *params.NodeConfig) {
|
|
// _, err := s.StatusNode.Stop()
|
|
// s.T().Logf("Stop() for network: %d, error: %v", config.NetworkID, err)
|
|
// progress <- struct{}{}
|
|
// },
|
|
// func(config *params.NodeConfig) {
|
|
// log.Info("Node()")
|
|
// _, err := s.StatusNode.GethNode()
|
|
// s.T().Logf("Node(), error: %v", err)
|
|
// progress <- struct{}{}
|
|
// },
|
|
// func(config *params.NodeConfig) {
|
|
// log.Info("IsRunning()")
|
|
// s.T().Logf("IsRunning(), result: %v", s.StatusNode.IsRunning())
|
|
// progress <- struct{}{}
|
|
// },
|
|
// // TODO(adam): quarantined until it uses a different datadir
|
|
// // as otherwise it wipes out cached blockchain data.
|
|
// // func(config *params.NodeConfig) {
|
|
// // log.Info("ResetChainData()")
|
|
// // _, err := s.StatusNode.ResetChainData()
|
|
// // s.T().Logf("ResetChainData(), error: %v", err)
|
|
// // progress <- struct{}{}
|
|
// // },
|
|
// func(config *params.NodeConfig) {
|
|
// log.Info("RestartNode()")
|
|
// _, err := s.StatusNode.RestartNode()
|
|
// s.T().Logf("RestartNode(), error: %v", err)
|
|
// progress <- struct{}{}
|
|
// },
|
|
// func(config *params.NodeConfig) {
|
|
// log.Info("Config()")
|
|
// _, err := s.StatusNode.Config()
|
|
// s.T().Logf("Config(), error: %v", err)
|
|
// progress <- struct{}{}
|
|
// },
|
|
// func(config *params.NodeConfig) {
|
|
// log.Info("LightEthereumService()")
|
|
// _, err := s.StatusNode.LightEthereumService()
|
|
// s.T().Logf("LightEthereumService(), error: %v", err)
|
|
// progress <- struct{}{}
|
|
// },
|
|
// func(config *params.NodeConfig) {
|
|
// log.Info("WhisperService()")
|
|
// _, err := s.StatusNode.WhisperService()
|
|
// s.T().Logf("WhisperService(), error: %v", err)
|
|
// progress <- struct{}{}
|
|
// },
|
|
// func(config *params.NodeConfig) {
|
|
// log.Info("AccountManager()")
|
|
// _, err := s.StatusNode.AccountManager()
|
|
// s.T().Logf("AccountManager(), error: %v", err)
|
|
// progress <- struct{}{}
|
|
// },
|
|
// func(config *params.NodeConfig) {
|
|
// log.Info("AccountKeyStore()")
|
|
// _, err := s.StatusNode.AccountKeyStore()
|
|
// s.T().Logf("AccountKeyStore(), error: %v", err)
|
|
// progress <- struct{}{}
|
|
// },
|
|
// func(config *params.NodeConfig) {
|
|
// log.Info("RPCClient()")
|
|
// s.StatusNode.RPCClient()
|
|
// progress <- struct{}{}
|
|
// },
|
|
// }
|
|
//
|
|
// // increase Start()/Stop() population
|
|
// for i := 0; i < 5; i++ {
|
|
// funcsToTest = append(funcsToTest, funcsToTest[0], funcsToTest[1])
|
|
// }
|
|
//
|
|
// for i := 0; i < cnt; i++ {
|
|
// randConfig := nodeConfigs[rnd.Intn(len(nodeConfigs))]
|
|
// randFunc := funcsToTest[rnd.Intn(len(funcsToTest))]
|
|
//
|
|
// if rnd.Intn(100) > 75 { // introduce random delays
|
|
// time.Sleep(500 * time.Millisecond)
|
|
// }
|
|
// go randFunc(randConfig)
|
|
// }
|
|
//
|
|
// for range progress {
|
|
// cnt -= 1
|
|
// if cnt <= 0 {
|
|
// break
|
|
// }
|
|
// }
|
|
//
|
|
// time.Sleep(2 * time.Second) // so that we see some logs
|
|
// nodeStopped, _ := s.StatusNode.Stop() // just in case we have a node running
|
|
//
|
|
// if nodeStopped != nil {
|
|
// <-nodeStopped
|
|
// }
|
|
//}
|