You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
125 lines
4.3 KiB
125 lines
4.3 KiB
package p2p
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNodeInfoValidate(t *testing.T) {
|
|
|
|
// empty fails
|
|
ni := DefaultNodeInfo{}
|
|
assert.Error(t, ni.Validate())
|
|
|
|
channels := make([]byte, maxNumChannels)
|
|
for i := 0; i < maxNumChannels; i++ {
|
|
channels[i] = byte(i)
|
|
}
|
|
dupChannels := make([]byte, 5)
|
|
copy(dupChannels, channels[:5])
|
|
dupChannels = append(dupChannels, testCh)
|
|
|
|
nonASCII := "¢§µ"
|
|
emptyTab := "\t"
|
|
emptySpace := " "
|
|
|
|
testCases := []struct {
|
|
testName string
|
|
malleateNodeInfo func(*DefaultNodeInfo)
|
|
expectErr bool
|
|
}{
|
|
{
|
|
"Too Many Channels",
|
|
func(ni *DefaultNodeInfo) { ni.Channels = append(channels, byte(maxNumChannels)) }, // nolint: gocritic
|
|
true,
|
|
},
|
|
{"Duplicate Channel", func(ni *DefaultNodeInfo) { ni.Channels = dupChannels }, true},
|
|
{"Good Channels", func(ni *DefaultNodeInfo) { ni.Channels = ni.Channels[:5] }, false},
|
|
|
|
{"Invalid NetAddress", func(ni *DefaultNodeInfo) { ni.ListenAddr = "not-an-address" }, true},
|
|
{"Good NetAddress", func(ni *DefaultNodeInfo) { ni.ListenAddr = "0.0.0.0:26656" }, false},
|
|
|
|
{"Non-ASCII Version", func(ni *DefaultNodeInfo) { ni.Version = nonASCII }, true},
|
|
{"Empty tab Version", func(ni *DefaultNodeInfo) { ni.Version = emptyTab }, true},
|
|
{"Empty space Version", func(ni *DefaultNodeInfo) { ni.Version = emptySpace }, true},
|
|
{"Empty Version", func(ni *DefaultNodeInfo) { ni.Version = "" }, false},
|
|
|
|
{"Non-ASCII Moniker", func(ni *DefaultNodeInfo) { ni.Moniker = nonASCII }, true},
|
|
{"Empty tab Moniker", func(ni *DefaultNodeInfo) { ni.Moniker = emptyTab }, true},
|
|
{"Empty space Moniker", func(ni *DefaultNodeInfo) { ni.Moniker = emptySpace }, true},
|
|
{"Empty Moniker", func(ni *DefaultNodeInfo) { ni.Moniker = "" }, true},
|
|
{"Good Moniker", func(ni *DefaultNodeInfo) { ni.Moniker = "hey its me" }, false},
|
|
|
|
{"Non-ASCII TxIndex", func(ni *DefaultNodeInfo) { ni.Other.TxIndex = nonASCII }, true},
|
|
{"Empty tab TxIndex", func(ni *DefaultNodeInfo) { ni.Other.TxIndex = emptyTab }, true},
|
|
{"Empty space TxIndex", func(ni *DefaultNodeInfo) { ni.Other.TxIndex = emptySpace }, true},
|
|
{"Empty TxIndex", func(ni *DefaultNodeInfo) { ni.Other.TxIndex = "" }, false},
|
|
{"Off TxIndex", func(ni *DefaultNodeInfo) { ni.Other.TxIndex = "off" }, false},
|
|
|
|
{"Non-ASCII RPCAddress", func(ni *DefaultNodeInfo) { ni.Other.RPCAddress = nonASCII }, true},
|
|
{"Empty tab RPCAddress", func(ni *DefaultNodeInfo) { ni.Other.RPCAddress = emptyTab }, true},
|
|
{"Empty space RPCAddress", func(ni *DefaultNodeInfo) { ni.Other.RPCAddress = emptySpace }, true},
|
|
{"Empty RPCAddress", func(ni *DefaultNodeInfo) { ni.Other.RPCAddress = "" }, false},
|
|
{"Good RPCAddress", func(ni *DefaultNodeInfo) { ni.Other.RPCAddress = "0.0.0.0:26657" }, false},
|
|
}
|
|
|
|
nodeKey := GenNodeKey()
|
|
name := "testing"
|
|
|
|
// test case passes
|
|
ni = testNodeInfo(nodeKey.ID, name).(DefaultNodeInfo)
|
|
ni.Channels = channels
|
|
assert.NoError(t, ni.Validate())
|
|
|
|
for _, tc := range testCases {
|
|
ni := testNodeInfo(nodeKey.ID, name).(DefaultNodeInfo)
|
|
ni.Channels = channels
|
|
tc.malleateNodeInfo(&ni)
|
|
err := ni.Validate()
|
|
if tc.expectErr {
|
|
assert.Error(t, err, tc.testName)
|
|
} else {
|
|
assert.NoError(t, err, tc.testName)
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
func TestNodeInfoCompatible(t *testing.T) {
|
|
|
|
nodeKey1 := GenNodeKey()
|
|
nodeKey2 := GenNodeKey()
|
|
name := "testing"
|
|
|
|
var newTestChannel byte = 0x2
|
|
|
|
// test NodeInfo is compatible
|
|
ni1 := testNodeInfo(nodeKey1.ID, name).(DefaultNodeInfo)
|
|
ni2 := testNodeInfo(nodeKey2.ID, name).(DefaultNodeInfo)
|
|
assert.NoError(t, ni1.CompatibleWith(ni2))
|
|
|
|
// add another channel; still compatible
|
|
ni2.Channels = []byte{newTestChannel, testCh}
|
|
assert.NoError(t, ni1.CompatibleWith(ni2))
|
|
|
|
// wrong NodeInfo type is not compatible
|
|
_, netAddr := CreateRoutableAddr()
|
|
ni3 := mockNodeInfo{netAddr}
|
|
assert.Error(t, ni1.CompatibleWith(ni3))
|
|
|
|
testCases := []struct {
|
|
testName string
|
|
malleateNodeInfo func(*DefaultNodeInfo)
|
|
}{
|
|
{"Wrong block version", func(ni *DefaultNodeInfo) { ni.ProtocolVersion.Block++ }},
|
|
{"Wrong network", func(ni *DefaultNodeInfo) { ni.Network += "-wrong" }},
|
|
{"No common channels", func(ni *DefaultNodeInfo) { ni.Channels = []byte{newTestChannel} }},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
ni := testNodeInfo(nodeKey2.ID, name).(DefaultNodeInfo)
|
|
tc.malleateNodeInfo(&ni)
|
|
assert.Error(t, ni1.CompatibleWith(ni))
|
|
}
|
|
}
|