135 lines
2.5 KiB
Go
135 lines
2.5 KiB
Go
package user_test
|
|
|
|
import (
|
|
"bytes"
|
|
"crypto/rand"
|
|
"testing"
|
|
|
|
"git.frostfs.info/TrueCloudLab/frostfs-sdk-go/api/refs"
|
|
. "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user"
|
|
usertest "git.frostfs.info/TrueCloudLab/frostfs-sdk-go/user/test"
|
|
"github.com/mr-tron/base58"
|
|
"github.com/nspcc-dev/neo-go/pkg/encoding/address"
|
|
"github.com/nspcc-dev/neo-go/pkg/util"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestID_WalletBytes(t *testing.T) {
|
|
var scriptHash util.Uint160
|
|
rand.Read(scriptHash[:])
|
|
|
|
var id ID
|
|
id.SetScriptHash(scriptHash)
|
|
|
|
w := id.WalletBytes()
|
|
|
|
var m refs.OwnerID
|
|
m.SetValue(w)
|
|
|
|
err := id.ReadFromV2(m)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestID_SetScriptHash(t *testing.T) {
|
|
var scriptHash util.Uint160
|
|
rand.Read(scriptHash[:])
|
|
|
|
var id ID
|
|
id.SetScriptHash(scriptHash)
|
|
|
|
var m refs.OwnerID
|
|
id.WriteToV2(&m)
|
|
|
|
var id2 ID
|
|
|
|
err := id2.ReadFromV2(m)
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, id2.Equals(id))
|
|
}
|
|
|
|
func TestID_ScriptHash(t *testing.T) {
|
|
userID := usertest.ID()
|
|
|
|
scriptHash, err := userID.ScriptHash()
|
|
require.NoError(t, err)
|
|
|
|
ownerAddress := userID.EncodeToString()
|
|
decodedScriptHash, err := address.StringToUint160(ownerAddress)
|
|
require.NoError(t, err)
|
|
|
|
require.True(t, scriptHash.Equals(decodedScriptHash))
|
|
}
|
|
|
|
func TestV2_ID(t *testing.T) {
|
|
id := usertest.ID()
|
|
var m refs.OwnerID
|
|
var id2 ID
|
|
|
|
t.Run("OK", func(t *testing.T) {
|
|
id.WriteToV2(&m)
|
|
|
|
err := id2.ReadFromV2(m)
|
|
require.NoError(t, err)
|
|
require.True(t, id2.Equals(id))
|
|
})
|
|
|
|
val := m.GetValue()
|
|
|
|
t.Run("invalid size", func(t *testing.T) {
|
|
m.SetValue(val[:24])
|
|
|
|
err := id2.ReadFromV2(m)
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("invalid prefix", func(t *testing.T) {
|
|
val := bytes.Clone(val)
|
|
val[0]++
|
|
|
|
m.SetValue(val)
|
|
|
|
err := id2.ReadFromV2(m)
|
|
require.Error(t, err)
|
|
})
|
|
|
|
t.Run("invalid checksum", func(t *testing.T) {
|
|
val := bytes.Clone(val)
|
|
val[21]++
|
|
|
|
m.SetValue(val)
|
|
|
|
err := id2.ReadFromV2(m)
|
|
require.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func TestID_EncodeToString(t *testing.T) {
|
|
id := usertest.ID()
|
|
|
|
s := id.EncodeToString()
|
|
|
|
_, err := base58.Decode(s)
|
|
require.NoError(t, err)
|
|
|
|
var id2 ID
|
|
|
|
err = id2.DecodeString(s)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, id, id2)
|
|
|
|
err = id2.DecodeString("_") // any invalid bas58 string
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestID_Equal(t *testing.T) {
|
|
id1 := usertest.ID()
|
|
id2 := usertest.ID()
|
|
id3 := id1
|
|
|
|
require.True(t, id1.Equals(id1)) // self-equality
|
|
require.True(t, id1.Equals(id3))
|
|
require.True(t, id3.Equals(id1)) // commutativity
|
|
require.False(t, id1.Equals(id2))
|
|
}
|