whisper/common_test.go


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

// Contains some common utility functions for testing.

package whisper

import (
    "bytes"
    "io/ioutil"

    "github.com/ethereum/go-ethereum/p2p"
)

// 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
        }
    }
}