fix to get test passing

README.md

@@ -11,9 +11,10 @@ Broadcast a message regularly.
   - `opts.interval` interval between packets in ms, default 1000
   - `opts.port` port to broadcast on.
 
-### `bc.receive(callback [opts])`
+### `const destroy = bc.receive(callback [opts])`
 
-Listen for messages
+Listen for messages.
+- `destroy` is a function which you can call with no arguments to stop listening.
 - `callback` is called whenever a ***unique*** message is found.  That is, whenever the message or address changes. The message is passed in the callback.
 - `opts` may include:
   - `opts.port` port to listen to

index.js

@@ -41,4 +41,6 @@ function receive (callback, opts = {}) {
       callback(msg)
     }
   })
+
+  return () => { stream.end() }
 }

test/index.test.js

@@ -1,18 +1,18 @@
 const { describe } = require('tape-plus')
 const { broadcast, receive } = require('..')
+const message = Buffer.from('boop')
 
 describe('basic', (context) => {
   context('broadcast and receive', (assert, next) => {
-    const destroy = broadcast(null, { interval: 1000 })
-
+    const destroyBroadcast = broadcast(message, { interval: 1000 })
+    const destroyReceive = receive((msg) => {
+      assert.same(msg, message, 'received the message that was broadcast')
+    })
 
-    console.log(destroy)
     setTimeout(() => {
-      destroy()
+      destroyBroadcast()
+      destroyReceive()
       next()
     }, 2000)
-    receive((msg) => {
-      console.log(msg)
-    })
   })
 })