You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
786 B
35 lines
786 B
6 years ago
|
package host
|
||
|
|
|
||
|
|
import (
|
||
|
|
"reflect"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/harmony-one/harmony/p2p"
|
||
|
|
"github.com/harmony-one/harmony/p2p/host/hostv2"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestSendMessage(test *testing.T) {
|
||
|
|
peer1 := p2p.Peer{IP: "127.0.0.1", Port: "9000"}
|
||
|
|
peer2 := p2p.Peer{IP: "127.0.0.1", Port: "9001"}
|
||
|
|
msg := []byte{0x00, 0x01, 0x02, 0x03, 0x04}
|
||
|
|
host1 := hostv2.New(peer1)
|
||
|
|
host2 := hostv2.New(peer2)
|
||
|
|
go host2.BindHandlerAndServe(handler)
|
||
|
|
SendMessage(host1, peer2, msg, nil)
|
||
|
|
time.Sleep(3 * time.Second)
|
||
|
|
}
|
||
|
|
|
||
|
|
func handler(s p2p.Stream) {
|
||
|
|
defer s.Close()
|
||
|
|
content, err := p2p.ReadMessageContent(s)
|
||
|
|
if err != nil {
|
||
|
|
panic("Read p2p data failed")
|
||
|
|
}
|
||
|
|
golden := []byte{0x00, 0x01, 0x02, 0x03, 0x04}
|
||
|
|
|
||
|
|
if !reflect.DeepEqual(content, golden) {
|
||
|
|
panic("received message not equal original message")
|
||
|
|
}
|
||
|
|
}
|