prysm-pulse/validator/accounts/v1/status_test.go

package accounts

import (
	"context"
	"testing"

	"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"
)

func TestFetchAccountStatuses_OK(t *testing.T) {
	ctx := context.Background()
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()
	pubkeys := make([][]byte, 10000)
	indices := make([]uint64, 10000)
	for i := 0; i < 10000; i++ {
		pubkeys[i] = []byte{byte(i)}
		indices[i] = uint64(i)
	}

	mockClient := mock.NewMockBeaconNodeValidatorClient(ctrl)
	mockClient.EXPECT().MultipleValidatorStatus(
		gomock.Any(),
		&ethpb.MultipleValidatorStatusRequest{PublicKeys: pubkeys},
	).Return(&ethpb.MultipleValidatorStatusResponse{PublicKeys: pubkeys, Indices: indices}, nil /*err*/)
	_, err := FetchAccountStatuses(ctx, mockClient, pubkeys)
	require.NoError(t, err, "FetchAccountStatuses failed with error")
}