2018-03-18 07:19:29 -07:00
|
|
|
package ibc
|
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
2018-06-29 18:10:15 -07:00
|
|
|
"github.com/stretchr/testify/require"
|
2018-03-18 07:19:29 -07:00
|
|
|
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
)
|
|
|
|
|
|
|
|
// --------------------------------
|
|
|
|
// IBCPacket Tests
|
|
|
|
|
|
|
|
func TestIBCPacketValidation(t *testing.T) {
|
|
|
|
cases := []struct {
|
|
|
|
valid bool
|
|
|
|
packet IBCPacket
|
|
|
|
}{
|
|
|
|
{true, constructIBCPacket(true)},
|
|
|
|
{false, constructIBCPacket(false)},
|
|
|
|
}
|
|
|
|
|
|
|
|
for i, tc := range cases {
|
|
|
|
err := tc.packet.ValidateBasic()
|
|
|
|
if tc.valid {
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err, "%d: %+v", i, err)
|
2018-03-18 07:19:29 -07:00
|
|
|
} else {
|
2018-06-29 18:10:15 -07:00
|
|
|
require.NotNil(t, err, "%d", i)
|
2018-03-18 07:19:29 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// -------------------------------
|
|
|
|
// IBCTransferMsg Tests
|
|
|
|
|
|
|
|
func TestIBCTransferMsg(t *testing.T) {
|
|
|
|
packet := constructIBCPacket(true)
|
|
|
|
msg := IBCTransferMsg{packet}
|
|
|
|
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, msg.Type(), "ibc")
|
2018-03-18 07:19:29 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestIBCTransferMsgValidation(t *testing.T) {
|
|
|
|
validPacket := constructIBCPacket(true)
|
|
|
|
invalidPacket := constructIBCPacket(false)
|
|
|
|
|
|
|
|
cases := []struct {
|
|
|
|
valid bool
|
|
|
|
msg IBCTransferMsg
|
|
|
|
}{
|
|
|
|
{true, IBCTransferMsg{validPacket}},
|
|
|
|
{false, IBCTransferMsg{invalidPacket}},
|
|
|
|
}
|
|
|
|
|
|
|
|
for i, tc := range cases {
|
|
|
|
err := tc.msg.ValidateBasic()
|
|
|
|
if tc.valid {
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err, "%d: %+v", i, err)
|
2018-03-18 07:19:29 -07:00
|
|
|
} else {
|
2018-06-29 18:10:15 -07:00
|
|
|
require.NotNil(t, err, "%d", i)
|
2018-03-18 07:19:29 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// -------------------------------
|
|
|
|
// IBCReceiveMsg Tests
|
|
|
|
|
|
|
|
func TestIBCReceiveMsg(t *testing.T) {
|
|
|
|
packet := constructIBCPacket(true)
|
|
|
|
msg := IBCReceiveMsg{packet, sdk.Address([]byte("relayer")), 0}
|
|
|
|
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Equal(t, msg.Type(), "ibc")
|
2018-03-18 07:19:29 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
func TestIBCReceiveMsgValidation(t *testing.T) {
|
|
|
|
validPacket := constructIBCPacket(true)
|
|
|
|
invalidPacket := constructIBCPacket(false)
|
|
|
|
|
|
|
|
cases := []struct {
|
|
|
|
valid bool
|
|
|
|
msg IBCReceiveMsg
|
|
|
|
}{
|
|
|
|
{true, IBCReceiveMsg{validPacket, sdk.Address([]byte("relayer")), 0}},
|
|
|
|
{false, IBCReceiveMsg{invalidPacket, sdk.Address([]byte("relayer")), 0}},
|
|
|
|
}
|
|
|
|
|
|
|
|
for i, tc := range cases {
|
|
|
|
err := tc.msg.ValidateBasic()
|
|
|
|
if tc.valid {
|
2018-06-29 18:10:15 -07:00
|
|
|
require.Nil(t, err, "%d: %+v", i, err)
|
2018-03-18 07:19:29 -07:00
|
|
|
} else {
|
2018-06-29 18:10:15 -07:00
|
|
|
require.NotNil(t, err, "%d", i)
|
2018-03-18 07:19:29 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// -------------------------------
|
|
|
|
// Helpers
|
|
|
|
|
|
|
|
func constructIBCPacket(valid bool) IBCPacket {
|
|
|
|
srcAddr := sdk.Address([]byte("source"))
|
|
|
|
destAddr := sdk.Address([]byte("destination"))
|
2018-06-15 14:16:45 -07:00
|
|
|
coins := sdk.Coins{sdk.NewCoin("atom", 10)}
|
2018-03-18 07:19:29 -07:00
|
|
|
srcChain := "source-chain"
|
|
|
|
destChain := "dest-chain"
|
|
|
|
|
|
|
|
if valid {
|
|
|
|
return NewIBCPacket(srcAddr, destAddr, coins, srcChain, destChain)
|
|
|
|
}
|
2018-04-18 21:49:24 -07:00
|
|
|
return NewIBCPacket(srcAddr, destAddr, coins, srcChain, srcChain)
|
2018-03-18 07:19:29 -07:00
|
|
|
}
|