Add tests for all messages
This commit is contained in:
parent
6c9b62d461
commit
36c508032b
|
@ -0,0 +1,112 @@
|
|||
package ibc
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
|
||||
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 {
|
||||
assert.Nil(t, err, "%d: %+v", i, err)
|
||||
} else {
|
||||
assert.NotNil(t, err, "%d", i)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// -------------------------------
|
||||
// IBCTransferMsg Tests
|
||||
|
||||
func TestIBCTransferMsg(t *testing.T) {
|
||||
packet := constructIBCPacket(true)
|
||||
msg := IBCTransferMsg{packet}
|
||||
|
||||
assert.Equal(t, msg.Type(), "ibc")
|
||||
}
|
||||
|
||||
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 {
|
||||
assert.Nil(t, err, "%d: %+v", i, err)
|
||||
} else {
|
||||
assert.NotNil(t, err, "%d", i)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// -------------------------------
|
||||
// IBCReceiveMsg Tests
|
||||
|
||||
func TestIBCReceiveMsg(t *testing.T) {
|
||||
packet := constructIBCPacket(true)
|
||||
msg := IBCReceiveMsg{packet, sdk.Address([]byte("relayer")), 0}
|
||||
|
||||
assert.Equal(t, msg.Type(), "ibc")
|
||||
}
|
||||
|
||||
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 {
|
||||
assert.Nil(t, err, "%d: %+v", i, err)
|
||||
} else {
|
||||
assert.NotNil(t, err, "%d", i)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// -------------------------------
|
||||
// Helpers
|
||||
|
||||
func constructIBCPacket(valid bool) IBCPacket {
|
||||
srcAddr := sdk.Address([]byte("source"))
|
||||
destAddr := sdk.Address([]byte("destination"))
|
||||
coins := sdk.Coins{{"atom", 10}}
|
||||
srcChain := "source-chain"
|
||||
destChain := "dest-chain"
|
||||
|
||||
if valid {
|
||||
return NewIBCPacket(srcAddr, destAddr, coins, srcChain, destChain)
|
||||
} else {
|
||||
return NewIBCPacket(srcAddr, destAddr, coins, srcChain, srcChain)
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue