unit-tests fix

Signed-off-by: Angelo De Caro <[email protected]>

token/services/network/common/rws/translator/translator.go

@@ -210,8 +210,6 @@ func (w *Translator) checkIssue(issue IssueAction) error {
 }
 
 func (w *Translator) checkTransfer(t TransferAction) error {
-	inputs := t.GetInputs()
-
 	// check inputs
 
 	// we must check that the serial number does not exist, if any are in the action
@@ -222,10 +220,14 @@ func (w *Translator) checkTransfer(t TransferAction) error {
 	}
 
 	// we must check that the serial number for serialized inputs must exist, if any are in the action
+	inputs := t.GetInputs()
 	serializedInputs, err := t.GetSerializedInputs()
 	if err != nil {
 		return errors.Wrapf(err, "failed to get serialized inputs")
 	}
+	if len(serializedInputs) != len(inputs) {
+		return errors.Errorf("inputs and serialized inputs length mismatch")
+	}
 	for i, input := range inputs {
 		key, err := w.KeyTranslator.CreateOutputSNKey(input.TxId, input.Index, serializedInputs[i])
 		if err != nil {

token/services/network/common/rws/translator/translator_test.go

@@ -186,7 +186,11 @@ var _ = Describe("Translator", func() {
 			fakeRWSet.GetStateReturnsOnCall(0, nil, nil)
 			fakeRWSet.GetStateReturnsOnCall(1, nil, nil)
 			fakeRWSet.GetStateReturnsOnCall(2, nil, nil)
+			fakeRWSet.GetStateReturnsOnCall(3, []byte("s3"), nil)
+			fakeRWSet.GetStateReturnsOnCall(4, []byte("s4"), nil)
+			fakeRWSet.GetStateReturnsOnCall(5, []byte("s5"), nil)
 			faketransfer.GetInputsReturns([]*token.ID{{TxId: "key1"}, {TxId: "key2"}, {TxId: "key3"}})
+			faketransfer.GetSerializedInputsReturns([][]byte{[]byte("i1"), []byte("i2"), []byte("i3")}, nil)
 			faketransfer.GetSerialNumbersReturns(sn)
 			faketransfer.NumOutputsReturns(2)
 			faketransfer.IsGraphHidingReturns(true)