From a8d41f596d232e6180b520dd526b0db57595e478 Mon Sep 17 00:00:00 2001 From: Pavel Karpy 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 --- 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, + }, + }, + {}, + }, + }, + } +}