duckdb 0.6.1-dev131.0 → 0.6.1-dev140.0

package/test/arrow.test.js

@@ -1,60 +1,51 @@
-var sqlite3 = require('..');
+var duckdb = require('..');
 var assert = require('assert');
 var fs = require('fs');
 
-describe('exec', function() {
-    var db;
-    before(function(done) {
-        db = new sqlite3.Database(':memory:', done);
-    });
-
+describe('arrow IPC API fails neatly when extension not loaded', function() {
     // Note: arrow IPC api requires the arrow extension to be loaded. The tests for this functionality reside in:
     //       https://github.com/duckdblabs/arrow
-    describe(`Arrow IPC API fails neatly when extension not loaded`, () => {
-        let db;
-        let conn;
-        before((done) => {
-            db = new sqlite3.Database(':memory:', {"allow_unsigned_extensions": "true"}, () => {
-                done();
-            });
+    let db;
+    let conn;
+    before((done) => {
+        db = new duckdb.Database(':memory:', {"allow_unsigned_extensions": "true"}, () => {
+            done();
         });
+    });
 
-        it(`Basic examples`, async () => {
-            const range_size = 130000;
-            const query = `SELECT * FROM range(0,${range_size}) tbl(i)`;
-
-            db.arrowIPCStream(query).then(
-                () => Promise.reject(new Error('Expected method to reject.')),
-                err => {
-                    assert(err.message.includes("Catalog Error: Function with name to_arrow_ipc is not on the catalog, but it exists in the arrow extension. To Install and Load the extension, run: INSTALL arrow; LOAD arrow;"))
-                }
-            );
+    it(`basic examples`, async () => {
+        const range_size = 130000;
+        const query = `SELECT * FROM range(0,${range_size}) tbl(i)`;
 
-            db.arrowIPCAll(`SELECT * FROM ipc_table`, function (err, result) {
-                if (err) {
-                    assert(err.message.includes("Catalog Error: Function with name to_arrow_ipc is not on the catalog, but it exists in the arrow extension. To Install and Load the extension, run: INSTALL arrow; LOAD arrow;"))
-                } else {
-                    assert.fail("Expected error");
-                }
-            });
+        db.arrowIPCStream(query).then(
+            () => Promise.reject(new Error('Expected method to reject.')),
+            err => {
+                assert(err.message.includes("Catalog Error: Function with name to_arrow_ipc is not on the catalog, but it exists in the arrow extension. To Install and Load the extension, run: INSTALL arrow; LOAD arrow;"))
+            }
+        );
 
-            assert.throws(() => db.register_buffer("ipc_table", [1,'a',1], true), TypeError, "Incorrect parameters");
+        db.arrowIPCAll(`SELECT * FROM ipc_table`, function (err, result) {
+            if (err) {
+                assert(err.message.includes("Catalog Error: Function with name to_arrow_ipc is not on the catalog, but it exists in the arrow extension. To Install and Load the extension, run: INSTALL arrow; LOAD arrow;"))
+            } else {
+                assert.fail("Expected error");
+            }
         });
 
-        it('Register buffer should be disabled currently', function(done) {
-            try {
-                db.register_buffer();
-                assert(0);
-            } catch (error) {
-                assert(error.message.includes('Register buffer currently not implemented'))
-            }
+        assert.throws(() => db.register_buffer("ipc_table", [1,'a',1], true), TypeError, "Incorrect parameters");
+    });
 
-            try {
-                db.unregister_buffer();
-                assert(0);
-            } catch (error) {
-                assert(error.message.includes('Register buffer currently not implemented'))
-            }
+    it('register buffer should be disabled currently', function(done) {
+        db.register_buffer("test", [new Uint8Array(new ArrayBuffer(10))], true, (err) => {
+            assert(err)
+            assert(err.includes("Function with name scan_arrow_ipc is not on the catalog, but it exists in the arrow extension. To Install and Load the extension, run: INSTALL arrow; LOAD arrow;"));
+            done()
+        });
+    });
+
+    it('unregister will silently do nothing', function(done) {
+        db.unregister_buffer("test", (err) => {
+            assert(!err)
             done()
         });
     });