forked from TrueCloudLab/frostfs-api-go
85 lines
1.8 KiB
Go
85 lines
1.8 KiB
Go
|
package accounting
|
||
|
|
|
||
|
|
import (
|
||
|
|
"io/ioutil"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/mr-tron/base58"
|
||
|
|
"github.com/nspcc-dev/neofs-crypto/test"
|
||
|
|
"github.com/nspcc-dev/neofs-proto/chain"
|
||
|
|
"github.com/nspcc-dev/neofs-proto/decimal"
|
||
|
|
"github.com/nspcc-dev/neofs-proto/refs"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCheque(t *testing.T) {
|
||
|
|
t.Run("new/valid", func(t *testing.T) {
|
||
|
|
id, err := NewChequeID()
|
||
|
|
require.NoError(t, err)
|
||
|
|
require.True(t, id.Valid())
|
||
|
|
|
||
|
|
d := make([]byte, chain.AddressLength+1)
|
||
|
|
|
||
|
|
// expected size + 1 byte
|
||
|
|
str := base58.Encode(d)
|
||
|
|
require.False(t, ChequeID(str).Valid())
|
||
|
|
|
||
|
|
// expected size - 1 byte
|
||
|
|
str = base58.Encode(d[:len(d)-2])
|
||
|
|
require.False(t, ChequeID(str).Valid())
|
||
|
|
|
||
|
|
// wrong encoding
|
||
|
|
d = d[:len(d)-1] // normal size
|
||
|
|
require.False(t, ChequeID(string(d)).Valid())
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("marshal/unmarshal", func(t *testing.T) {
|
||
|
|
var b2 = new(Cheque)
|
||
|
|
|
||
|
|
key1 := test.DecodeKey(0)
|
||
|
|
key2 := test.DecodeKey(1)
|
||
|
|
|
||
|
|
id, err := NewChequeID()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
owner, err := refs.NewOwnerID(&key1.PublicKey)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
b1 := &Cheque{
|
||
|
|
ID: id,
|
||
|
|
Owner: owner,
|
||
|
|
Height: 100,
|
||
|
|
Amount: decimal.NewGAS(100),
|
||
|
|
}
|
||
|
|
|
||
|
|
require.NoError(t, b1.Sign(key1))
|
||
|
|
require.NoError(t, b1.Sign(key2))
|
||
|
|
|
||
|
|
data, err := b1.MarshalBinary()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
require.Len(t, data, b1.Size())
|
||
|
|
require.NoError(t, b2.UnmarshalBinary(data))
|
||
|
|
require.Equal(t, b1, b2)
|
||
|
|
|
||
|
|
require.NoError(t, b1.Verify())
|
||
|
|
require.NoError(t, b2.Verify())
|
||
|
|
})
|
||
|
|
|
||
|
|
t.Run("example from SC", func(t *testing.T) {
|
||
|
|
var pathToCheque = "fixtures/cheque_data"
|
||
|
|
expect, err := ioutil.ReadFile(pathToCheque)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
var cheque Cheque
|
||
|
|
require.NoError(t, cheque.UnmarshalBinary(expect))
|
||
|
|
|
||
|
|
actual, err := cheque.MarshalBinary()
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
require.Equal(t, expect, actual)
|
||
|
|
|
||
|
|
require.NoError(t, cheque.Verify())
|
||
|
|
})
|
||
|
|
}
|