wormhole/sdk/vaa/payloads_test.go

87 lines
4.2 KiB
Go

package vaa
import (
"encoding/hex"
"testing"
"github.com/ethereum/go-ethereum/common"
"github.com/stretchr/testify/assert"
)
func TestCoreModule(t *testing.T) {
hexifiedCoreModule := "00000000000000000000000000000000000000000000000000000000436f7265"
assert.Equal(t, hex.EncodeToString(CoreModule), hexifiedCoreModule)
}
func TestBodyContractUpgrade(t *testing.T) {
addr := Address{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4}
test := BodyContractUpgrade{ChainID: 1, NewContract: addr}
assert.Equal(t, test.ChainID, ChainID(1))
assert.Equal(t, test.NewContract, addr)
}
func TestBodyGuardianSetUpdate(t *testing.T) {
keys := []common.Address{
common.HexToAddress("0x5aaeb6053f3e94c9b9a09f33669435e7ef1beaed"),
common.HexToAddress("0x5aaeb6053f3e94c9b9a09f33669435e7ef1beaee"),
}
test := BodyGuardianSetUpdate{Keys: keys, NewIndex: uint32(1)}
assert.Equal(t, test.Keys, keys)
assert.Equal(t, test.NewIndex, uint32(1))
}
func TestBodyTokenBridgeRegisterChain(t *testing.T) {
module := "test"
addr := Address{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4}
test := BodyTokenBridgeRegisterChain{Module: module, ChainID: 1, EmitterAddress: addr}
assert.Equal(t, test.Module, module)
assert.Equal(t, test.ChainID, ChainID(1))
assert.Equal(t, test.EmitterAddress, addr)
}
func TestBodyTokenBridgeUpgradeContract(t *testing.T) {
module := "test"
addr := Address{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4}
test := BodyTokenBridgeUpgradeContract{Module: module, TargetChainID: 1, NewContract: addr}
assert.Equal(t, test.Module, module)
assert.Equal(t, test.TargetChainID, ChainID(1))
assert.Equal(t, test.NewContract, addr)
}
func TestBodyContractUpgradeSerialize(t *testing.T) {
addr := Address{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4}
bodyContractUpgrade := BodyContractUpgrade{ChainID: 1, NewContract: addr}
expected := "00000000000000000000000000000000000000000000000000000000436f72650100010000000000000000000000000000000000000000000000000000000000000004"
serializedBodyContractUpgrade := bodyContractUpgrade.Serialize()
assert.Equal(t, expected, hex.EncodeToString(serializedBodyContractUpgrade))
}
func TestBodyGuardianSetUpdateSerialize(t *testing.T) {
keys := []common.Address{
common.HexToAddress("0x5aaeb6053f3e94c9b9a09f33669435e7ef1beaed"),
common.HexToAddress("0x5aaeb6053f3e94c9b9a09f33669435e7ef1beaee"),
}
bodyGuardianSetUpdate := BodyGuardianSetUpdate{Keys: keys, NewIndex: uint32(1)}
expected := "00000000000000000000000000000000000000000000000000000000436f726502000000000001025aaeb6053f3e94c9b9a09f33669435e7ef1beaed5aaeb6053f3e94c9b9a09f33669435e7ef1beaee"
serializedBodyGuardianSetUpdate := bodyGuardianSetUpdate.Serialize()
assert.Equal(t, expected, hex.EncodeToString(serializedBodyGuardianSetUpdate))
}
func TestBodyTokenBridgeRegisterChainSerialize(t *testing.T) {
module := "test"
addr := Address{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4}
bodyTokenBridgeRegisterChain := BodyTokenBridgeRegisterChain{Module: module, ChainID: 1, EmitterAddress: addr}
expected := "000000000000000000000000000000000000000000000000000000007465737401000000010000000000000000000000000000000000000000000000000000000000000004"
serializedBodyTokenBridgeRegisterChain := bodyTokenBridgeRegisterChain.Serialize()
assert.Equal(t, expected, hex.EncodeToString(serializedBodyTokenBridgeRegisterChain))
}
func TestBodyTokenBridgeUpgradeContractSerialize(t *testing.T) {
module := "test"
addr := Address{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4}
bodyTokenBridgeUpgradeContract := BodyTokenBridgeUpgradeContract{Module: module, TargetChainID: 1, NewContract: addr}
expected := "00000000000000000000000000000000000000000000000000000000746573740200010000000000000000000000000000000000000000000000000000000000000004"
serializedBodyTokenBridgeUpgradeContract := bodyTokenBridgeUpgradeContract.Serialize()
assert.Equal(t, expected, hex.EncodeToString(serializedBodyTokenBridgeUpgradeContract))
}