diff --git a/pkg/multiresolver/multi_test.go b/pkg/multiresolver/multi_test.go index 99cafea2..895be347 100644 --- a/pkg/multiresolver/multi_test.go +++ b/pkg/multiresolver/multi_test.go @@ -97,7 +97,7 @@ func TestMultiResolver(t *testing.T) { ctx, cancelFunc := context.WithTimeout(context.Background(), 30*time.Second) defer cancelFunc() - grpcClient := proto.NewICosignerGRPCServerClient(connDNS) + grpcClient := proto.NewICosignerGRPCClient(connDNS) _, err = grpcClient.GetLeader(ctx, &proto.CosignerGRPCGetLeaderRequest{}) require.NoError(t, err) @@ -110,7 +110,7 @@ func TestMultiResolver(t *testing.T) { require.NoError(t, err) defer connIP.Close() - grpcClient = proto.NewICosignerGRPCServerClient(connIP) + grpcClient = proto.NewICosignerGRPCClient(connIP) _, err = grpcClient.GetLeader(ctx, &proto.CosignerGRPCGetLeaderRequest{}) require.NoError(t, err) } diff --git a/pkg/node/raft_events.go b/pkg/node/raft_events.go index ec4b388b..1efc4b9d 100644 --- a/pkg/node/raft_events.go +++ b/pkg/node/raft_events.go @@ -51,7 +51,7 @@ func (f *fsm) handleLSSEvent(value string) { _ = f.thresholdValidator.myCosigner.SaveLastSignedState(lss.ChainID, lss.SignStateConsensus) } -func (s *RaftStore) getLeaderGRPCClient() (proto.ICosignerGRPCServerClient, *grpc.ClientConn, error) { +func (s *RaftStore) getLeaderGRPCClient() (proto.ICosignerGRPCClient, *grpc.ClientConn, error) { var leader string for i := 0; i < 30; i++ { leader = string(s.GetLeader()) @@ -68,7 +68,7 @@ func (s *RaftStore) getLeaderGRPCClient() (proto.ICosignerGRPCServerClient, *grp if err != nil { return nil, nil, err } - return proto.NewICosignerGRPCServerClient(conn), conn, nil + return proto.NewICosignerGRPCClient(conn), conn, nil } func (s *RaftStore) SignBlock(req ValidatorSignBlockRequest) (*ValidatorSignBlockResponse, error) {