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

     1  /*
     2  Copyright hechain. All Rights Reserved.
     3  
     4  SPDX-License-Identifier: Apache-2.0
     5  */
     6  
     7  package node
     8  
     9  import (
    10  	"testing"
    11  
    12  	"github.com/stretchr/testify/require"
    13  )
    14  
    15  func TestRollbackCmd(t *testing.T) {
    16  	t.Run("when the channelID is not supplied", func(t *testing.T) {
    17  		cmd := rollbackCmd()
    18  		args := []string{}
    19  		cmd.SetArgs(args)
    20  		err := cmd.Execute()
    21  		require.Equal(t, "Must supply channel ID", err.Error())
    22  	})
    23  
    24  	t.Run("when the specified channelID does not exist", func(t *testing.T) {
    25  		cmd := rollbackCmd()
    26  		args := []string{"-c", "ch1", "-b", "10"}
    27  		cmd.SetArgs(args)
    28  		err := cmd.Execute()
    29  		// this should return an error as no ledger has been set up
    30  		require.Contains(t, err.Error(), "error while checking if any ledger has been bootstrapped from snapshot")
    31  	})
    32  }