improve binary data sending

libp2p · Aug 30, 2016 · d6c6aea · d6c6aea
1 parent 4d4fa5a
commit d6c6aea
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/src/index.js b/src/index.js
@@ -24,7 +24,10 @@ module.exports = class WebSockets {
 
     const url = `ws://${maOpts.host}:${maOpts.port}`
     log('dialing %s', url)
-    const socket = connect(url, cb)
+    const socket = connect(url, {
+      binary: true,
+      onConnect: cb
+    })
     socket.getObservedAddrs = (cb) => {
       return cb(null, [ma])
     }

diff --git a/test/browser.js b/test/browser.js
@@ -40,10 +40,10 @@ describe('libp2p-websockets', () => {
       const message = rawMessage.toString('hex')
 
       const s = goodbye({
-        source: pull.values([message]),
+        source: pull.values([rawMessage]),
         sink: pull.collect((err, results) => {
           expect(err).to.not.exist
-          expect(results).to.be.eql([message])
+          expect(results).to.be.eql([rawMessage])
           done()
         })
       })
@@ -55,7 +55,7 @@ describe('libp2p-websockets', () => {
         source: pull(
           pull.infinite(),
           pull.take(1000),
-          pull.map((val) => val.toString())
+          pull.map((val) => Buffer(val.toString()))
         ),
         sink: pull.collect((err, result) => {
           expect(err).to.not.exist