proto fixes

nitronit · Sep 7, 2023 · e2cca71 · e2cca71
1 parent 41d514d
commit e2cca71
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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) {