sqlite/ext/fiddle/testing1.js

/*
  2022-05-22

  The author disclaims copyright to this source code.  In place of a
  legal notice, here is a blessing:

  *   May you do good and not evil.
  *   May you find forgiveness for yourself and forgive others.
  *   May you share freely, never taking more than you give.

  ***********************************************************************

  A basic test script for sqlite3-api.js.
*/
(function(){
    const T = self.SqliteTestUtil;
    const log = console.log.bind(console);

    const test1 = function(db,api){
        log("Basic sanity tests...");
        T.assert(db._pDb);
        let st = db.prepare("select 3 as a");
        //log("statement =",st);
        T.assert(st._pStmt)
            .assert(!st._mayGet)
            .assert('a' === st.getColumnName(0))
            .assert(st === db._statements[st._pStmt])
            .assert(1===st.columnCount)
            .assert(0===st.parameterCount)
            .mustThrow(()=>st.bind(1,null))
            .assert(true===st.step())
            .assert(3 === st.get(0))
            .mustThrow(()=>st.get(1))
            .mustThrow(()=>st.get(0,~api.SQLITE_INTEGER))
            .assert(3 === st.get(0,api.SQLITE_INTEGER))
            .assert(3 === st.getInt(0))
            .assert('3' === st.get(0,api.SQLITE_TEXT))
            .assert('3' === st.getString(0))
            .assert(3.0 === st.get(0,api.SQLITE_FLOAT))
            .assert(3.0 === st.getFloat(0))
            .assert(st.get(0,api.SQLITE_BLOB) instanceof Uint8Array)
            .assert(st.getBlob(0) instanceof Uint8Array)
            .assert(3 === st.get([])[0])
            .assert(3 === st.get({}).a)
            .assert(3 === st.getJSON(0))
            .assert(st._mayGet)
            .assert(false===st.step())
            .assert(!st._mayGet)
        ;
        let pId = st._pStmt;
        st.finalize();
        T.assert(!st._pStmt)
            .assert(!db._statements[pId]);

        let list = [];
        db.exec({
            sql:`CREATE TABLE t(a,b);
INSERT INTO t(a,b) VALUES(1,2),(3,4),(?,?);`,
            multi: true,
            saveSql: list,
            bind: [5,6]
        });
        T.assert(2 === list.length);
        //log("Exec'd SQL:", list);
        let counter = 0, colNames = [];
        db.exec("SELECT a a, b b FROM t",{
            rowMode: 'object',
            callback: function(row,stmt){
                if(!counter) stmt.getColumnNames(colNames);
                ++counter;
                T.assert(row.a%2 && row.a<6);
            }
        });
        assert(2 === colNames.length);
        assert('a' === colNames[0]);
        T.assert(3 === counter);
        db.exec("SELECT a a, b b FROM t",{
            rowMode: 'array',
            callback: function(row,stmt){
                ++counter;
                assert(Array.isArray(row));
                T.assert(0===row[1]%2 && row[1]<7);
            }
        });
        T.assert(6 === counter);
    };

    const testUDF = function(db){
        log("Testing UDF...");
        db.createFunction("foo",function(a,b){return a+b});
        T.assert(7===db.selectValue("select foo(3,4)")).
            assert(5===db.selectValue("select foo(3,?)",2)).
            assert(5===db.selectValue("select foo(?,?)",[1,4])).
            assert(5===db.selectValue("select foo($a,$b)",{$a:0,$b:5}));
        db.createFunction("bar", {
            arity: -1,
            callback: function(){
                var rc = 0;
                for(let i = 0; i < arguments.length; ++i) rc += arguments[i];
                return rc;
            }
        });

        log("Testing DB::selectValue() w/ UDF...");
        T.assert(0===db.selectValue("select bar()")).
            assert(1===db.selectValue("select bar(1)")).
            assert(3===db.selectValue("select bar(1,2)")).
            assert(-1===db.selectValue("select bar(1,2,-4)"));

        T.assert('hi' === db.selectValue("select ?",'hi')).
            assert(null===db.selectValue("select null")).
            assert(null === db.selectValue("select ?",null)).
            assert(null === db.selectValue("select ?",[null])).
            assert(null === db.selectValue("select $a",{$a:null}));
    };

    const runTests = function(namespace){
        T.assert(Module._free instanceof Function).
            assert(Module.allocate instanceof Function).
            assert(Module.addFunction instanceof Function).
            assert(Module.removeFunction instanceof Function);
        const api = namespace.api;
        const oo = namespace.SQLite3;
        console.log("Loaded module:",api.sqlite3_libversion(),
                    api.sqlite3_sourceid());
        log("Build options:",oo.compileOptionUsed());
        const db = new oo.DB();
        try {
            log("DB:",db.filename);
            [
                test1, testUDF
            ].forEach((f)=>f(db, api));
        }finally{
            db.close();
        }
        log("Total Test count:",T.counter);
    };

    self.Module.postRun.push(function(theModule){
        /** Use a timeout so that we are (hopefully) out from under the
            module init stack when our setup gets run. Just on principle,
            not because we _need_ to be. */
        setTimeout(()=>theModule.loadSqliteAPI(runTests), 0);
    });
})(self/*window or worker*/);