v0.1.4

Dharitri-org · Mar 21, 2024 · c938087 · c938087
1 parent 7ce80af
commit c938087
Show file tree

Hide file tree

Showing 6 changed files with 208 additions and 139 deletions.
diff --git a/builtInFunctions/setGuardian.go b/builtInFunctions/setGuardian.go
@@ -49,6 +49,9 @@ func (sg *setGuardian) ProcessBuiltinFunction(
 	if vmInput == nil {
 		return nil, ErrNilVmInput
 	}
+	if len(vmInput.Arguments) != noOfArgsSetGuardian {
+		return nil, fmt.Errorf("%w, expected %d, got %d ", ErrInvalidNumberOfArguments, noOfArgsSetGuardian, len(vmInput.Arguments))
+	}
 
 	senderAddr := acntSnd.AddressBytes()
 	senderIsNotCaller := !bytes.Equal(senderAddr, vmInput.CallerAddr)

diff --git a/builtInFunctions/setGuardian_test.go b/builtInFunctions/setGuardian_test.go
@@ -145,6 +145,28 @@ func TestSetGuardian_ProcessBuiltinFunctionCheckArguments(t *testing.T) {
 			receiverAccount: account,
 			expectedErr:     ErrNilVmInput,
 		},
+		{
+			testname: "nil vm input arguments should error",
+			vmInput: func() *vmcommon.ContractCallInput {
+				input := *vmInput
+				input.Arguments = nil
+				return &input
+			},
+			senderAccount:   account,
+			receiverAccount: account,
+			expectedErr:     ErrInvalidNumberOfArguments,
+		},
+		{
+			testname: "one input arguments should error",
+			vmInput: func() *vmcommon.ContractCallInput {
+				input := *vmInput
+				input.Arguments = [][]byte{guardianAddress}
+				return &input
+			},
+			senderAccount:   account,
+			receiverAccount: account,
+			expectedErr:     ErrInvalidNumberOfArguments,
+		},
 		{
 			testname: "sender different to caller should error",
 			vmInput: func() *vmcommon.ContractCallInput {