103 lines
3.2 KiB
Go
103 lines
3.2 KiB
Go
package cli
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/pkg/errors"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
"github.com/ethereum/go-ethereum/common"
|
|
)
|
|
|
|
func cosmosAddressFromArg(addr string) (sdk.AccAddress, error) {
|
|
if strings.HasPrefix(addr, sdk.GetConfig().GetBech32AccountAddrPrefix()) {
|
|
// Check to see if address is Cosmos bech32 formatted
|
|
toAddr, err := sdk.AccAddressFromBech32(addr)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "invalid bech32 formatted address")
|
|
}
|
|
return toAddr, nil
|
|
}
|
|
|
|
// Strip 0x prefix if exists
|
|
addr = strings.TrimPrefix(addr, "0x")
|
|
|
|
return sdk.AccAddressFromHexUnsafe(addr)
|
|
}
|
|
|
|
func TestAddressFormats(t *testing.T) {
|
|
testCases := []struct {
|
|
name string
|
|
addrString string
|
|
expectedHex string
|
|
expectErr bool
|
|
}{
|
|
{"Cosmos Address", "cosmos18wvvwfmq77a6d8tza4h5sfuy2yj3jj88yqg82a", "0x3B98c72760f7BBa69D62ED6f48278451251948e7", false},
|
|
{"hex without 0x", "3B98C72760F7BBA69D62ED6F48278451251948E7", "0x3B98c72760f7BBa69D62ED6f48278451251948e7", false},
|
|
{"hex with mixed casing", "3b98C72760f7BBA69D62ED6F48278451251948e7", "0x3B98c72760f7BBa69D62ED6f48278451251948e7", false},
|
|
{"hex with 0x", "0x3B98C72760F7BBA69D62ED6F48278451251948E7", "0x3B98c72760f7BBa69D62ED6f48278451251948e7", false},
|
|
{"invalid hex ethereum address", "0x3B98C72760F7BBA69D62ED6F48278451251948E", "", true},
|
|
{"invalid Cosmos address", "cosmos18wvvwfmq77a6d8tza4h5sfuy2yj3jj88", "", true},
|
|
{"empty string", "", "", true},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
tc := tc
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
hex, err := accountToHex(tc.addrString)
|
|
require.Equal(t, tc.expectErr, err != nil, err)
|
|
|
|
if !tc.expectErr {
|
|
require.Equal(t, hex, tc.expectedHex)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestCosmosToEthereumTypes(t *testing.T) {
|
|
hexString := "0x3B98D72760f7bbA69d62Ed6F48278451251948E7"
|
|
cosmosAddr, err := sdk.AccAddressFromHexUnsafe(hexString[2:])
|
|
require.NoError(t, err)
|
|
|
|
cosmosFormatted := cosmosAddr.String()
|
|
|
|
// Test decoding a cosmos formatted address
|
|
decodedHex, err := accountToHex(cosmosFormatted)
|
|
require.NoError(t, err)
|
|
require.Equal(t, hexString, decodedHex)
|
|
|
|
// Test converting cosmos address with eth address from hex
|
|
hexEth := common.HexToAddress(hexString)
|
|
convertedEth := common.BytesToAddress(cosmosAddr.Bytes())
|
|
require.Equal(t, hexEth, convertedEth)
|
|
|
|
// Test decoding eth hex output against hex string
|
|
ethDecoded, err := accountToHex(hexEth.Hex())
|
|
require.NoError(t, err)
|
|
require.Equal(t, hexString, ethDecoded)
|
|
}
|
|
|
|
func TestAddressToCosmosAddress(t *testing.T) {
|
|
baseAddr, err := sdk.AccAddressFromHexUnsafe("6A98D72760f7bbA69d62Ed6F48278451251948E7")
|
|
require.NoError(t, err)
|
|
|
|
// Test cosmos string back to address
|
|
cosmosFormatted, err := cosmosAddressFromArg(baseAddr.String())
|
|
require.NoError(t, err)
|
|
require.Equal(t, baseAddr, cosmosFormatted)
|
|
|
|
// Test account address from Ethereum address
|
|
ethAddr := common.BytesToAddress(baseAddr.Bytes())
|
|
ethFormatted, err := cosmosAddressFromArg(ethAddr.Hex())
|
|
require.NoError(t, err)
|
|
require.Equal(t, baseAddr, ethFormatted)
|
|
|
|
// Test encoding without the 0x prefix
|
|
ethFormatted, err = cosmosAddressFromArg(ethAddr.Hex()[2:])
|
|
require.NoError(t, err)
|
|
require.Equal(t, baseAddr, ethFormatted)
|
|
}
|