duckdb/test/query_result.test.ts

DuckDB node.js API

import * as duckdb from '..';
import * as assert from 'assert';

describe('QueryResult', () => {
    const total = 1000;

    let db: duckdb.Database;
    let conn: duckdb.Connection;
    before((done) => {
        db = new duckdb.Database(':memory:', () => {
            conn = new duckdb.Connection(db, done);
        });
    });

    it('streams results', async () => {
        let retrieved = 0;
        const stream = conn.stream('SELECT * FROM range(0, ?)', total);
        for await (const row of stream) {
            retrieved++;
        }
        assert.equal(total, retrieved)
    })

    it('streams results using the database object', async () => {
        let retrieved = 0;
        const stream = db.stream('SELECT * FROM range(0, ?)', total);
        for await (const row of stream) {
            retrieved++;
        }
        assert.equal(total, retrieved)
    })
})