mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-04 08:44:28 +00:00
a170c69653
* upgrading linter from gometalinter to golangci-lint * fixed golangci-lint linting * removed linting before_script command * removed disable-all command * Fixed golang config file * fixed golang config file v2 * removed gosec issue rule * formatting * fixed travis build to run golangci-lint * Add install golangci-lint command * fixing golangci-lint script * removed https:// * Added golangci-lint cmd script * added go get for local lint install * created a before_script * add install before script * Added get script * added go mod download * removed go mod downloads * changed * removed before script * Added before script go get lint * added exit zero to see what went wrong * removed golang run script * removed before script * change lint command * verbose output * removed verbose * change linter enable and disable configuration * Update .golangci.yml Removed gotype as a linter * Update .golangci.yml Added typecheck linter * Update .golangci.yml Added fixed lint version * Update .golangci.yml Added gotype * Update .golangci.yml Added typecheck * removed env:lint * Added env lint * fixing lint upgrade * Changing travis configuration * FIxed spelling errors * disabled typecheck * Enabled typecheck * remove binary * Deleting lib binary * adding more linters * fixed constants * fix spelling * fixed all lint issues * Revert "Changing travis configuration" This reverts commit 334afe9d05e96261b01f275aa3ada20e7f36aac4. * Merge branch 'master' of https://github.com/prysmaticlabs/prysm into update-linter * Changed from Infof to Info * Fixing commits * fixing commits with linter config * added install * Fixing * fix log statement
114 lines
2.9 KiB
Go
114 lines
2.9 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"encoding/hex"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/prysmaticlabs/prysm/shared"
|
|
"github.com/prysmaticlabs/prysm/shared/featureconfig"
|
|
"github.com/prysmaticlabs/prysm/shared/keystore"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil"
|
|
"github.com/prysmaticlabs/prysm/validator/accounts"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
)
|
|
|
|
var _ = shared.Service(&ValidatorService{})
|
|
var validatorKey *keystore.Key
|
|
var keyMap map[string]*keystore.Key
|
|
var keyMapThreeValidators map[string]*keystore.Key
|
|
|
|
func keySetup() {
|
|
keyMap = make(map[string]*keystore.Key)
|
|
keyMapThreeValidators = make(map[string]*keystore.Key)
|
|
|
|
validatorKey, _ = keystore.NewKey(rand.Reader)
|
|
keyMap[hex.EncodeToString(validatorKey.PublicKey.Marshal())] = validatorKey
|
|
|
|
for i := 0; i < 3; i++ {
|
|
vKey, _ := keystore.NewKey(rand.Reader)
|
|
keyMapThreeValidators[hex.EncodeToString(vKey.PublicKey.Marshal())] = vKey
|
|
}
|
|
}
|
|
|
|
func init() {
|
|
featureconfig.InitFeatureConfig(&featureconfig.FeatureFlagConfig{
|
|
CacheTreeHash: false,
|
|
})
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
dir := testutil.TempDir() + "/keystore1"
|
|
defer os.RemoveAll(dir)
|
|
accounts.NewValidatorAccount(dir, "1234")
|
|
keySetup()
|
|
os.Exit(m.Run())
|
|
}
|
|
|
|
func TestStop_CancelsContext(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
vs := &ValidatorService{
|
|
ctx: ctx,
|
|
cancel: cancel,
|
|
}
|
|
|
|
if err := vs.Stop(); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
select {
|
|
case <-time.After(1 * time.Second):
|
|
t.Error("ctx not cancelled within 1s")
|
|
case <-vs.ctx.Done():
|
|
}
|
|
}
|
|
|
|
func TestLifecycle(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
// Use canceled context so that the run function exits immediately..
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
validatorService := &ValidatorService{
|
|
ctx: ctx,
|
|
cancel: cancel,
|
|
endpoint: "merkle tries",
|
|
withCert: "alice.crt",
|
|
keys: keyMap,
|
|
}
|
|
validatorService.Start()
|
|
if err := validatorService.Stop(); err != nil {
|
|
t.Fatalf("Could not stop service: %v", err)
|
|
}
|
|
testutil.AssertLogsContain(t, hook, "Stopping service")
|
|
}
|
|
|
|
func TestLifecycle_Insecure(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
// Use canceled context so that the run function exits immediately.
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
cancel()
|
|
validatorService := &ValidatorService{
|
|
ctx: ctx,
|
|
cancel: cancel,
|
|
endpoint: "merkle tries",
|
|
keys: keyMap,
|
|
}
|
|
validatorService.Start()
|
|
testutil.AssertLogsContain(t, hook, "You are using an insecure gRPC connection")
|
|
if err := validatorService.Stop(); err != nil {
|
|
t.Fatalf("Could not stop service: %v", err)
|
|
}
|
|
testutil.AssertLogsContain(t, hook, "Stopping service")
|
|
}
|
|
|
|
func TestStatus_NoConnectionError(t *testing.T) {
|
|
validatorService := &ValidatorService{}
|
|
if err := validatorService.Status(); !strings.Contains(err.Error(), "no connection") {
|
|
t.Errorf("Expected status check to fail if no connection is found, received: %v", err)
|
|
}
|
|
}
|