gaspersic/go-ethereum
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

whisper: push work in progress for bug report

Browse Source
This commit is contained in:
Péter Szilágyi
2015-04-13 16:19:34 +03:00
parent 89358d25a4
commit cb707ba50c
2 changed files with 19 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
whisper/whisper_test.go
View File

@ -7,6 +7,7 @@ import (
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/crypto"
"github.com/ethereum/go-ethereum/logger/glog"
"github.com/ethereum/go-ethereum/p2p"
"github.com/ethereum/go-ethereum/p2p/nat"
)
@ -17,6 +18,7 @@ type testNode struct {
}
func startNodes(n int) ([]*testNode, error) {
// Start up the cluster of nodes
cluster := make([]*testNode, 0, n)
for i := 0; i < n; i++ {
shh := New()
@ -46,6 +48,11 @@ func startNodes(n int) ([]*testNode, error) {
client: shh,
})
}
// Manually wire together the cluster nodes
root := cluster[0].server.Self()
for _, node := range cluster[1:] {
node.server.SuggestPeer(root)
}
return cluster, nil
}
@ -56,6 +63,7 @@ func stopNodes(cluster []*testNode) {
}
func TestSelfMessage(t *testing.T) {
// Start the single node cluster
cluster, err := startNodes(1)
if err != nil {
t.Fatalf("failed to boot test cluster: %v", err)
@ -96,6 +104,10 @@ func TestSelfMessage(t *testing.T) {
}
func TestDirectMessage(t *testing.T) {
glog.SetV(6)
glog.SetToStderr(true)
// Start the sender-recipient cluster
cluster, err := startNodes(2)
if err != nil {
t.Fatalf("failed to boot test cluster: %v", err)