mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-12-11 15:30:07 +00:00
94 lines
2.8 KiB
Go
94 lines
2.8 KiB
Go
package manifest
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/nspcc-dev/neo-go/internal/random"
|
|
"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestNewPermission(t *testing.T) {
|
|
require.Panics(t, func() { NewPermission(PermissionWildcard, util.Uint160{}) })
|
|
require.Panics(t, func() { NewPermission(PermissionHash) })
|
|
require.Panics(t, func() { NewPermission(PermissionHash, 1) })
|
|
require.Panics(t, func() { NewPermission(PermissionGroup) })
|
|
require.Panics(t, func() { NewPermission(PermissionGroup, util.Uint160{}) })
|
|
}
|
|
|
|
func TestPermission_MarshalJSON(t *testing.T) {
|
|
t.Run("wildcard", func(t *testing.T) {
|
|
expected := NewPermission(PermissionWildcard)
|
|
expected.Methods.Restrict()
|
|
testMarshalUnmarshal(t, expected, NewPermission(PermissionWildcard))
|
|
})
|
|
|
|
t.Run("group", func(t *testing.T) {
|
|
expected := NewPermission(PermissionWildcard)
|
|
expected.Contract.Type = PermissionGroup
|
|
priv, err := keys.NewPrivateKey()
|
|
require.NoError(t, err)
|
|
expected.Contract.Value = priv.PublicKey()
|
|
expected.Methods.Add("method1")
|
|
expected.Methods.Add("method2")
|
|
testMarshalUnmarshal(t, expected, NewPermission(PermissionWildcard))
|
|
})
|
|
|
|
t.Run("hash", func(t *testing.T) {
|
|
expected := NewPermission(PermissionWildcard)
|
|
expected.Contract.Type = PermissionHash
|
|
expected.Contract.Value = random.Uint160()
|
|
testMarshalUnmarshal(t, expected, NewPermission(PermissionWildcard))
|
|
})
|
|
}
|
|
|
|
func TestPermissionDesc_MarshalJSON(t *testing.T) {
|
|
t.Run("uint160 with 0x", func(t *testing.T) {
|
|
u := random.Uint160()
|
|
s := u.StringLE()
|
|
js := []byte(fmt.Sprintf(`"0x%s"`, s))
|
|
d := new(PermissionDesc)
|
|
require.NoError(t, json.Unmarshal(js, d))
|
|
require.Equal(t, u, d.Value.(util.Uint160))
|
|
})
|
|
|
|
t.Run("invalid uint160", func(t *testing.T) {
|
|
d := new(PermissionDesc)
|
|
s := random.String(util.Uint160Size * 2)
|
|
js := []byte(fmt.Sprintf(`"ok%s"`, s))
|
|
require.Error(t, json.Unmarshal(js, d))
|
|
|
|
js = []byte(fmt.Sprintf(`"%s"`, s))
|
|
require.Error(t, json.Unmarshal(js, d))
|
|
})
|
|
|
|
t.Run("invalid public key", func(t *testing.T) {
|
|
d := new(PermissionDesc)
|
|
s := random.String(65)
|
|
s = "k" + s // not a hex
|
|
js := []byte(fmt.Sprintf(`"%s"`, s))
|
|
require.Error(t, json.Unmarshal(js, d))
|
|
})
|
|
|
|
t.Run("not a string", func(t *testing.T) {
|
|
d := new(PermissionDesc)
|
|
js := []byte(`123`)
|
|
require.Error(t, json.Unmarshal(js, d))
|
|
})
|
|
|
|
t.Run("invalid string", func(t *testing.T) {
|
|
d := new(PermissionDesc)
|
|
js := []byte(`"invalid length"`)
|
|
require.Error(t, json.Unmarshal(js, d))
|
|
})
|
|
}
|
|
|
|
func testMarshalUnmarshal(t *testing.T, expected, actual interface{}) {
|
|
data, err := json.Marshal(expected)
|
|
require.NoError(t, err)
|
|
require.NoError(t, json.Unmarshal(data, actual))
|
|
require.Equal(t, expected, actual)
|
|
}
|