whisper: mock tests to use simulated peers

1 files changed, 32 insertions, 0 deletions

diff --git a/whisper/common_test.go b/whisper/common_test.go
index 8c12f8aeb..4e221d6b1 100644
--- a/whisper/common_test.go
+++ b/whisper/common_test.go
@@ -3,7 +3,9 @@
 package whisper
 
 import (
+	"bytes"
 	"fmt"
+	"io/ioutil"
 	"math/rand"
 
 	"github.com/ethereum/go-ethereum/p2p"
@@ -36,3 +38,33 @@ func whisperCaps() []p2p.Cap {
 		},
 	}
 }
+
+// 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
+		}
+	}
+}