68 lines
1.5 KiB
Go
68 lines
1.5 KiB
Go
package util
|
|
|
|
import (
|
|
"encoding/hex"
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestUint160UnmarshalJSON(t *testing.T) {
|
|
str := "2d3b96ae1bcc5a585e075e3b81920210dec16302"
|
|
expected, _ := Uint160DecodeString(str)
|
|
|
|
// UnmarshalJSON should decode hex-strings
|
|
var u1 Uint160
|
|
s, _ := json.Marshal(str)
|
|
assert.Nil(t, json.Unmarshal(s, &u1))
|
|
assert.True(t, expected.Equals(u1))
|
|
|
|
// UnmarshalJSON should decode hex-strings prefixed by 0x
|
|
var u2 Uint160
|
|
s, _ = json.Marshal("0x" + str)
|
|
assert.Nil(t, json.Unmarshal(s, &u2))
|
|
assert.True(t, expected.Equals(u2))
|
|
}
|
|
|
|
func TestUInt160DecodeString(t *testing.T) {
|
|
hexStr := "2d3b96ae1bcc5a585e075e3b81920210dec16302"
|
|
val, err := Uint160DecodeString(hexStr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, hexStr, val.String())
|
|
}
|
|
|
|
func TestUint160DecodeBytes(t *testing.T) {
|
|
hexStr := "2d3b96ae1bcc5a585e075e3b81920210dec16302"
|
|
b, err := hex.DecodeString(hexStr)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
val, err := Uint160DecodeBytes(b)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
assert.Equal(t, hexStr, val.String())
|
|
}
|
|
|
|
func TestUInt160Equals(t *testing.T) {
|
|
a := "2d3b96ae1bcc5a585e075e3b81920210dec16302"
|
|
b := "4d3b96ae1bcc5a585e075e3b81920210dec16302"
|
|
|
|
ua, err := Uint160DecodeString(a)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
ub, err := Uint160DecodeString(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)
|
|
}
|
|
}
|