neo-go/pkg/util/uint160_test.go

119 lines
3 KiB
Go
Raw Normal View History

package util
import (
"encoding/hex"
"testing"
"github.com/nspcc-dev/neo-go/internal/testserdes"
"github.com/stretchr/testify/assert"
2019-11-24 14:55:50 +00:00
"github.com/stretchr/testify/require"
)
func TestUint160UnmarshalJSON(t *testing.T) {
str := "0263c1de100292813b5e075e585acc1bae963b2d"
expected, err := Uint160DecodeStringLE(str)
2019-11-24 14:55:50 +00:00
assert.NoError(t, err)
2019-10-22 14:56:03 +00:00
// UnmarshalJSON decodes hex-strings
var u1, u2 Uint160
2019-11-24 14:55:50 +00:00
assert.NoError(t, u1.UnmarshalJSON([]byte(`"`+str+`"`)))
assert.True(t, expected.Equals(u1))
testserdes.MarshalUnmarshalJSON(t, &expected, &u2)
2019-11-25 08:43:13 +00:00
assert.Error(t, u2.UnmarshalJSON([]byte(`123`)))
}
func TestUInt160DecodeString(t *testing.T) {
hexStr := "2d3b96ae1bcc5a585e075e3b81920210dec16302"
2019-11-27 09:20:31 +00:00
val, err := Uint160DecodeStringBE(hexStr)
2019-11-24 14:55:50 +00:00
assert.NoError(t, err)
assert.Equal(t, hexStr, val.String())
2019-11-25 08:43:13 +00:00
2019-12-03 13:14:44 +00:00
valLE, err := Uint160DecodeStringLE(hexStr)
assert.NoError(t, err)
assert.Equal(t, val, valLE.Reverse())
2019-11-27 09:20:31 +00:00
_, err = Uint160DecodeStringBE(hexStr[1:])
2019-11-25 08:43:13 +00:00
assert.Error(t, err)
2019-12-03 13:14:44 +00:00
_, err = Uint160DecodeStringLE(hexStr[1:])
assert.Error(t, err)
2019-11-25 08:43:13 +00:00
hexStr = "zz3b96ae1bcc5a585e075e3b81920210dec16302"
2019-11-27 09:20:31 +00:00
_, err = Uint160DecodeStringBE(hexStr)
2019-11-25 08:43:13 +00:00
assert.Error(t, err)
2019-12-03 13:14:44 +00:00
_, err = Uint160DecodeStringLE(hexStr)
assert.Error(t, err)
}
func TestUint160DecodeBytes(t *testing.T) {
hexStr := "2d3b96ae1bcc5a585e075e3b81920210dec16302"
b, err := hex.DecodeString(hexStr)
2019-11-24 14:55:50 +00:00
require.NoError(t, err)
2019-11-27 09:20:31 +00:00
val, err := Uint160DecodeBytesBE(b)
2019-11-24 14:55:50 +00:00
assert.NoError(t, err)
assert.Equal(t, hexStr, val.String())
2019-11-25 08:43:13 +00:00
2019-12-03 13:12:26 +00:00
valLE, err := Uint160DecodeBytesLE(b)
assert.NoError(t, err)
assert.Equal(t, val, valLE.Reverse())
_, err = Uint160DecodeBytesLE(b[1:])
assert.Error(t, err)
2019-11-27 09:20:31 +00:00
_, err = Uint160DecodeBytesBE(b[1:])
2019-11-25 08:43:13 +00:00
assert.Error(t, err)
}
func TestUInt160Equals(t *testing.T) {
a := "2d3b96ae1bcc5a585e075e3b81920210dec16302"
b := "4d3b96ae1bcc5a585e075e3b81920210dec16302"
2019-11-27 09:20:31 +00:00
ua, err := Uint160DecodeStringBE(a)
2019-11-24 14:55:50 +00:00
require.NoError(t, err)
2019-11-27 09:20:31 +00:00
ub, err := Uint160DecodeStringBE(b)
2019-11-24 14:55:50 +00:00
require.NoError(t, err)
assert.False(t, ua.Equals(ub), "%s and %s cannot be equal", ua, ub)
assert.True(t, ua.Equals(ua), "%s and %s must be equal", ua, ua)
}
func TestUInt160Less(t *testing.T) {
a := "2d3b96ae1bcc5a585e075e3b81920210dec16302"
b := "2d3b96ae1bcc5a585e075e3b81920210dec16303"
2019-11-27 09:20:31 +00:00
ua, err := Uint160DecodeStringBE(a)
assert.Nil(t, err)
2019-11-27 09:20:31 +00:00
ua2, err := Uint160DecodeStringBE(a)
assert.Nil(t, err)
2019-11-27 09:20:31 +00:00
ub, err := Uint160DecodeStringBE(b)
assert.Nil(t, err)
assert.Equal(t, true, ua.Less(ub))
assert.Equal(t, false, ua.Less(ua2))
assert.Equal(t, false, ub.Less(ua))
}
func TestUInt160String(t *testing.T) {
hexStr := "b28427088a3729b2536d10122960394e8be6721f"
hexRevStr := "1f72e68b4e39602912106d53b229378a082784b2"
2019-11-27 09:20:31 +00:00
val, err := Uint160DecodeStringBE(hexStr)
assert.Nil(t, err)
assert.Equal(t, hexStr, val.String())
2019-11-27 09:20:31 +00:00
assert.Equal(t, hexRevStr, val.StringLE())
}
2019-12-03 13:07:15 +00:00
func TestUint160_Reverse(t *testing.T) {
hexStr := "b28427088a3729b2536d10122960394e8be6721f"
val, err := Uint160DecodeStringBE(hexStr)
require.NoError(t, err)
assert.Equal(t, hexStr, val.Reverse().StringLE())
assert.Equal(t, val, val.Reverse().Reverse())
}