whisper: add basic tests for the whiper peers

1 files changed, 38 insertions, 0 deletions

diff --git a/whisper/common_test.go b/whisper/common_test.go
new file mode 100644
index 000000000..8c12f8aeb
--- /dev/null
+++ b/whisper/common_test.go
@@ -0,0 +1,38 @@
+// Contains some common utility functions for testing.
+
+package whisper
+
+import (
+	"fmt"
+	"math/rand"
+
+	"github.com/ethereum/go-ethereum/p2p"
+
+	"github.com/ethereum/go-ethereum/common"
+	"github.com/ethereum/go-ethereum/p2p/discover"
+)
+
+// randomNodeID generates and returns a random P2P discovery node id for the
+// whisper tests.
+func randomNodeID() (id discover.NodeID) {
+	for i := range id {
+		id[i] = byte(rand.Intn(255))
+	}
+	return id
+}
+
+// randomNodeName generates and returns a random P2P node name for the whisper
+// tests.
+func randomNodeName() string {
+	return common.MakeName(fmt.Sprintf("whisper-go-test-%3d", rand.Intn(999)), "1.0")
+}
+
+// whisperCaps returns the node capabilities for running the whisper sub-protocol.
+func whisperCaps() []p2p.Cap {
+	return []p2p.Cap{
+		p2p.Cap{
+			Name:    protocolName,
+			Version: uint(protocolVersion),
+		},
+	}
+}