Fix EventSwitch usage in WaitForOneEvent

2017-02-27 14:55:29 +01:00 · 2017-02-27 14:55:29 +01:00 · 21501815dd
parent d56cb2ab4b
commit 21501815dd
1 changed files with 2 additions and 3 deletions
--- a/rpc/client/helpers.go
+++ b/rpc/client/helpers.go
@ -72,9 +72,8 @@ func WaitForOneEvent(evsw types.EventSwitch,
 		evts <- data
 	})
 	// make sure to unregister after the test is over
-	// TODO: why doesn't the other call work???
-	// defer evsw.RemoveListenerForEvent(listener, evtTyp)
-	defer evsw.RemoveListener(listener)
+	defer evsw.RemoveListenerForEvent(evtTyp, listener)
+	// defer evsw.RemoveListener(listener)  // this also works

 	select {
 	case <-quit: