added unit tests with lz4 disabled

Author: shivam2680

tests/unit/result/ArrowResultHandler.test.ts

@@ -66,6 +66,10 @@ const sampleRowSet4: TRowSet = {
   ],
 };
 
+declare global {
+  var LZ4: typeof import('lz4') | undefined; // Declare LZ4 on globalThis with proper type
+}
+
 describe('ArrowResultHandler', () => {
   it('should return data', async () => {
     const rowSetProvider = new ResultsProviderStub([sampleRowSet1], undefined);
@@ -229,4 +233,17 @@ describe('ArrowResultHandler', () => {
     });
     expect(await result.hasMore()).to.be.false;
   });
+
+  it('should handle data without LZ4 compression', async () => {
+    const rowSetProvider = new ResultsProviderStub([sampleRowSet1], undefined);
+    const result = new ArrowResultHandler(new ClientContextStub(), rowSetProvider, {
+      status: { statusCode: TStatusCode.SUCCESS_STATUS },
+      arrowSchema: sampleArrowSchema,
+      lz4Compressed: false,
+    });
+
+    const { batches } = await result.fetchNext({ limit: 10000 });
+    const expectedBatches = sampleRowSet1.arrowBatches?.map(({ batch }) => batch) ?? []; // Ensure iterable
+    expect(batches).to.deep.eq([sampleArrowSchema, ...expectedBatches]);
+  });
 });

tests/unit/result/CloudFetchResultHandler.test.ts

@@ -379,4 +379,24 @@ describe('CloudFetchResultHandler', () => {
       expect(context.invokeWithRetryStub.callCount).to.be.equal(1);
     }
   });
+
+  it('should handle data without LZ4 compression', async () => {
+    const context = new ClientContextStub();
+    const rowSetProvider = new ResultsProviderStub([sampleRowSet1], undefined);
+
+    const result = new CloudFetchResultHandler(context, rowSetProvider, {
+      lz4Compressed: false,
+      status: { statusCode: TStatusCode.SUCCESS_STATUS },
+    });
+
+    context.invokeWithRetryStub.callsFake(async () => ({
+      request: new Request('localhost'),
+      response: new Response(sampleArrowBatch, { status: 200 }), // Return only the batch
+    }));
+
+    const { batches } = await result.fetchNext({ limit: 10000 });
+
+    // Ensure the batches array matches the expected structure
+    expect(batches).to.deep.eq([sampleArrowBatch]);
+  });
 });