Add cleanup on abort

`close()` is NOT called if CSVWriter errors, only `abort()`. So need to call `close()` within `abort()` to finish cleanup.

diff --git a/src/core.js b/src/core.js
index 03cac7f55db4dbd155ea85a23c728857fba651fd..4e09f5decd14022502465d67048683569d216da4 100644 (file)
--- a/src/core.js
+++ b/src/core.js
@@ -375,6 +375,9 @@ export class CSVWriter extends WritableStream {
         await handle.close();
         status.elapsed = performance.now() - status.start;
         status.done = true;
+      },
+      async abort() {
+        await this.close();
       }
     });
 

diff --git a/test/core.test.js b/test/core.test.js
index 283939767a49f4a69835cced61baea2f54673d8f..a02e4bd95fa6db80ae87f4be854dbb2efd71d0d7 100644 (file)
--- a/test/core.test.js
+++ b/test/core.test.js
@@ -522,14 +522,16 @@ suite('CSVWriter', { concurrency: true }, () => {
   test('flushes progress to disk on uncaught error', { concurrency: true }, async (t) => {
     const temp = await createTempFile();
     t.after(async () => await unlink(temp));
+    const writer = new CSVWriter(temp);
     await rejects(async () => {
       await ReadableStream.from((function* () {
         yield '["header1","header2"]';
         yield '["1","1"]';
         throw new Error();
-      })()).pipeTo(new CSVWriter(temp))
+      })()).pipeTo(writer)
     });
-    await new CSVReader(temp).pipeTo(
+    deepStrictEqual(writer.status.done, true); // Confirm cleanup actions
+    await new CSVReader(temp).pipeTo( // Confirm flush to disk
       csvStreamEqualWritable([
         ['header1', 'header2'],
         ['1', '1'],