mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-08 10:41:19 +00:00
74101612ce
* first working implementation * assertions tests * adds to requires * merges assert and require tests into a single suite * gazelle * Merge branch 'merge-assert-require-tests' into assert-logs-contains-move-to-assertions * gazelle * updates references * fixes build issue * Merge branch 'master' into assert-logs-contains-move-to-assertions * Merge refs/heads/master into assert-logs-contains-move-to-assertions * Merge branch 'master' into assert-logs-contains-move-to-assertions * fixes build issue * Merge branch 'assert-logs-contains-move-to-assertions' of github.com:prysmaticlabs/prysm into assert-logs-contains-move-to-assertions * Merge refs/heads/master into assert-logs-contains-move-to-assertions * Merge refs/heads/master into assert-logs-contains-move-to-assertions
89 lines
2.4 KiB
Go
89 lines
2.4 KiB
Go
package beaconclient
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/golang/mock/gomock"
|
|
ethpb "github.com/prysmaticlabs/ethereumapis/eth/v1alpha1"
|
|
"github.com/prysmaticlabs/prysm/shared/mock"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
logTest "github.com/sirupsen/logrus/hooks/test"
|
|
)
|
|
|
|
func TestService_ChainHead(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
client := mock.NewMockBeaconChainClient(ctrl)
|
|
|
|
bs := Service{
|
|
beaconClient: client,
|
|
}
|
|
wanted := ðpb.ChainHead{
|
|
HeadSlot: 4,
|
|
HeadEpoch: 0,
|
|
HeadBlockRoot: make([]byte, 32),
|
|
}
|
|
client.EXPECT().GetChainHead(gomock.Any(), gomock.Any()).Return(wanted, nil)
|
|
res, err := bs.ChainHead(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !proto.Equal(res, wanted) {
|
|
t.Errorf("Wanted %v, received %v", wanted, res)
|
|
}
|
|
}
|
|
|
|
func TestService_GenesisValidatorsRoot(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
|
|
client := mock.NewMockNodeClient(ctrl)
|
|
bs := Service{
|
|
nodeClient: client,
|
|
}
|
|
wanted := ðpb.Genesis{
|
|
GenesisValidatorsRoot: []byte("I am genesis"),
|
|
}
|
|
client.EXPECT().GetGenesis(gomock.Any(), gomock.Any()).Return(wanted, nil)
|
|
res, err := bs.GenesisValidatorsRoot(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !bytes.Equal(res, wanted.GenesisValidatorsRoot) {
|
|
t.Errorf("Wanted %#x, received %#x", wanted.GenesisValidatorsRoot, res)
|
|
}
|
|
// test next fetch uses memory and not the rpc call.
|
|
res, err = bs.GenesisValidatorsRoot(context.Background())
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !bytes.Equal(res, wanted.GenesisValidatorsRoot) {
|
|
t.Errorf("Wanted %#x, received %#x", wanted.GenesisValidatorsRoot, res)
|
|
}
|
|
}
|
|
|
|
func TestService_QuerySyncStatus(t *testing.T) {
|
|
hook := logTest.NewGlobal()
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
client := mock.NewMockNodeClient(ctrl)
|
|
|
|
bs := Service{
|
|
nodeClient: client,
|
|
}
|
|
syncStatusPollingInterval = time.Millisecond
|
|
client.EXPECT().GetSyncStatus(gomock.Any(), gomock.Any()).Return(ðpb.SyncStatus{
|
|
Syncing: true,
|
|
}, nil)
|
|
client.EXPECT().GetSyncStatus(gomock.Any(), gomock.Any()).Return(ðpb.SyncStatus{
|
|
Syncing: false,
|
|
}, nil)
|
|
bs.querySyncStatus(context.Background())
|
|
require.LogsContain(t, hook, "Waiting for beacon node to be fully synced...")
|
|
require.LogsContain(t, hook, "Beacon node is fully synced")
|
|
}
|