mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-10-16 15:43:25 +00:00
95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
package util
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/io"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestUint256UnmarshalJSON(t *testing.T) {
|
|
str := "f037308fa0ab18155bccfc08485468c112409ea5064595699e98c545f245f32d"
|
|
expected, err := Uint256DecodeReverseString(str)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// UnmarshalJSON decodes hex-strings
|
|
var u1, u2 Uint256
|
|
|
|
if err = u1.UnmarshalJSON([]byte(`"` + str + `"`)); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.True(t, expected.Equals(u1))
|
|
|
|
s, err := expected.MarshalJSON()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// UnmarshalJSON decodes hex-strings prefixed by 0x
|
|
if err = u2.UnmarshalJSON(s); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.True(t, expected.Equals(u1))
|
|
}
|
|
|
|
func TestUint256DecodeString(t *testing.T) {
|
|
hexStr := "f037308fa0ab18155bccfc08485468c112409ea5064595699e98c545f245f32d"
|
|
val, err := Uint256DecodeReverseString(hexStr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, hexStr, val.ReverseString())
|
|
}
|
|
|
|
func TestUint256DecodeBytes(t *testing.T) {
|
|
hexStr := "f037308fa0ab18155bccfc08485468c112409ea5064595699e98c545f245f32d"
|
|
b, err := hex.DecodeString(hexStr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
val, err := Uint256DecodeReverseBytes(b)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, hexStr, val.ReverseString())
|
|
}
|
|
|
|
func TestUInt256Equals(t *testing.T) {
|
|
a := "f037308fa0ab18155bccfc08485468c112409ea5064595699e98c545f245f32d"
|
|
b := "e287c5b29a1b66092be6803c59c765308ac20287e1b4977fd399da5fc8f66ab5"
|
|
|
|
ua, err := Uint256DecodeReverseString(a)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
ub, err := Uint256DecodeReverseString(b)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if ua.Equals(ub) {
|
|
t.Fatalf("%s and %s cannot be equal", ua, ub)
|
|
}
|
|
if !ua.Equals(ua) {
|
|
t.Fatalf("%s and %s must be equal", ua, ua)
|
|
}
|
|
}
|
|
|
|
func TestUint256_Serializable(t *testing.T) {
|
|
a := Uint256{
|
|
1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16,
|
|
17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32,
|
|
}
|
|
|
|
w := io.NewBufBinWriter()
|
|
a.EncodeBinary(w.BinWriter)
|
|
require.NoError(t, w.Err)
|
|
|
|
var b Uint256
|
|
r := io.NewBinReaderFromBuf(w.Bytes())
|
|
b.DecodeBinary(r)
|
|
require.Equal(t, a, b)
|
|
}
|