mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-31 23:41:22 +00:00
b06876d698
* add node server * stub implementations * node server impl * gaz * only missing genesis info now * fmt imports * all tests pass * fmt * revert change * punctuation * use internal err code * view permission * using real reflection * spacing * lint
113 lines
2.9 KiB
Go
113 lines
2.9 KiB
Go
package rpc
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
"github.com/gogo/protobuf/proto"
|
|
ptypes "github.com/gogo/protobuf/types"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/internal"
|
|
pb "github.com/prysmaticlabs/prysm/proto/beacon/p2p/v1"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/shared/version"
|
|
"google.golang.org/grpc"
|
|
"google.golang.org/grpc/reflection"
|
|
)
|
|
|
|
type mockSyncChecker struct {
|
|
syncing bool
|
|
}
|
|
|
|
func (m *mockSyncChecker) Syncing() bool {
|
|
return m.syncing
|
|
}
|
|
|
|
func TestNodeServer_GetSyncStatus(t *testing.T) {
|
|
mSync := &mockSyncChecker{false}
|
|
ns := &NodeServer{
|
|
syncChecker: mSync,
|
|
}
|
|
res, err := ns.GetSyncStatus(context.Background(), &ptypes.Empty{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res.Syncing != mSync.syncing {
|
|
t.Errorf("Wanted GetSyncStatus() = %v, received %v", mSync.syncing, res.Syncing)
|
|
}
|
|
mSync.syncing = true
|
|
res, err = ns.GetSyncStatus(context.Background(), &ptypes.Empty{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res.Syncing != mSync.syncing {
|
|
t.Errorf("Wanted GetSyncStatus() = %v, received %v", mSync.syncing, res.Syncing)
|
|
}
|
|
}
|
|
|
|
func TestNodeServer_GetGenesis(t *testing.T) {
|
|
beaconDB := internal.SetupDB(t)
|
|
defer internal.TeardownDB(t, beaconDB)
|
|
ctx := context.Background()
|
|
addr := [20]byte{1, 2, 3, 4, 5, 6}
|
|
beaconDB.VerifyContractAddress(ctx, common.Address(addr))
|
|
beaconState := &pb.BeaconState{
|
|
Slot: 0,
|
|
GenesisTime: 0,
|
|
}
|
|
if err := beaconDB.SaveFinalizedState(beaconState); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
ns := &NodeServer{
|
|
beaconDB: beaconDB,
|
|
}
|
|
res, err := ns.GetGenesis(ctx, &ptypes.Empty{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !bytes.Equal(res.DepositContractAddress, addr[:]) {
|
|
t.Errorf("Wanted GetGenesis().DepositContractAddress = %#x, received %#x", addr, res.DepositContractAddress)
|
|
}
|
|
genesisTimestamp := time.Unix(0, 0)
|
|
protoTimestamp, err := ptypes.TimestampProto(genesisTimestamp)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !proto.Equal(res.GenesisTime, protoTimestamp) {
|
|
t.Errorf("Wanted GetGenesis().GenesisTime = %v, received %v", protoTimestamp, res.GenesisTime)
|
|
}
|
|
}
|
|
|
|
func TestNodeServer_GetVersion(t *testing.T) {
|
|
v := version.GetVersion()
|
|
ns := &NodeServer{}
|
|
res, err := ns.GetVersion(context.Background(), &ptypes.Empty{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res.Version != v {
|
|
t.Errorf("Wanted GetVersion() = %s, received %s", v, res.Version)
|
|
}
|
|
}
|
|
|
|
func TestNodeServer_GetImplementedServices(t *testing.T) {
|
|
server := grpc.NewServer()
|
|
ns := &NodeServer{
|
|
server: server,
|
|
}
|
|
ethpb.RegisterNodeServer(server, ns)
|
|
reflection.Register(server)
|
|
|
|
res, err := ns.ListImplementedServices(context.Background(), &ptypes.Empty{})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
// We verify the services include the node service + the registered reflection service.
|
|
if len(res.Services) != 2 {
|
|
t.Errorf("Expected 2 services, received %d: %v", len(res.Services), res.Services)
|
|
}
|
|
}
|