forked from TrueCloudLab/neoneo-go
dba248236c
And use it where appropriate. Some of our code was just plain wrong (like the one in GAS contract) and unification is always useful here.
85 lines
2.2 KiB
Go
85 lines
2.2 KiB
Go
package smartcontract
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/io"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/emit"
|
|
"github.com/nspcc-dev/neo-go/pkg/vm/opcode"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCreateMultiSigRedeemScript(t *testing.T) {
|
|
val1, _ := keys.NewPublicKeyFromString("03b209fd4f53a7170ea4444e0cb0a6bb6a53c2bd016926989cf85f9b0fba17a70c")
|
|
val2, _ := keys.NewPublicKeyFromString("02df48f60e8f3e01c48ff40b9b7f1310d7a8b2a193188befe1c2e3df740e895093")
|
|
val3, _ := keys.NewPublicKeyFromString("03b8d9d5771d8f513aa0869b9cc8d50986403b78c6da36890638c3d46a5adce04a")
|
|
|
|
validators := []*keys.PublicKey{val1, val2, val3}
|
|
|
|
out, err := CreateMultiSigRedeemScript(3, validators)
|
|
require.NoError(t, err)
|
|
|
|
br := io.NewBinReaderFromBuf(out)
|
|
assert.Equal(t, opcode.PUSH3, opcode.Opcode(br.ReadB()))
|
|
|
|
for i := 0; i < len(validators); i++ {
|
|
assert.EqualValues(t, opcode.PUSHDATA1, br.ReadB())
|
|
bb := br.ReadVarBytes()
|
|
require.NoError(t, br.Err)
|
|
assert.Equal(t, validators[i].Bytes(), bb)
|
|
}
|
|
|
|
assert.Equal(t, opcode.PUSH3, opcode.Opcode(br.ReadB()))
|
|
assert.Equal(t, opcode.PUSHNULL, opcode.Opcode(br.ReadB()))
|
|
assert.Equal(t, opcode.SYSCALL, opcode.Opcode(br.ReadB()))
|
|
assert.Equal(t, emit.InteropNameToID([]byte("Neo.Crypto.CheckMultisigWithECDsaSecp256r1")), br.ReadU32LE())
|
|
}
|
|
|
|
func TestCreateDefaultMultiSigRedeemScript(t *testing.T) {
|
|
var validators = make([]*keys.PublicKey, 0)
|
|
|
|
var addKey = func() {
|
|
key, err := keys.NewPrivateKey()
|
|
require.NoError(t, err)
|
|
validators = append(validators, key.PublicKey())
|
|
}
|
|
var checkM = func(m int) {
|
|
validScript, err := CreateMultiSigRedeemScript(m, validators)
|
|
require.NoError(t, err)
|
|
defaultScript, err := CreateDefaultMultiSigRedeemScript(validators)
|
|
require.NoError(t, err)
|
|
require.Equal(t, validScript, defaultScript)
|
|
}
|
|
|
|
// 1 out of 1
|
|
addKey()
|
|
checkM(1)
|
|
|
|
// 2 out of 2
|
|
addKey()
|
|
checkM(2)
|
|
|
|
// 3 out of 4
|
|
for i := 0; i < 2; i++ {
|
|
addKey()
|
|
}
|
|
checkM(3)
|
|
|
|
// 5 out of 6
|
|
for i := 0; i < 2; i++ {
|
|
addKey()
|
|
}
|
|
checkM(5)
|
|
|
|
// 5 out of 7
|
|
addKey()
|
|
checkM(5)
|
|
|
|
// 7 out of 10
|
|
for i := 0; i < 3; i++ {
|
|
addKey()
|
|
}
|
|
checkM(7)
|
|
}
|