mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 04:47:18 +00:00
a840fa563d
* remove accounts-v1 * get all tests to not panic * all client tests passing * fix node test * eliminate old flags * tidy up
84 lines
2.1 KiB
Go
84 lines
2.1 KiB
Go
package client
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/prysmaticlabs/prysm/shared"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
)
|
|
|
|
var _ shared.Service = (*ValidatorService)(nil)
|
|
|
|
func TestMain(m *testing.M) {
|
|
dir := testutil.TempDir() + "/keystore1"
|
|
cleanup := func() {
|
|
if err := os.RemoveAll(dir); err != nil {
|
|
log.WithError(err).Debug("Cannot remove keystore folder")
|
|
}
|
|
}
|
|
defer cleanup()
|
|
code := m.Run()
|
|
// os.Exit will prevent defer from being called
|
|
cleanup()
|
|
os.Exit(code)
|
|
}
|
|
|
|
func TestStop_CancelsContext(t *testing.T) {
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|
vs := &ValidatorService{
|
|
ctx: ctx,
|
|
cancel: cancel,
|
|
}
|
|
|
|
assert.NoError(t, vs.Stop())
|
|
|
|
select {
|
|
case <-time.After(1 * time.Second):
|
|
t.Error("Context not canceled 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",
|
|
}
|
|
validatorService.Start()
|
|
require.NoError(t, validatorService.Stop(), "Could not stop service")
|
|
require.LogsContain(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",
|
|
}
|
|
validatorService.Start()
|
|
require.LogsContain(t, hook, "You are using an insecure gRPC connection")
|
|
require.NoError(t, validatorService.Stop(), "Could not stop service")
|
|
require.LogsContain(t, hook, "Stopping service")
|
|
}
|
|
|
|
func TestStatus_NoConnectionError(t *testing.T) {
|
|
validatorService := &ValidatorService{}
|
|
assert.ErrorContains(t, "no connection", validatorService.Status())
|
|
}
|