frostfs-api-go-pogpp/pkg/owner/id_test.go
Evgeniy Kulikov 4e6350f6d4 [#147] Work around owner package
- implement Stringer interface and Parse method for owner.ID
- increase test coverage

closes #147

Signed-off-by: Evgeniy Kulikov <kim@nspcc.ru>
2020-10-15 11:05:37 +03:00

65 lines
1.4 KiB
Go

package owner
import (
"crypto/rand"
"strconv"
"testing"
"github.com/mr-tron/base58"
"github.com/nspcc-dev/neofs-crypto/test"
"github.com/stretchr/testify/require"
)
func TestIDV2(t *testing.T) {
id := NewID()
wallet := new(NEO3Wallet)
_, err := rand.Read(wallet.Bytes())
require.NoError(t, err)
id.SetNeo3Wallet(wallet)
idV2 := id.ToV2()
require.Equal(t, wallet.Bytes(), idV2.GetValue())
}
func TestNewIDFromNeo3Wallet(t *testing.T) {
wallet, err := NEO3WalletFromPublicKey(&test.DecodeKey(1).PublicKey)
require.NoError(t, err)
id := NewIDFromNeo3Wallet(wallet)
require.Equal(t, id.ToV2().GetValue(), wallet.Bytes())
}
func TestID_Parse(t *testing.T) {
t.Run("should parse successful", func(t *testing.T) {
for i := 0; i < 10; i++ {
j := i
t.Run(strconv.Itoa(j), func(t *testing.T) {
wallet, err := NEO3WalletFromPublicKey(&test.DecodeKey(j).PublicKey)
require.NoError(t, err)
eid := NewIDFromNeo3Wallet(wallet)
aid := NewID()
require.NoError(t, aid.Parse(eid.String()))
require.Equal(t, eid, aid)
})
}
})
t.Run("should failure on parse", func(t *testing.T) {
for i := 0; i < 10; i++ {
j := i
t.Run(strconv.Itoa(j), func(t *testing.T) {
cs := []byte{1, 2, 3, 4, 5, byte(j)}
str := base58.Encode(cs)
cid := NewID()
require.EqualError(t, cid.Parse(str), ErrBadID.Error())
})
}
})
}