whisper: remove some unneeded testing complexity

pull/718/head
Péter Szilágyi 10 years ago
parent bcf41797ca
commit 46ea193a49
  1. 29
      whisper/common_test.go
  2. 5
      whisper/peer_test.go
  3. 3
      whisper/whisper_test.go

@ -4,40 +4,11 @@ package whisper
import ( import (
"bytes" "bytes"
"fmt"
"io/ioutil" "io/ioutil"
"math/rand"
"github.com/ethereum/go-ethereum/common"
"github.com/ethereum/go-ethereum/p2p" "github.com/ethereum/go-ethereum/p2p"
"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),
},
}
}
// bufMsgPipe creates a buffered message pipe between two endpoints. // bufMsgPipe creates a buffered message pipe between two endpoints.
func bufMsgPipe() (*p2p.MsgPipeRW, *p2p.MsgPipeRW) { func bufMsgPipe() (*p2p.MsgPipeRW, *p2p.MsgPipeRW) {
A, midA := p2p.MsgPipe() A, midA := p2p.MsgPipe()

@ -5,6 +5,7 @@ import (
"time" "time"
"github.com/ethereum/go-ethereum/p2p" "github.com/ethereum/go-ethereum/p2p"
"github.com/ethereum/go-ethereum/p2p/discover"
) )
type testPeer struct { type testPeer struct {
@ -15,7 +16,7 @@ type testPeer struct {
func startTestPeer() *testPeer { func startTestPeer() *testPeer {
// Create a simulated P2P remote peer and data streams to it // Create a simulated P2P remote peer and data streams to it
remote := p2p.NewPeer(randomNodeID(), randomNodeName(), whisperCaps()) remote := p2p.NewPeer(discover.NodeID{}, "", nil)
tester, tested := p2p.MsgPipe() tester, tested := p2p.MsgPipe()
// Create a whisper client and connect with it to the tester peer // Create a whisper client and connect with it to the tester peer
@ -30,7 +31,7 @@ func startTestPeer() *testPeer {
client.handlePeer(remote, tested) client.handlePeer(remote, tested)
}() }()
// Assemble and return the test peer
return &testPeer{ return &testPeer{
client: client, client: client,
stream: tester, stream: tester,

@ -5,13 +5,14 @@ import (
"time" "time"
"github.com/ethereum/go-ethereum/p2p" "github.com/ethereum/go-ethereum/p2p"
"github.com/ethereum/go-ethereum/p2p/discover"
) )
func startTestCluster(n int) []*Whisper { func startTestCluster(n int) []*Whisper {
// Create the batch of simulated peers // Create the batch of simulated peers
nodes := make([]*p2p.Peer, n) nodes := make([]*p2p.Peer, n)
for i := 0; i < n; i++ { for i := 0; i < n; i++ {
nodes[i] = p2p.NewPeer(randomNodeID(), randomNodeName(), whisperCaps()) nodes[i] = p2p.NewPeer(discover.NodeID{}, "", nil)
} }
whispers := make([]*Whisper, n) whispers := make([]*Whisper, n)
for i := 0; i < n; i++ { for i := 0; i < n; i++ {

Loading…
Cancel
Save