grenache-nodejs-http 1.0.2 → 2.0.0

package/test/peer-rpc-server-secure-buffered.js

@@ -1,236 +1,189 @@
-/* eslint-env mocha */
+'use strict'
 
-const assert = require('assert')
+const test = require('brittle')
 const { PassThrough } = require('stream')
 const fs = require('fs')
 const path = require('path')
-const { PeerRPCClient, PeerRPCServer } = require('../')
-const Link = require('grenache-nodejs-link')
-const setupHooks = require('./before-after.js')
-
-const PORT = 1337
-let link, peer, peerSrvBuf, serviceBuf, stop
-describe('RPC integration', () => {
-  setupHooks((grapes) => {
-    link = new Link({
-      grape: 'http://127.0.0.1:30001'
-    })
-    link.start()
-
-    peer = new PeerRPCClient(link, {
-      secure: {
-        key: fs.readFileSync(path.join(__dirname, './certs/client1-key.pem')),
-        cert: fs.readFileSync(path.join(__dirname, './certs/client1-crt.pem')),
-        ca: fs.readFileSync(path.join(__dirname, './certs/ca-crt.pem')),
-        rejectUnauthorized: false // take care, can be dangerous in production!
-      }
-    })
-    peer.init()
-
-    peerSrvBuf = new PeerRPCServer(link, {
-      secure: {
-        key: fs.readFileSync(path.join(__dirname, './certs/server-key.pem')),
-        cert: fs.readFileSync(path.join(__dirname, './certs/server-crt.pem')),
-        ca: fs.readFileSync(path.join(__dirname, './certs/ca-crt.pem')),
-        requestCert: true,
-        rejectUnauthorized: false // take care, can be dangerous in production!
-      },
-      timeout: 300000
-    })
+const { rpcTestCase } = require('./before-after')
 
-    peerSrvBuf.init()
-    serviceBuf = peerSrvBuf.transport('buffered')
-    serviceBuf.listen(PORT)
+const rpcTest = rpcTestCase({ transport: 'buffered', isSecure: true })
 
-    link.announce('rpc_buf', serviceBuf.port, {}, (err, res) => {
-      if (err) throw Error('error in announce, setup')
-    })
+test('RPC integration', async (t) => {
+  await t.test('buffered parsing works, objects', { timeout: 7000 }, rpcTest(async ({ t, service, peer }) => {
+    service.on('request', (rid, key, payload, handler, cert, meta) => {
+      t.ok(typeof rid === 'string')
+      t.is(key, 'rpc_test')
+      t.alike(payload, { hello: 'world' })
 
-    stop = () => {
-      peer.stop()
-      link.stop()
-      serviceBuf.stop()
-    }
-  })
-
-  it('buffered parsing works, objects', (done) => {
-    serviceBuf.on('request', (rid, key, payload, handler, cert, meta) => {
-      assert.ok(typeof rid === 'string')
-      assert.strictEqual(key, 'rpc_buf')
-      assert.deepStrictEqual(payload, { hello: 'world' })
-
-      assert.strictEqual(meta.isStream, false)
+      t.is(meta.isStream, false)
 
       handler.reply(null, { hello: 'helloworld' })
     })
 
     const opts = { timeout: 100000 }
-    peer.request('rpc_buf', { hello: 'world' }, opts, (err, result) => {
-      if (err) throw err
-
-      assert.deepStrictEqual(result, { hello: 'helloworld' })
-
-      stop()
-      done()
+    await new Promise(resolve => {
+      peer.request('rpc_test', { hello: 'world' }, opts, (err, result) => {
+        if (err) throw err
+        t.alike(result, { hello: 'helloworld' })
+        resolve()
+      })
     })
-  }).timeout(7000)
+  }))
 
-  it('buffered parsing works, strings', (done) => {
-    serviceBuf.on('request', (rid, key, payload, handler, cert, meta) => {
-      assert.ok(typeof rid === 'string')
-      assert.strictEqual(key, 'rpc_buf')
-      assert.strictEqual(payload, 'hello')
+  await t.test('buffered parsing works, strings', { timeout: 7000 }, rpcTest(async ({ t, service, peer }) => {
+    service.on('request', (rid, key, payload, handler, cert, meta) => {
+      t.ok(typeof rid === 'string')
+      t.is(key, 'rpc_test')
+      t.is(payload, 'hello')
 
-      assert.strictEqual(meta.isStream, false)
+      t.is(meta.isStream, false)
 
       handler.reply(null, 'world')
     })
 
     const opts = { timeout: 100000 }
-    peer.request('rpc_buf', 'hello', opts, (err, result) => {
-      if (err) throw err
-
-      assert.strictEqual(result, 'world')
-
-      stop()
-      done()
+    await new Promise(resolve => {
+      peer.request('rpc_test', 'hello', opts, (err, result) => {
+        if (err) throw err
+        t.is(result, 'world')
+        resolve()
+      })
     })
-  }).timeout(7000)
+  }))
 
-  it('buffered parsing works, compressed strings', (done) => {
-    serviceBuf.on('request', (rid, key, payload, handler, cert, meta) => {
-      assert.ok(typeof rid === 'string')
-      assert.strictEqual(key, 'rpc_buf')
-      assert.strictEqual(payload, 'hello')
+  await t.test('buffered parsing works, compressed strings', { timeout: 7000 }, rpcTest(async ({ t, service, peer }) => {
+    service.on('request', (rid, key, payload, handler, cert, meta) => {
+      t.ok(typeof rid === 'string')
+      t.is(key, 'rpc_test')
+      t.is(payload, 'hello')
 
-      assert.strictEqual(meta.isStream, false)
-      assert.strictEqual(meta.compress, true)
+      t.is(meta.isStream, false)
+      t.is(meta.compress, true)
 
       handler.reply(null, 'world')
     })
 
     const opts = { timeout: 100000, compress: true }
-    peer.request('rpc_buf', 'hello', opts, (err, result) => {
-      if (err) throw err
-
-      assert.strictEqual(result, 'world')
-
-      stop()
-      done()
+    await new Promise(resolve => {
+      peer.request('rpc_test', 'hello', opts, (err, result) => {
+        if (err) throw err
+        t.is(result, 'world')
+        resolve()
+      })
     })
-  }).timeout(7000)
-
-  it('buffered parsing works, strings from streaming client', (done) => {
-    serviceBuf.on('request', (rid, key, payload, handler, cert, meta) => {
-      assert.ok(typeof rid === 'string')
-      assert.strictEqual(key, 'rpc_buf')
-      assert.strictEqual(payload, 'hello')
-
-      assert.strictEqual(meta.isStream, true)
+  }))
 
+  await t.test('buffered parsing works, strings from streaming client', { timeout: 7000 }, rpcTest(async ({ t, service, peer }) => {
+    service.on('request', (rid, key, payload, handler, cert, meta) => {
+      t.ok(typeof rid === 'string')
+      t.is(key, 'rpc_test')
+      t.is(payload, 'hello')
+      t.is(meta.isStream, true)
       handler.reply(null, 'world')
     })
 
-    const req = peer.stream('rpc_buf', {
+    const req = peer.stream('rpc_test', {
       timeout: 10000
     })
 
-    req
-      .on('data', (data) => {
-        const d = JSON.parse(data.toString())
-        // ["65c5737f-bbce-40f9-b48a-af8bd8869d66",null,"world"]
-        assert.strictEqual(d[1], null)
-        assert.strictEqual(d[2], 'world')
-        stop()
-        done()
-      })
-      .on('end', () => {})
-      .on('error', (e) => { console.error(e) })
+    const p = new Promise(resolve => {
+      req
+        .on('data', (data) => {
+          const d = JSON.parse(data.toString())
+          // ["65c5737f-bbce-40f9-b48a-af8bd8869d66",null,"world"]
+          t.is(d[1], null)
+          t.is(d[2], 'world')
+          resolve()
+        })
+        .on('end', () => {})
+        .on('error', (e) => { console.error(e) })
+    })
 
     const writable = new PassThrough()
-    writable.write('["UUID", "rpc_buf", "hello"]')
+    writable.write('["UUID", "rpc_test", "hello"]')
     writable.end()
     writable.pipe(req)
-  }).timeout(7000)
-
-  it('buffered parsing works, objects from streaming client', (done) => {
-    serviceBuf.on('request', (rid, key, payload, handler, cert, meta) => {
-      assert.ok(typeof rid === 'string')
-      assert.strictEqual(key, 'rpc_buf')
-      assert.deepStrictEqual(payload, { hello: 'world' })
 
-      assert.strictEqual(meta.isStream, true)
+    await p
+  }))
 
+  await t.test('buffered parsing works, objects from streaming client', { timeout: 7000 }, rpcTest(async ({ t, service, peer }) => {
+    service.on('request', (rid, key, payload, handler, cert, meta) => {
+      t.ok(typeof rid === 'string')
+      t.is(key, 'rpc_test')
+      t.alike(payload, { hello: 'world' })
+      t.is(meta.isStream, true)
       handler.reply(null, { hello: 'helloworld' })
     })
 
-    const req = peer.stream('rpc_buf', {
+    const req = peer.stream('rpc_test', {
       timeout: 10000
     })
 
-    req
-      .on('data', (data) => {
-        const d = JSON.parse(data.toString())
-        assert.strictEqual(d[1], null)
-        assert.deepStrictEqual(d[2], { hello: 'helloworld' })
-        stop()
-        done()
-      })
-      .on('end', () => {})
-      .on('error', (e) => { console.error(e) })
+    const p = new Promise(resolve => {
+      req
+        .on('data', (data) => {
+          const d = JSON.parse(data.toString())
+          t.is(d[1], null)
+          t.alike(d[2], { hello: 'helloworld' })
+          resolve()
+        })
+        .on('end', () => {})
+        .on('error', (e) => { console.error(e) })
+    })
 
     const writable = new PassThrough()
-    writable.write('["UUID", "rpc_buf", { "hello": "world" }]')
+    writable.write('["UUID", "rpc_test", { "hello": "world" }]')
     writable.end()
     writable.pipe(req)
-  }).timeout(7000)
-
-  it('buffered parsing works, extra headers', (done) => {
-    serviceBuf.on('request', (rid, key, payload, handler, cert, meta) => {
-      assert.ok(typeof rid === 'string')
-      assert.strictEqual(key, 'rpc_buf')
-      assert.deepStrictEqual(payload, { hello: 'world' })
-
-      assert.strictEqual(meta.action, 'uploadPublic')
-      assert.deepStrictEqual(meta.args, { foo: 'bar' })
-      assert.strictEqual(meta.isStream, true)
 
+    await p
+  }))
+
+  await t.test('buffered parsing works, extra headers', { timeout: 7000 }, rpcTest(async ({ t, service, peer }) => {
+    service.on('request', (rid, key, payload, handler, cert, meta) => {
+      t.ok(typeof rid === 'string')
+      t.is(key, 'rpc_test')
+      t.alike(payload, { hello: 'world' })
+      t.is(meta.action, 'uploadPublic')
+      t.alike(meta.args, { foo: 'bar' })
+      t.is(meta.isStream, true)
       handler.reply(null, { hello: 'helloworld' })
     })
 
-    const req = peer.stream('rpc_buf', {
+    const req = peer.stream('rpc_test', {
       timeout: 10000,
       headers: { _a: 'uploadPublic', _ar: { foo: 'bar' } }
     })
 
-    req
-      .on('data', (data) => {
-        const d = JSON.parse(data.toString())
-        assert.strictEqual(d[1], null)
-        assert.deepStrictEqual(d[2], { hello: 'helloworld' })
-        stop()
-        done()
-      })
-      .on('end', () => {})
-      .on('error', (e) => { console.error(e) })
+    const p = new Promise(resolve => {
+      req
+        .on('data', (data) => {
+          const d = JSON.parse(data.toString())
+          t.is(d[1], null)
+          t.alike(d[2], { hello: 'helloworld' })
+          resolve()
+        })
+        .on('end', () => {})
+        .on('error', (e) => { console.error(e) })
+    })
 
     const writable = new PassThrough()
-    writable.write('["UUID", "rpc_buf", { "hello": "world" }]')
+    writable.write('["UUID", "rpc_test", { "hello": "world" }]')
     writable.end()
     writable.pipe(req)
-  }).timeout(7000)
-
-  it('buf/buf: serialized files work', (done) => {
-    serviceBuf.on('request', (rid, key, payload, handler, cert, meta) => {
-      assert.ok(typeof rid === 'string')
-      assert.strictEqual(key, 'rpc_buf')
-
-      assert.strictEqual(payload.action, 'uploadPublic')
-      assert.strictEqual(payload.args[1].key, 'example-amazon-dynamo-sosp2007-4.pdf')
-      assert.strictEqual(typeof payload.args[0], 'string')
-
-      assert.strictEqual(meta.isStream, false)
 
+    await p
+  }))
+
+  await t.test('buf/buf: serialized files work', { timeout: 7000 }, rpcTest(async ({ t, service, peer }) => {
+    service.on('request', (rid, key, payload, handler, cert, meta) => {
+      t.ok(typeof rid === 'string')
+      t.is(key, 'rpc_test')
+      t.is(payload.action, 'uploadPublic')
+      t.is(payload.args[1].key, 'example-amazon-dynamo-sosp2007-4.pdf')
+      t.is(typeof payload.args[0], 'string')
+      t.is(meta.isStream, false)
       handler.reply(null, 'ok')
     })
 
@@ -250,13 +203,12 @@ describe('RPC integration', () => {
       action: 'uploadPublic',
       args: [buf.toString('hex'), optsPdf]
     }
-    peer.request('rpc_buf', queryUploadPublicPdf, opts, (err, result) => {
-      if (err) throw err
-
-      assert.deepStrictEqual(result, 'ok')
-
-      stop()
-      done()
+    await new Promise(resolve => {
+      peer.request('rpc_test', queryUploadPublicPdf, opts, (err, result) => {
+        if (err) throw err
+        t.alike(result, 'ok')
+        resolve()
+      })
     })
-  }).timeout(7000)
+  }))
 })