88 lines
2.2 KiB
Go
88 lines
2.2 KiB
Go
|
package ibc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"strings"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/tendermint/basecoin/testutils"
|
||
|
|
"github.com/tendermint/basecoin/types"
|
||
|
|
cmn "github.com/tendermint/go-common"
|
||
|
|
"github.com/tendermint/go-wire"
|
||
|
|
tm "github.com/tendermint/tendermint/types"
|
||
|
|
)
|
||
|
|
|
||
|
|
func genGenesisDoc(chainID string, numVals int) (*tm.GenesisDoc, []*tm.Validator) {
|
||
|
|
var vals []*tm.Validator
|
||
|
|
genDoc := &tm.GenesisDoc{
|
||
|
|
ChainID: chainID,
|
||
|
|
Validators: nil,
|
||
|
|
}
|
||
|
|
|
||
|
|
for i := 0; i < numVals; i++ {
|
||
|
|
name := cmn.Fmt("%v_val_%v", chainID, i)
|
||
|
|
valPrivAcc := testutils.PrivAccountFromSecret(name)
|
||
|
|
val := tm.NewValidator(valPrivAcc.Account.PubKey, 1)
|
||
|
|
genDoc.Validators = append(genDoc.Validators, tm.GenesisValidator{
|
||
|
|
PubKey: val.PubKey,
|
||
|
|
Amount: 1,
|
||
|
|
Name: name,
|
||
|
|
})
|
||
|
|
vals = append(vals, val)
|
||
|
|
}
|
||
|
|
|
||
|
|
return genDoc, vals
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestIBCPlugin(t *testing.T) {
|
||
|
|
|
||
|
|
store := types.NewKVCache(nil)
|
||
|
|
store.SetLogging() // Log all activity
|
||
|
|
|
||
|
|
ibcPlugin := New()
|
||
|
|
ctx := types.CallContext{
|
||
|
|
CallerAddress: nil,
|
||
|
|
CallerAccount: nil,
|
||
|
|
Coins: types.Coins{},
|
||
|
|
}
|
||
|
|
|
||
|
|
chainID_1 := "test_chain"
|
||
|
|
genDoc_1, vals_1 := genGenesisDoc(chainID_1, 4)
|
||
|
|
genDocJSON_1 := wire.JSONBytesPretty(genDoc_1)
|
||
|
|
|
||
|
|
// Register a malformed chain
|
||
|
|
res := ibcPlugin.RunTx(store, ctx, wire.BinaryBytes(struct{ IBCTx }{IBCRegisterChainTx{
|
||
|
|
BlockchainGenesis{
|
||
|
|
ChainID: "test_chain",
|
||
|
|
Genesis: "<THIS IS NOT JSON>",
|
||
|
|
},
|
||
|
|
}}))
|
||
|
|
assert.Equal(t, res.Code, IBCCodeEncodingError)
|
||
|
|
t.Log(">>", strings.Join(store.GetLogLines(), "\n"))
|
||
|
|
store.ClearLogLines()
|
||
|
|
|
||
|
|
// Successfully register a chain
|
||
|
|
res = ibcPlugin.RunTx(store, ctx, wire.BinaryBytes(struct{ IBCTx }{IBCRegisterChainTx{
|
||
|
|
BlockchainGenesis{
|
||
|
|
ChainID: "test_chain",
|
||
|
|
Genesis: string(genDocJSON_1),
|
||
|
|
},
|
||
|
|
}}))
|
||
|
|
assert.True(t, res.IsOK(), res)
|
||
|
|
t.Log(">>", strings.Join(store.GetLogLines(), "\n"))
|
||
|
|
store.ClearLogLines()
|
||
|
|
|
||
|
|
// Duplicate request fails
|
||
|
|
res = ibcPlugin.RunTx(store, ctx, wire.BinaryBytes(struct{ IBCTx }{IBCRegisterChainTx{
|
||
|
|
BlockchainGenesis{
|
||
|
|
ChainID: "test_chain",
|
||
|
|
Genesis: string(genDocJSON_1),
|
||
|
|
},
|
||
|
|
}}))
|
||
|
|
assert.Equal(t, res.Code, IBCCodeChainAlreadyExists, res)
|
||
|
|
t.Log(">>", strings.Join(store.GetLogLines(), "\n"))
|
||
|
|
store.ClearLogLines()
|
||
|
|
|
||
|
|
t.Log(">>", vals_1)
|
||
|
|
}
|