fix(net:swarm) test... use netmsg interface
Brian Tiger Chow committed
Sep 22, 2014 at 04:27 UTC
7cac1ef93d1f273a484a82e683881b465595f189
1 file changed
+4
-4
net/swarm/swarm_test.go
+4
-4
@@ -115,7 +115,7 @@ func TestSwarm(t *testing.T) {
115
MsgNum := 1000
116
for k := 0; k < MsgNum; k++ {
117
for _, p := range peers {
118
- swarm.Outgoing <- &msg.Message{Peer: p, Data: []byte("ping")}
118
+ swarm.Outgoing <- msg.New(p, []byte("ping"))
119
}
120
}
121
@@ -123,12 +123,12 @@ func TestSwarm(t *testing.T) {
123
124
for k := 0; k < (MsgNum * len(peers)); k++ {
125
msg := <-swarm.Incoming
126
- if string(msg.Data) != "pong" {
126
+ if string(msg.Data()) != "pong" {
127
t.Error("unexpected conn output", msg.Data)
128
}
129
130
- n, _ := got[msg.Peer.Key()]
131
- got[msg.Peer.Key()] = n + 1
130
+ n, _ := got[msg.Peer().Key()]
131
+ got[msg.Peer().Key()] = n + 1
132
}
133
134
if len(peers) != len(got) {