Merge pull request #2 from xicombd/echo-tests

Add echo tests

src/index.js

@@ -17,10 +17,10 @@ function WebSockets () {
       options = {}
     }
 
-    options.ready = options.ready || function noop () {}
+    options.connect = options.connect || function noop () {}
     const maOpts = multiaddr.toOptions()
     const conn = new SWS('ws://' + maOpts.host + ':' + maOpts.port)
-    conn.on('ready', options.ready)
+    conn.on('connect', options.connect)
     conn.getObservedAddrs = () => {
       return [multiaddr]
     }

tests/libp2p-websockets-test.js

@@ -63,4 +63,45 @@ describe('libp2p-websockets', function () {
     expect(valid[0]).to.deep.equal(mh3)
     done()
   })
+
+  it('echo', (done) => {
+    const mh = multiaddr('/ip4/127.0.0.1/tcp/9090/websockets')
+    ws.createListener(mh, (conn) => {
+      conn.pipe(conn)
+    }, () => {
+      const conn = ws.dial(mh)
+      const message = 'Hello World!'
+      conn.write(message)
+      conn.on('data', (data) => {
+        expect(data.toString()).to.equal(message)
+        conn.end()
+        ws.close(() => {
+          done()
+        })
+      })
+    })
+  })
+
+  it('echo with connect event and send', (done) => {
+    const mh = multiaddr('/ip4/127.0.0.1/tcp/9090/websockets')
+    ws.createListener(mh, (conn) => {
+      conn.pipe(conn)
+    }, () => {
+      const message = 'Hello World!'
+
+      const conn = ws.dial(mh, {
+        connect: () => {
+          conn.send(message)
+        }
+      })
+
+      conn.on('data', (data) => {
+        expect(data.toString()).to.equal(message)
+        conn.end()
+        ws.close(() => {
+          done()
+        })
+      })
+    })
+  })
 })