Updates

messaging-streams/hyper-p2p-stream-transform/test/test.js

@@ -2,25 +2,30 @@ require('bare-process/global')
 const test = require('brittle')
 const { HyperP2PStreamTransform, PROTOCOL } = require('../index.js')
 
-test('exports', (t) => {
-  t.ok(HyperP2PStreamTransform)
-  t.ok(PROTOCOL)
+test('exports', (t) => { t.ok(HyperP2PStreamTransform); t.is(PROTOCOL, 'stream-transform/v1') })
+
+test('transform chunks', async (t) => {
+  const m = new HyperP2PStreamTransform({ transform: (buf) => require('b4a').from(buf.toString().toUpperCase()) })
+  m.write(require('b4a').from('hi'))
+  t.is(m.read().toString(), 'HI')
+  await m.close()
 })
 
-test('basic operation', async (t) => {
+test('setTransform', async (t) => {
   const m = new HyperP2PStreamTransform()
-  m.write('hi'); t.ok(m.read())
+  m.setTransform((b) => b)
+  t.ok(m.write('x'))
   await m.close()
 })
 
 test('validation', async (t) => {
   const m = new HyperP2PStreamTransform()
-  try { m.write(null) } catch (e) { t.ok(e) }
+  try { m.setTransform(null) } catch (e) { t.ok(e) }
   await m.close()
 })
 
 test('getStats', async (t) => {
   const m = new HyperP2PStreamTransform()
-  t.ok(m.getStats().protocol)
+  t.is(m.getStats().protocol, 'stream-transform/v1')
   await m.close()
 })