github.com/hechain20/hechain@v0.0.0-20220316014945-b544036ba106/internal/peer/snapshot/client_test.go (about)

     1  /*
     2  Copyright hechain. All Rights Reserved.
     3  
     4  SPDX-License-Identifier: Apache-2.0
     5  */
     6  
     7  package snapshot
     8  
     9  import (
    10  	"testing"
    11  
    12  	"github.com/hechain20/hechain/internal/peer/common"
    13  	pb "github.com/hyperledger/fabric-protos-go/peer"
    14  	"github.com/spf13/viper"
    15  	"github.com/stretchr/testify/require"
    16  )
    17  
    18  //go:generate counterfeiter -o mock/snapshot_client.go -fake-name SnapshotClient . snapshotClient
    19  
    20  type snapshotClient interface {
    21  	pb.SnapshotClient
    22  }
    23  
    24  //go:generate counterfeiter -o mock/signer.go -fake-name Signer . signer
    25  
    26  type signer interface {
    27  	common.Signer
    28  }
    29  
    30  func TestValidatePeerConnectionParameters(t *testing.T) {
    31  	viper.Set("peer.tls.enabled", false)
    32  	require.NoError(t, validatePeerConnectionParameters())
    33  
    34  	viper.Set("peer.tls.enabled", true)
    35  	expectedErrMsg := "the required parameter 'tlsRootCertFile' is empty. Rerun the command with --tlsRootCertFile flag"
    36  	require.EqualError(t, validatePeerConnectionParameters(), expectedErrMsg)
    37  
    38  	tlsRootCertFile = "cert1_file"
    39  	require.NoError(t, validatePeerConnectionParameters())
    40  
    41  	// test error propagation
    42  	args := []string{"-c", "mychannel"}
    43  	resetFlags()
    44  	cmd := listPendingCmd(nil, nil)
    45  	cmd.SetArgs(args)
    46  	err := cmd.Execute()
    47  	require.EqualError(t, err, expectedErrMsg)
    48  
    49  	resetFlags()
    50  	cmd = submitRequestCmd(nil, nil)
    51  	cmd.SetArgs(args)
    52  	err = cmd.Execute()
    53  	require.EqualError(t, err, expectedErrMsg)
    54  
    55  	resetFlags()
    56  	cmd = cancelRequestCmd(nil, nil)
    57  	// append required block number parameter for cancel request
    58  	args = append(args, []string{"-b", "10"}...)
    59  	cmd.SetArgs(args)
    60  	err = cmd.Execute()
    61  	require.EqualError(t, err, expectedErrMsg)
    62  }