path: root/whisper/common_test.go

// Contains some common utility functions for testing.

package whisper

import (
    "bytes"
    "fmt"
    "io/ioutil"
    "math/rand"

    "github.com/ethereum/go-ethereum/common"
    "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.
func bufMsgPipe() (*p2p.MsgPipeRW, *p2p.MsgPipeRW) {
    A, midA := p2p.MsgPipe()
    midB, B := p2p.MsgPipe()

    go copyMsgPipe(midA, midB)
    go copyMsgPipe(midB, midA)

    return A, B
}

// copyMsgPipe copies messages from the src pipe to the dest.
func copyMsgPipe(dst, src *p2p.MsgPipeRW) {
    defer dst.Close()
    for {
        msg, err := src.ReadMsg()
        if err != nil {
            return
        }
        data, err := ioutil.ReadAll(msg.Payload)
        if err != nil {
            return
        }
        msg.Payload = bytes.NewReader(data)
        if err := dst.WriteMsg(msg); err != nil {
            return
        }
    }
}