133 lines
3.3 KiB
Go
133 lines
3.3 KiB
Go
package api_test
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"math/rand"
|
|
"os"
|
|
"strconv"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/log"
|
|
"github.com/status-im/status-go/geth/api"
|
|
"github.com/status-im/status-go/geth/params"
|
|
. "github.com/status-im/status-go/geth/testing"
|
|
"github.com/stretchr/testify/suite"
|
|
)
|
|
|
|
func TestAPI(t *testing.T) {
|
|
suite.Run(t, new(APITestSuite))
|
|
}
|
|
|
|
type APITestSuite struct {
|
|
suite.Suite
|
|
api *api.StatusAPI
|
|
}
|
|
|
|
func (s *APITestSuite) SetupTest() {
|
|
require := s.Require()
|
|
statusAPI := api.NewStatusAPI()
|
|
require.NotNil(statusAPI)
|
|
require.IsType(&api.StatusAPI{}, statusAPI)
|
|
s.api = statusAPI
|
|
}
|
|
|
|
func (s *APITestSuite) TestCHTUpdate() {
|
|
require := s.Require()
|
|
require.NotNil(s.api)
|
|
|
|
tmpDir, err := ioutil.TempDir(os.TempDir(), "cht-updates")
|
|
require.NoError(err)
|
|
defer os.RemoveAll(tmpDir)
|
|
|
|
url := "https://gist.githubusercontent.com/tiabc/83ed515fbb0c0e9d39700a6279072b6a/raw/a8c7b08488fab3c1d9139b18af33da3df823e3ff/cht.json?u=" + strconv.Itoa(int(time.Now().Unix()))
|
|
configJSON := `{
|
|
"NetworkId": ` + strconv.Itoa(params.RopstenNetworkID) + `,
|
|
"DataDir": "` + tmpDir + `",
|
|
"LogEnabled": true,
|
|
"LogLevel": "INFO",
|
|
"RPCEnabled": true,
|
|
"LightEthConfig": {
|
|
"CHTRootConfigURL": "` + url + `"
|
|
}
|
|
}`
|
|
//nodeConfig, err := params.LoadNodeConfig(configJSON)
|
|
_, err = params.LoadNodeConfig(configJSON)
|
|
require.NoError(err)
|
|
|
|
// start node
|
|
//nodeConfig.DevMode = true
|
|
//s.api.StartNode(nodeConfig)
|
|
//s.api.StopNode()
|
|
// TODO(tiabc): Test that CHT is really updated.
|
|
}
|
|
|
|
func (s *APITestSuite) TestRaceConditions() {
|
|
require := s.Require()
|
|
require.NotNil(s.api)
|
|
|
|
cnt := 25
|
|
progress := make(chan struct{}, cnt)
|
|
rnd := rand.New(rand.NewSource(time.Now().UnixNano()))
|
|
|
|
nodeConfig1, err := MakeTestNodeConfig(params.RopstenNetworkID)
|
|
require.NoError(err)
|
|
|
|
nodeConfig2, err := MakeTestNodeConfig(params.RinkebyNetworkID)
|
|
require.NoError(err)
|
|
|
|
nodeConfigs := []*params.NodeConfig{nodeConfig1, nodeConfig2}
|
|
|
|
var funcsToTest = []func(*params.NodeConfig){
|
|
func(config *params.NodeConfig) {
|
|
log.Info("StartNodeAsync()")
|
|
_, err := s.api.StartNodeAsync(config)
|
|
s.T().Logf("StartNodeAsync() for network: %d, error: %v", config.NetworkID, err)
|
|
progress <- struct{}{}
|
|
},
|
|
func(config *params.NodeConfig) {
|
|
log.Info("StopNodeAsync()")
|
|
_, err := s.api.StopNodeAsync()
|
|
s.T().Logf("StopNodeAsync(), error: %v", err)
|
|
progress <- struct{}{}
|
|
},
|
|
func(config *params.NodeConfig) {
|
|
log.Info("RestartNodeAsync()")
|
|
_, err := s.api.RestartNodeAsync()
|
|
s.T().Logf("RestartNodeAsync(), error: %v", err)
|
|
progress <- struct{}{}
|
|
},
|
|
func(config *params.NodeConfig) {
|
|
log.Info("ResetChainDataAsync()")
|
|
_, err := s.api.ResetChainDataAsync()
|
|
s.T().Logf("ResetChainDataAsync(), error: %v", err)
|
|
progress <- struct{}{}
|
|
},
|
|
}
|
|
|
|
// increase StartNode()/StopNode() 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
|
|
s.api.StopNode() // just in case we have a node running
|
|
}
|