pausing-transform 1.0.0 → 1.0.2

package/.vscode/launch.json

@@ -0,0 +1,17 @@
+{
+	// Use IntelliSense to learn about possible attributes.
+	// Hover to view descriptions of existing attributes.
+	// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+	"version": "0.2.0",
+	"configurations": [
+		{
+			"name": "Attach",
+			"port": 9229,
+			"request": "attach",
+			"skipFiles": [
+				"<node_internals>/**"
+			],
+			"type": "node"
+		}
+	]
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pausing-transform",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "description": "A transform stream that will buffer data written until flushed.",
   "main": "pausing-transform.js",
   "scripts": {

package/pausing-transform.js

@@ -8,13 +8,21 @@ class PausingTransform extends Transform {
 		this.transformer = transformer
 	}
 	_transform(chunk, encoding, callback) {
-		
 		return callback(null, this.transformer ? this.transformer(chunk.toString()) : chunk)
 	}
 	_flush(callback) {
-		let ret = super.write(this.pausedData, null, callback)
-		this.pausedData = ''
-		return ret
+		if(this.pausedData) {
+			let p = this.pausedData
+			this.pausedData = ''
+			let ret = super.write(p, null, () => {
+				if(callback) {
+					return callback()
+				}
+			})
+		}
+		else if(callback) {
+			callback()
+		}
 	}
 	write(chunk, encoding, callback) {
 		if(this.paused) {
@@ -34,6 +42,16 @@ class PausingTransform extends Transform {
 		this.paused = false
 		this._flush()
 	}
+	end() {
+		if(this.pausedData) {
+			this._flush(() => {
+				super.end()
+			})
+		}
+		else {
+			super.end()
+		}
+	}
 }
 
-module.exports = PausingTransform
+module.exports = PausingTransform

package/test/basics.mjs

@@ -0,0 +1,70 @@
+import test from 'node:test';
+import assert from 'node:assert'
+import wait from './wait.mjs';
+
+import PausingTransform from '../pausing-transform.js'
+
+test("basics", async (t) => {
+
+	await t.test('paused and run', async (t) => {
+		let final = new PausingTransform()
+		let head = new PausingTransform()
+		
+		head.pipe(final)
+		
+		head.write('hello')
+		
+		assert.equal(final.pausedData, '', 'Paused data should be empty.')
+		assert.equal(head.pausedData, 'hello', 'Paused data should be empty.')
+		
+		head.run()
+		await wait(300)
+		assert.equal(head.pausedData, '', 'Paused data should be empty.')
+		assert.equal(final.pausedData, 'hello', 'Paused data should be empty.')
+		
+
+	})
+
+	await t.test('transform', async (t) => {
+		let final = new PausingTransform()
+		let head = new PausingTransform(msg => msg.toUpperCase())
+		
+		head.pipe(final)
+		
+		head.write('hello')
+		
+		assert.equal(final.pausedData, '', 'Paused data should be empty.')
+		assert.equal(head.pausedData, 'hello', 'Paused data should be empty.')
+		
+		head.run()
+		await wait(300)
+		assert.equal(head.pausedData, '', 'Paused data should be empty.')
+		assert.equal(final.pausedData, 'HELLO', 'Paused data should be empty.')
+	})
+
+	await t.test('unpipe', async (t) => {
+		let final = new PausingTransform()
+		let head = new PausingTransform(msg => msg.toUpperCase())
+		
+		head.pipe(final)
+		
+		head.write('hello')
+		
+		assert.equal(final.pausedData, '', 'Paused data should be empty.')
+		assert.equal(head.pausedData, 'hello', 'Final data does not match.')
+		
+		head.run()
+		await wait(300)
+		assert.equal(head.pausedData, '', 'Paused data should be empty.')
+		assert.equal(final.pausedData, 'HELLO', 'Final data does not match.')
+		
+		head.unpipe()
+		await wait(300)
+		head.run()
+		head.pipe(final)
+		
+		head.write('there')
+		await wait(300)
+		assert.equal(final.pausedData, 'HELLOTHERE', 'Final data does not match.')
+	})
+})

package/test/wait.mjs

@@ -0,0 +1,6 @@
+export default async function wait(millis) {
+	let pr = new Promise((resolve, reject) => {
+		setTimeout(resolve, millis)
+	})
+	return pr
+}