code.vegaprotocol.io/vega@v0.79.0/commands/cancel_transfer_test.go (about)

     1  // Copyright (C) 2023 Gobalsky Labs Limited
     2  //
     3  // This program is free software: you can redistribute it and/or modify
     4  // it under the terms of the GNU Affero General Public License as
     5  // published by the Free Software Foundation, either version 3 of the
     6  // License, or (at your option) any later version.
     7  //
     8  // This program is distributed in the hope that it will be useful,
     9  // but WITHOUT ANY WARRANTY; without even the implied warranty of
    10  // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    11  // GNU Affero General Public License for more details.
    12  //
    13  // You should have received a copy of the GNU Affero General Public License
    14  // along with this program.  If not, see <http://www.gnu.org/licenses/>.
    15  
    16  package commands_test
    17  
    18  import (
    19  	"errors"
    20  	"testing"
    21  
    22  	"code.vegaprotocol.io/vega/commands"
    23  	commandspb "code.vegaprotocol.io/vega/protos/vega/commands/v1"
    24  
    25  	"github.com/stretchr/testify/assert"
    26  )
    27  
    28  func TestNilCancelTransferFails(t *testing.T) {
    29  	err := checkCancelTransfer(nil)
    30  
    31  	assert.Contains(t, err.Get("cancel_transfer"), commands.ErrIsRequired)
    32  }
    33  
    34  func TestCancelTransfer(t *testing.T) {
    35  	cases := []struct {
    36  		ctransfer commandspb.CancelTransfer
    37  		errString string
    38  	}{
    39  		{
    40  			ctransfer: commandspb.CancelTransfer{
    41  				TransferId: "18f8b607aad9ef2cd57f2d233766b0c576b27a3e0c50c9db713c00e518c0bbdc",
    42  			},
    43  		},
    44  		{
    45  			ctransfer: commandspb.CancelTransfer{},
    46  			errString: "cancel_transfer.transfer_id (is required)",
    47  		},
    48  	}
    49  
    50  	for _, c := range cases {
    51  		err := commands.CheckCancelTransfer(&c.ctransfer)
    52  		if len(c.errString) <= 0 {
    53  			assert.NoError(t, err)
    54  			continue
    55  		}
    56  		assert.EqualError(t, err, c.errString)
    57  	}
    58  }
    59  
    60  func checkCancelTransfer(cmd *commandspb.CancelTransfer) commands.Errors {
    61  	err := commands.CheckCancelTransfer(cmd)
    62  
    63  	var e commands.Errors
    64  	if ok := errors.As(err, &e); !ok {
    65  		return commands.NewErrors()
    66  	}
    67  
    68  	return e
    69  }