From 255090c5c0ac0a4caa02a5468246f509f63d4e84 Mon Sep 17 00:00:00 2001 From: Darien Raymond Date: Tue, 3 Jul 2018 21:53:36 +0200 Subject: [PATCH] fix dispatcher test --- transport/internet/udp/dispatcher_test.go | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/transport/internet/udp/dispatcher_test.go b/transport/internet/udp/dispatcher_test.go index 2572c980..d642e36e 100644 --- a/transport/internet/udp/dispatcher_test.go +++ b/transport/internet/udp/dispatcher_test.go @@ -59,13 +59,15 @@ func TestSameDestinationDispatching(t *testing.T) { b := buf.New() b.AppendBytes('a', 'b', 'c', 'd') - dispatcher := NewDispatcher(td) + var msgCount uint32 - dispatcher.Dispatch(ctx, dest, b, func(payload *buf.Buffer) { + dispatcher := NewDispatcher(td, func(ctx context.Context, payload *buf.Buffer) { atomic.AddUint32(&msgCount, 1) }) + + dispatcher.Dispatch(ctx, dest, b) for i := 0; i < 5; i++ { - dispatcher.Dispatch(ctx, dest, b, func(payload *buf.Buffer) {}) + dispatcher.Dispatch(ctx, dest, b) } time.Sleep(time.Second)