From a8d41f596d232e6180b520dd526b0db57595e478 Mon Sep 17 00:00:00 2001
From: Pavel Karpy <carpawell@nspcc.ru>
Date: Fri, 10 Sep 2021 20:29:32 +0300
Subject: [PATCH] [#822] event/notaryPreparator: Add unit test for correct NR

Signed-off-by: Pavel Karpy <carpawell@nspcc.ru>
---
 pkg/morph/event/notary_preparator_test.go | 157 ++++++++++++++++++++++
 1 file changed, 157 insertions(+)
 create mode 100644 pkg/morph/event/notary_preparator_test.go

diff --git a/pkg/morph/event/notary_preparator_test.go b/pkg/morph/event/notary_preparator_test.go
new file mode 100644
index 000000000..c18d5c727
--- /dev/null
+++ b/pkg/morph/event/notary_preparator_test.go
@@ -0,0 +1,157 @@
+package event
+
+import (
+	"encoding/binary"
+	"testing"
+
+	"github.com/nspcc-dev/neo-go/pkg/core/interop/interopnames"
+	"github.com/nspcc-dev/neo-go/pkg/core/transaction"
+	"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/io"
+	"github.com/nspcc-dev/neo-go/pkg/network/payload"
+	"github.com/nspcc-dev/neo-go/pkg/smartcontract"
+	"github.com/nspcc-dev/neo-go/pkg/smartcontract/callflag"
+	"github.com/nspcc-dev/neo-go/pkg/util"
+	"github.com/nspcc-dev/neo-go/pkg/vm/emit"
+	"github.com/nspcc-dev/neo-go/pkg/vm/opcode"
+	"github.com/nspcc-dev/neofs-node/pkg/morph/client"
+	"github.com/stretchr/testify/require"
+)
+
+var (
+	alphaKeys keys.PublicKeys
+
+	dummyInvocationScript = append([]byte{byte(opcode.PUSHDATA1), 64}, make([]byte, 64)...)
+	contractSysCall       = make([]byte, 4)
+
+	scriptHash util.Uint160
+)
+
+func init() {
+	privat, _ := keys.NewPrivateKey()
+	pub := privat.PublicKey()
+
+	alphaKeys = keys.PublicKeys{pub}
+
+	binary.LittleEndian.PutUint32(contractSysCall, interopnames.ToID([]byte(interopnames.SystemContractCall)))
+
+	scriptHash, _ = util.Uint160DecodeStringLE("21fce15191428e9c2f0e8d0329ff6d3dd14882de")
+}
+
+type blockCounter struct {
+	epoch uint32
+	err   error
+}
+
+func (b blockCounter) BlockCount() (res uint32, err error) {
+	return b.epoch, b.err
+}
+
+func TestPrepare_CorrectNR(t *testing.T) {
+	tests := []struct {
+		hash   util.Uint160
+		method string
+		args   []interface{}
+	}{
+		{
+			scriptHash,
+			"test1",
+			nil,
+		},
+		{
+			scriptHash,
+			"test2",
+			[]interface{}{
+				int64(4),
+				"test",
+				[]interface{}{
+					int64(4),
+					false,
+					true,
+				},
+			},
+		},
+	}
+
+	preparator := notaryPreparator(
+		PreparatorPrm{
+			alphaKeysSource(),
+			blockCounter{100, nil},
+		},
+	)
+
+	for _, test := range tests {
+		nr := correctNR(script(test.hash, test.method, test.args...))
+
+		event, err := preparator.Prepare(nr)
+
+		require.NoError(t, err)
+		require.Equal(t, test.method, event.Type().String())
+		require.Equal(t, test.hash.StringLE(), event.ScriptHash().StringLE())
+	}
+}
+
+func alphaKeysSource() client.AlphabetKeys {
+	return func() (keys.PublicKeys, error) {
+		return alphaKeys, nil
+	}
+}
+
+func script(hash util.Uint160, method string, args ...interface{}) []byte {
+	bw := io.NewBufBinWriter()
+
+	if len(args) > 0 {
+		emit.AppCall(bw.BinWriter, hash, method, callflag.All, args)
+	} else {
+		emit.AppCallNoArgs(bw.BinWriter, hash, method, callflag.All)
+	}
+
+	return bw.Bytes()
+}
+
+func correctNR(script []byte) *payload.P2PNotaryRequest {
+	alphaVerificationScript, _ := smartcontract.CreateMultiSigRedeemScript(len(alphaKeys)*2/3+1, alphaKeys)
+
+	return &payload.P2PNotaryRequest{
+		MainTransaction: &transaction.Transaction{
+			Signers: []transaction.Signer{
+				{},
+				{
+					Account: hash.Hash160(alphaVerificationScript),
+				},
+				{},
+			},
+			Scripts: []transaction.Witness{
+				{},
+				{
+					InvocationScript:   dummyInvocationScript,
+					VerificationScript: alphaVerificationScript,
+				},
+				{
+					InvocationScript: dummyInvocationScript,
+				},
+			},
+			Attributes: []transaction.Attribute{
+				{
+					Value: &transaction.NotaryAssisted{
+						NKeys: uint8(len(alphaKeys)),
+					},
+				},
+			},
+			Script: script,
+		},
+		FallbackTransaction: &transaction.Transaction{
+			Attributes: []transaction.Attribute{
+				{},
+				{
+					Type: transaction.NotValidBeforeT,
+					Value: &transaction.NotValidBefore{
+						Height: 1000,
+					},
+				},
+				{},
+			},
+		},
+	}
+}