You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
250 lines
7.3 KiB
250 lines
7.3 KiB
// Copyright 2015 The go-ethereum Authors
|
|
// This file is part of the go-ethereum library.
|
|
//
|
|
// The go-ethereum library is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU Lesser General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// The go-ethereum library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public License
|
|
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
package common
|
|
|
|
import (
|
|
"database/sql/driver"
|
|
"encoding/json"
|
|
"math/big"
|
|
"reflect"
|
|
"strings"
|
|
"testing"
|
|
|
|
ethCommon "github.com/ethereum/go-ethereum/common"
|
|
)
|
|
|
|
func TestIsBech32Address(t *testing.T) {
|
|
tests := []struct {
|
|
str string
|
|
exp bool
|
|
}{
|
|
{"one1ay37rp2pc3kjarg7a322vu3sa8j9puahg679z3", true},
|
|
{"one1fdv7u7rll9epgcqv9xxh9lhwq427nsqldp8ua9", true},
|
|
{"tone1fdv7u7rll9epgcqv9xxh9lhwq427nsqlr5wca5", true},
|
|
{"tone1fdv7u7rll9epgcqv9xxh9lhwq427nsqlr5wca1", false},
|
|
{"xone1fdv7u7rll9epgcqv9xxh9lhwq427nsqlr5wca5", false},
|
|
{"ne1fdv7u7rll9epgcqv9xxh9lhwq427nsqlr5wca5", false},
|
|
{"1one1fdv7u7rll9epgcqv9xxh9lhwq427nsqlr5wca5", false},
|
|
{"one2fdv7u7rll9epgcqv9xxh9lhwq427nsqlr5wca5", false},
|
|
}
|
|
|
|
for _, test := range tests {
|
|
if result := IsBech32Address(test.str); result != test.exp {
|
|
t.Errorf("IsBech32Address(%s) == %v; expected %v",
|
|
test.str, result, test.exp)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestHashJsonValidation(t *testing.T) {
|
|
var tests = []struct {
|
|
Prefix string
|
|
Size int
|
|
Error string
|
|
}{
|
|
{"", 62, "json: cannot unmarshal hex string without 0x prefix into Go value of type common.Hash"},
|
|
{"0x", 66, "hex string has length 66, want 64 for common.Hash"},
|
|
{"0x", 63, "json: cannot unmarshal hex string of odd length into Go value of type common.Hash"},
|
|
{"0x", 0, "hex string has length 0, want 64 for common.Hash"},
|
|
{"0x", 64, ""},
|
|
{"0X", 64, ""},
|
|
}
|
|
for _, test := range tests {
|
|
input := `"` + test.Prefix + strings.Repeat("0", test.Size) + `"`
|
|
var v Hash
|
|
err := json.Unmarshal([]byte(input), &v)
|
|
if err == nil {
|
|
if test.Error != "" {
|
|
t.Errorf("%s: error mismatch: have nil, want %q", input, test.Error)
|
|
}
|
|
} else {
|
|
if err.Error() != test.Error {
|
|
t.Errorf("%s: error mismatch: have %q, want %q", input, err, test.Error)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAddressUnmarshalJSON(t *testing.T) {
|
|
var tests = []struct {
|
|
Input string
|
|
ShouldErr bool
|
|
Output *big.Int
|
|
}{
|
|
{"", true, nil},
|
|
{`""`, true, nil},
|
|
{`"0x"`, true, nil},
|
|
{`"0x00"`, true, nil},
|
|
{`"0xG000000000000000000000000000000000000000"`, true, nil},
|
|
{`"0x0000000000000000000000000000000000000000"`, false, big.NewInt(0)},
|
|
{`"0x0000000000000000000000000000000000000010"`, false, big.NewInt(16)},
|
|
}
|
|
for i, test := range tests {
|
|
var v Address
|
|
err := json.Unmarshal([]byte(test.Input), &v)
|
|
if err != nil && !test.ShouldErr {
|
|
t.Errorf("test #%d: unexpected error: %v", i, err)
|
|
}
|
|
if err == nil {
|
|
if test.ShouldErr {
|
|
t.Errorf("test #%d: expected error, got none", i)
|
|
}
|
|
if got := new(big.Int).SetBytes(v.Bytes()); got.Cmp(test.Output) != 0 {
|
|
t.Errorf("test #%d: address mismatch: have %v, want %v", i, got, test.Output)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestAddressHexChecksum(t *testing.T) {
|
|
var tests = []struct {
|
|
Input string
|
|
Output string
|
|
}{
|
|
// Test cases from https://github.com/ethereum/EIPs/blob/master/EIPS/eip-55.md#specification
|
|
{"0x5aaeb6053f3e94c9b9a09f33669435e7ef1beaed", "0x5aAeb6053F3E94C9b9A09f33669435E7Ef1BeAed"},
|
|
{"0xfb6916095ca1df60bb79ce92ce3ea74c37c5d359", "0xfB6916095ca1df60bB79Ce92cE3Ea74c37c5d359"},
|
|
{"0xdbf03b407c01e7cd3cbea99509d93f8dddc8c6fb", "0xdbF03B407c01E7cD3CBea99509d93f8DDDC8C6FB"},
|
|
{"0xd1220a0cf47c7b9be7a2e6ba89f429762e7b9adb", "0xD1220A0cf47c7B9Be7A2E6BA89F429762e7b9aDb"},
|
|
// Ensure that non-standard length input values are handled correctly
|
|
{"0xa", "0x000000000000000000000000000000000000000A"},
|
|
{"0x0a", "0x000000000000000000000000000000000000000A"},
|
|
{"0x00a", "0x000000000000000000000000000000000000000A"},
|
|
{"0x000000000000000000000000000000000000000a", "0x000000000000000000000000000000000000000A"},
|
|
}
|
|
for i, test := range tests {
|
|
output := HexToAddress(test.Input).Bech32()
|
|
if output != test.Output {
|
|
t.Errorf("test #%d: failed to match when it should (%s != %s)", i, output, test.Output)
|
|
}
|
|
}
|
|
}
|
|
|
|
func BenchmarkAddressBech32(b *testing.B) {
|
|
testAddr := HexToAddress("0x5aaeb6053f3e94c9b9a09f33669435e7ef1beaed")
|
|
for n := 0; n < b.N; n++ {
|
|
testAddr.Bech32()
|
|
}
|
|
}
|
|
|
|
func TestAddressToBech32(t *testing.T) {
|
|
adr := ethCommon.HexToAddress("0x15a128e599b74842bccba860311efa92991bffb5")
|
|
if address, err := AddressToBech32(adr); err == nil {
|
|
if address != "one1zksj3evekayy90xt4psrz8h6j2v3hla4qwz4ur" {
|
|
t.Errorf("error on parseAddr")
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestParseAddr(t *testing.T) {
|
|
adr := ethCommon.HexToAddress("0x15a128e599b74842bccba860311efa92991bffb5")
|
|
adr2 := ParseAddr("one1zksj3evekayy90xt4psrz8h6j2v3hla4qwz4ur")
|
|
if adr.Hex() != adr2.Hex() {
|
|
t.Errorf("error on ParseAddr")
|
|
}
|
|
// Parsing incorrect address
|
|
adr3 := ParseAddr("helloworldone1zksj3evekayy90xt4psrz8h6j2v3hla4qwz4ufdfsrfasdfadfas")
|
|
if adr3.Hex() != "0x0000000000000000000000000000000000000000" {
|
|
t.Errorf("error on ParseAddr")
|
|
}
|
|
}
|
|
|
|
func TestAddress_Scan(t *testing.T) {
|
|
type args struct {
|
|
src interface{}
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "working scan",
|
|
args: args{src: []byte{
|
|
0xb2, 0x6f, 0x2b, 0x34, 0x2a, 0xab, 0x24, 0xbc, 0xf6, 0x3e,
|
|
0xa2, 0x18, 0xc6, 0xa9, 0x27, 0x4d, 0x30, 0xab, 0x9a, 0x15,
|
|
}},
|
|
wantErr: false,
|
|
},
|
|
{
|
|
name: "non working scan",
|
|
args: args{src: int64(1234567890)},
|
|
wantErr: true,
|
|
},
|
|
{
|
|
name: "invalid length scan",
|
|
args: args{src: []byte{
|
|
0xb2, 0x6f, 0x2b, 0x34, 0x2a, 0xab, 0x24, 0xbc, 0xf6, 0x3e,
|
|
0xa2, 0x18, 0xc6, 0xa9, 0x27, 0x4d, 0x30, 0xab, 0x9a,
|
|
}},
|
|
wantErr: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
a := &Address{}
|
|
if err := a.Scan(tt.args.src); (err != nil) != tt.wantErr {
|
|
t.Errorf("Address.Scan() error = %v, wantErr %v", err, tt.wantErr)
|
|
}
|
|
|
|
if !tt.wantErr {
|
|
for i := range a {
|
|
if a[i] != tt.args.src.([]byte)[i] {
|
|
t.Errorf(
|
|
"Address.Scan() didn't scan the %d src correctly (have %X, want %X)",
|
|
i, a[i], tt.args.src.([]byte)[i],
|
|
)
|
|
}
|
|
}
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestAddress_Value(t *testing.T) {
|
|
b := []byte{
|
|
0xb2, 0x6f, 0x2b, 0x34, 0x2a, 0xab, 0x24, 0xbc, 0xf6, 0x3e,
|
|
0xa2, 0x18, 0xc6, 0xa9, 0x27, 0x4d, 0x30, 0xab, 0x9a, 0x15,
|
|
}
|
|
var usedA Address
|
|
usedA.SetBytes(b)
|
|
tests := []struct {
|
|
name string
|
|
a Address
|
|
want driver.Value
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "Working value",
|
|
a: usedA,
|
|
want: b,
|
|
wantErr: false,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := tt.a.Value()
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("Address.Value() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
if !reflect.DeepEqual(got, tt.want) {
|
|
t.Errorf("Address.Value() = %v, want %v", got, tt.want)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|