From b1e716368968617aed2c50592b7331274bc68389 Mon Sep 17 00:00:00 2001 From: Anton Kaliaev Date: Wed, 8 Nov 2017 13:12:48 -0500 Subject: [PATCH] rewrite node test to use new pubsub --- node/node_test.go | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/node/node_test.go b/node/node_test.go index 389dd1d2..01099459 100644 --- a/node/node_test.go +++ b/node/node_test.go @@ -1,6 +1,7 @@ package node import ( + "context" "testing" "time" @@ -22,10 +23,9 @@ func TestNodeStartStop(t *testing.T) { t.Logf("Started node %v", n.sw.NodeInfo()) // wait for the node to produce a block - blockCh := make(chan struct{}) - types.AddListenerForEvent(n.EventSwitch(), "node_test", types.EventStringNewBlock(), func(types.TMEventData) { - blockCh <- struct{}{} - }) + blockCh := make(chan interface{}) + err = n.EventBus().Subscribe(context.Background(), "node_test", types.EventQueryNewBlock, blockCh) + assert.NoError(t, err) select { case <-blockCh: case <-time.After(5 * time.Second):