[#975] morph/event: Update notary preparator tests

Now includes testcases for notary
requests with 3 and 4 witnesses.

Signed-off-by: Alex Vanin <alexey@nspcc.ru>
This commit is contained in:
Alex Vanin 2021-11-25 19:40:33 +03:00 committed by Alex Vanin
parent a0ff5b1bf8
commit 20f0b29a6e

View file

@ -69,7 +69,7 @@ func TestPrepare_IncorrectScript(t *testing.T) {
emit.Bytes(bw.BinWriter, scriptHash.BytesBE())
emit.Syscall(bw.BinWriter, interopnames.SystemContractCallNative) // any != interopnames.SystemContractCall
nr := correctNR(bw.Bytes())
nr := correctNR(bw.Bytes(), false)
_, err := preparator.Prepare(nr)
@ -84,7 +84,7 @@ func TestPrepare_IncorrectScript(t *testing.T) {
emit.Bytes(bw.BinWriter, scriptHash.BytesBE())
emit.Syscall(bw.BinWriter, interopnames.SystemContractCall)
nr := correctNR(bw.Bytes())
nr := correctNR(bw.Bytes(), false)
_, err := preparator.Prepare(nr)
@ -129,12 +129,14 @@ func TestPrepare_IncorrectNR(t *testing.T) {
tests := []struct {
name string
addW bool // additional witness for non alphabet invocations
mTX mTX
fbTX fbTX
expErr error
}{
{
name: "incorrect witness amount",
addW: false,
mTX: mTX{
scripts: []transaction.Witness{{}},
},
@ -142,6 +144,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "not dummy invocation script",
addW: false,
mTX: mTX{
scripts: []transaction.Witness{
{},
@ -155,6 +158,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "incorrect main TX signers amount",
addW: false,
mTX: mTX{
sigs: []transaction.Signer{{}},
},
@ -162,6 +166,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "incorrect main TX Alphabet signer",
addW: false,
mTX: mTX{
sigs: []transaction.Signer{
{},
@ -175,6 +180,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "incorrect main TX attribute amount",
addW: false,
mTX: mTX{
attrs: []transaction.Attribute{{}, {}},
},
@ -182,6 +188,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "incorrect main TX attribute",
addW: false,
mTX: mTX{
attrs: []transaction.Attribute{
{
@ -195,6 +202,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "incorrect main TX proxy witness",
addW: false,
mTX: mTX{
scripts: []transaction.Witness{
{
@ -210,6 +218,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "incorrect main TX Alphabet witness",
addW: false,
mTX: mTX{
scripts: []transaction.Witness{
{},
@ -224,6 +233,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "incorrect main TX Notary witness",
addW: false,
mTX: mTX{
scripts: []transaction.Witness{
{},
@ -240,6 +250,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "incorrect fb TX attributes amount",
addW: false,
fbTX: fbTX{
attrs: []transaction.Attribute{{}},
},
@ -247,6 +258,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "incorrect fb TX attributes",
addW: false,
fbTX: fbTX{
attrs: []transaction.Attribute{{}, {}, {}},
},
@ -254,6 +266,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
{
name: "expired fb TX",
addW: false,
fbTX: fbTX{
[]transaction.Attribute{
{},
@ -268,6 +281,36 @@ func TestPrepare_IncorrectNR(t *testing.T) {
},
expErr: ErrMainTXExpired,
},
{
name: "incorrect invoker TX Alphabet witness",
addW: true,
mTX: mTX{
scripts: []transaction.Witness{
{},
{
VerificationScript: alphaVerificationScript,
InvocationScript: dummyInvocationScript,
},
{},
{},
},
},
expErr: errIncorrectInvokerWitnesses,
},
{
name: "incorrect main TX attribute with invoker",
addW: true,
mTX: mTX{
attrs: []transaction.Attribute{
{
Value: &transaction.NotaryAssisted{
NKeys: uint8(len(alphaKeys) + 2),
},
},
},
},
expErr: errIncorrectAttribute,
},
}
preparator := notaryPreparator(
@ -284,7 +327,7 @@ func TestPrepare_IncorrectNR(t *testing.T) {
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
correctNR := correctNR(nil)
correctNR := correctNR(nil, test.addW)
incorrectNR = setIncorrectFields(*correctNR, test.mTX, test.fbTX)
_, err = preparator.Prepare(&incorrectNR)
@ -328,7 +371,9 @@ func TestPrepare_CorrectNR(t *testing.T) {
)
for _, test := range tests {
nr := correctNR(script(test.hash, test.method, test.args...))
for i := 0; i < 1; i++ { // run tests against 3 and 4 witness NR
additionalWitness := i == 0
nr := correctNR(script(test.hash, test.method, test.args...), additionalWitness)
event, err := preparator.Prepare(nr)
@ -363,6 +408,7 @@ func TestPrepare_CorrectNR(t *testing.T) {
require.Equal(t, opcode.RET, opCode)
}
}
}
func alphaKeysSource() client.AlphabetKeys {
return func() (keys.PublicKeys, error) {
@ -382,19 +428,22 @@ func script(hash util.Uint160, method string, args ...interface{}) []byte {
return bw.Bytes()
}
func correctNR(script []byte) *payload.P2PNotaryRequest {
func correctNR(script []byte, additionalWitness bool) *payload.P2PNotaryRequest {
alphaVerificationScript, _ := smartcontract.CreateMultiSigRedeemScript(len(alphaKeys)*2/3+1, alphaKeys)
return &payload.P2PNotaryRequest{
MainTransaction: &transaction.Transaction{
Signers: []transaction.Signer{
signers := []transaction.Signer{
{},
{
Account: hash.Hash160(alphaVerificationScript),
},
{},
},
Scripts: []transaction.Witness{
}
if additionalWitness { // insert on element with index 2
signers = append(signers[:2+1], signers[2:]...)
signers[2] = transaction.Signer{Account: hash.Hash160(alphaVerificationScript)}
}
scripts := []transaction.Witness{
{},
{
InvocationScript: dummyInvocationScript,
@ -403,11 +452,28 @@ func correctNR(script []byte) *payload.P2PNotaryRequest {
{
InvocationScript: dummyInvocationScript,
},
},
}
if additionalWitness { // insert on element with index 2
scripts = append(scripts[:2+1], scripts[2:]...)
scripts[2] = transaction.Witness{
InvocationScript: dummyInvocationScript,
VerificationScript: alphaVerificationScript,
}
}
nKeys := uint8(len(alphaKeys))
if additionalWitness {
nKeys++
}
return &payload.P2PNotaryRequest{
MainTransaction: &transaction.Transaction{
Signers: signers,
Scripts: scripts,
Attributes: []transaction.Attribute{
{
Value: &transaction.NotaryAssisted{
NKeys: uint8(len(alphaKeys)),
NKeys: nKeys,
},
},
},