Add streaming data tests

Author: ptiurin

test/integration/v2/stream.test.ts

@@ -74,6 +74,7 @@ describe("streams", () => {
         resolve(null);
       });
     });
+    expect(sum).toEqual(250000);
   });
   it("stream transformers", async () => {
     class SerializeRowStream extends stream.Transform {
@@ -153,4 +154,86 @@ describe("streams", () => {
         "        ^"
     );
   });
+  it("check data types", async () => {
+    const seriesNum = 2;
+    const generateLargeResultQuery = (rows: number) => `
+      SELECT 
+        i as id,
+        'user_' || i::string as username,
+        'email_' || i::string || '@example.com' as email,
+        CASE WHEN i % 2 = 0 THEN 'active' ELSE 'inactive' END as status,
+        CAST('100000000000000000' as BIGINT) as big_number,
+        '2024-01-01'::date + (i % 365) as created_date,
+        RANDOM() * 1000 as score,
+        'Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed do eiusmod tempor incididunt ut labore et dolore magna aliqua.' as description
+      FROM generate_series(1, ${rows}) as i
+    `;
+
+    const firebolt = Firebolt({
+      apiEndpoint: process.env.FIREBOLT_API_ENDPOINT as string
+    });
+
+    const connection = await firebolt.connect(connectionParams);
+
+    const statement = await connection.executeStream(
+      generateLargeResultQuery(seriesNum),
+      {
+        response: {
+          normalizeData: true
+        }
+      }
+    );
+
+    const { data } = await statement.streamResult();
+    let rowCount = 0;
+    // const meta = await stream.once(data, "meta");
+    data.on("meta", meta => {
+      console.log("Meta:", meta);
+    });
+    // console.log("Meta:", meta);
+    // expect(meta).toEqual([
+    //   { name: "id", type: "int" },
+    //   { name: "username", type: "text" },
+    //   { name: "email", type: "text" },
+    //   { name: "status", type: "text" },
+    //   { name: "big_number", type: "bigint" },
+    //   { name: "created_date", type: "date" },
+    //   { name: "score", type: "double precision" },
+    //   { name: "description", type: "text" }
+    // ]);
+
+    data.on("data", row => {
+      rowCount++;
+      // Verify first row data types
+      if (rowCount === 1) {
+        expect(typeof row[0]).toBe("number"); // id
+        expect(typeof row[1]).toBe("string"); // username
+        expect(typeof row[2]).toBe("string"); // email
+        expect(typeof row[3]).toBe("string"); // status
+        expect(typeof row[4]).toBe("number"); // big_number (should be handled as number or BigNumber)
+        expect(row[5]).toBeInstanceOf(Date); // created_date
+        expect(typeof row[6]).toBe("number"); // score
+        expect(typeof row[7]).toBe("string"); // description
+
+        // Verify actual values for first row
+        expect(row[0]).toBe(1);
+        expect(row[1]).toBe("user_1");
+        expect(row[2]).toBe("[email protected]");
+        expect(row[3]).toBe("inactive"); // i=1, 1%2=1, so 'inactive'
+        expect(row[4]).toBe(100000000000000000);
+        expect(row[5]).toEqual(new Date("2024-01-02")); // 2024-01-01 + 1 day
+        expect(typeof row[6]).toBe("number");
+        expect(row[7]).toBe(
+          "Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed do eiusmod tempor incididunt ut labore et dolore magna aliqua."
+        );
+      }
+    });
+
+    await new Promise(resolve => {
+      data.on("end", () => {
+        expect(rowCount).toEqual(seriesNum);
+        resolve(null);
+      });
+    });
+  });
 });