test: fix more tests

dad1x · Oct 16, 2018 · 4cc1fff · 4cc1fff
1 parent 52af003
commit 4cc1fff
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/examples/cluster_benchmark/main.go b/examples/cluster_benchmark/main.go
@@ -16,6 +16,7 @@ import (
 	"github.com/perlin-network/noise/network"
 	"github.com/perlin-network/noise/network/backoff"
 	"github.com/perlin-network/noise/network/discovery"
+	"github.com/perlin-network/noise/types/opcode"
 )
 
 const MESSAGE_THRESHOLD uint64 = 2000
@@ -96,6 +97,7 @@ func main() {
 	log.Info().Str("private_key", keys.PrivateKeyHex()).Msg("")
 	log.Info().Str("public_key", keys.PublicKeyHex()).Msg("")
 
+	opcode.RegisterMessageType(opcode.Opcode(1000), &messages.Empty{})
 	builder := network.NewBuilder()
 	builder.SetKeys(keys)
 	builder.SetAddress(network.FormatAddress(protocol, host, port))

diff --git a/examples/topologies/topologies.go b/examples/topologies/topologies.go
@@ -9,6 +9,7 @@ import (
 	"github.com/perlin-network/noise/crypto/ed25519"
 	"github.com/perlin-network/noise/examples/topologies/messages"
 	"github.com/perlin-network/noise/network"
+	"github.com/perlin-network/noise/types/opcode"
 )
 
 const host = "127.0.0.1"
@@ -182,6 +183,7 @@ func setupTreeNodes(startPort int) ([]int, map[string]map[string]struct{}) {
 func setupNodes(ports []int) ([]*network.Network, []*MockPlugin, error) {
 	var nodes []*network.Network
 	var plugins []*MockPlugin
+	opcode.RegisterMessageType(opcode.Opcode(1000), &messages.BasicMessage{})
 
 	for i, port := range ports {
 		builder := network.NewBuilder()