Handle undefined in buffer

guardian · Jan 18, 2024 · 3529b72 · 3529b72
1 parent ef2e6eb
commit 3529b72
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/scrooge-generator-typescript/src/test/resources/decode-encode/decode-encode.ts b/scrooge-generator-typescript/src/test/resources/decode-encode/decode-encode.ts
@@ -41,7 +41,11 @@ async function decode(buffer: Buffer): Promise<School> {
 async function encode(school: School): Promise<Buffer> {
     return new Promise((resolve, reject) => {
         const outputBufferTransport = new TBufferedTransport(undefined, (buffer, seqId) => {
-            resolve(buffer);
+            if (buffer !== undefined) {
+                resolve(buffer);
+            } else {
+                reject();
+            }
         });
         const outputProtocol = new TCompactProtocol(outputBufferTransport);
         SchoolSerde.write(outputProtocol, school);