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) }