testchain: implement Sign function

Sign any data by all consensus nodes.

pkg/core/helper_test.go

@@ -71,17 +71,7 @@ func newBlock(cfg config.ProtocolConfiguration, index uint32, prev util.Uint256,
 	}
 	_ = b.RebuildMerkleRoot()
 
-	buf := io.NewBufBinWriter()
+	b.Script.InvocationScript = testchain.Sign(b.GetSignedPart())
-	for i := 0; i < testchain.Size(); i++ {
-		pKey := testchain.PrivateKey(i)
-		b := b.GetSignedPart()
-		sig := pKey.Sign(b)
-		if len(sig) != 64 {
-			panic("wrong signature length")
-		}
-		emit.Bytes(buf.BinWriter, sig)
-	}
-	b.Script.InvocationScript = buf.Bytes()
 	return b
 }
 
@@ -508,15 +498,8 @@ func signTx(bc *Blockchain, txs ...*transaction.Transaction) error {
 	}
 	for _, tx := range txs {
 		data := tx.GetSignedPart()
-
-		var invoc []byte
-		for i := 0; i < testchain.Size(); i++ {
-			priv := testchain.PrivateKey(i)
-			invoc = append(invoc, getInvocationScript(data, priv)...)
-		}
-
 		tx.Scripts = []transaction.Witness{{
-			InvocationScript:   invoc,
+			InvocationScript:   testchain.Sign(data),
 			VerificationScript: rawScript,
 		}}
 	}

pkg/internal/testchain/address.go

@@ -4,8 +4,10 @@ import (
 	"github.com/nspcc-dev/neo-go/pkg/crypto/hash"
 	"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
 	"github.com/nspcc-dev/neo-go/pkg/encoding/address"
+	"github.com/nspcc-dev/neo-go/pkg/io"
 	"github.com/nspcc-dev/neo-go/pkg/smartcontract"
 	"github.com/nspcc-dev/neo-go/pkg/util"
+	"github.com/nspcc-dev/neo-go/pkg/vm/emit"
 )
 
 // privNetKeys is a list of unencrypted WIFs sorted by public key.
@@ -78,3 +80,17 @@ func MultisigScriptHash() util.Uint160 {
 func MultisigAddress() string {
 	return address.Uint160ToString(MultisigScriptHash())
 }
+
+// Sign signs data by all consensus nodes and returns invocation script.
+func Sign(data []byte) []byte {
+	buf := io.NewBufBinWriter()
+	for i := 0; i < Size(); i++ {
+		pKey := PrivateKey(i)
+		sig := pKey.Sign(data)
+		if len(sig) != 64 {
+			panic("wrong signature length")
+		}
+		emit.Bytes(buf.BinWriter, sig)
+	}
+	return buf.Bytes()
+}