mock: fix notif test

jbenet · jbenet · commit e77cda695722 · 2015-06-26T23:57:01.000-07:00
License: MIT
Signed-off-by: Juan Batiz-Benet &lt;juan@benet.ai&gt;
diff --git a/p2p/net/mock/mock_notif_test.go b/p2p/net/mock/mock_notif_test.go
@@ -52,7 +52,7 @@ func TestNotifications(t *testing.T) {
 					case <-time.After(timeout):
 						t.Fatal("timeout")
 					}
-					for _, c2 := range cos {
+					for _, c2 := range s.ConnsToPeer(s2.LocalPeer()) {
 						if c == c2 {
 							t.Log("got notif for conn")
 							return

Original file line number	Diff line number	Diff line change
`@@ -52,7 +52,7 @@ func TestNotifications(t *testing.T) {`
`52`	`52`	`case <-time.After(timeout):`
`53`	`53`	`t.Fatal("timeout")`
`54`	`54`	`}`
`55`		`- for _, c2 := range cos {`
	`55`	`+ for _, c2 := range s.ConnsToPeer(s2.LocalPeer()) {`
`56`	`56`	`if c == c2 {`
`57`	`57`	`t.Log("got notif for conn")`
`58`	`58`	`return`