@malloydata/malloy-tests 0.0.135 → 0.0.136-dev240326234246

package/dist/databases/duckdb/duckdb.spec.js

@@ -0,0 +1,124 @@
+"use strict";
+/*
+ * Copyright 2023 Google LLC
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files
+ * (the "Software"), to deal in the Software without restriction,
+ * including without limitation the rights to use, copy, modify, merge,
+ * publish, distribute, sublicense, and/or sell copies of the Software,
+ * and to permit persons to whom the Software is furnished to do so,
+ * subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+ * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+ * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+ * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+const luxon_1 = require("luxon");
+const runtimes_1 = require("../../runtimes");
+require("../../util/db-jest-matchers");
+const util_1 = require("../../util");
+// TODO identify which tests need to run on wasm and move them into their own file
+const runtimes = ['duckdb', 'duckdb_wasm'];
+const [describe, databases] = (0, util_1.describeIfDatabaseAvailable)(runtimes);
+const allDucks = new runtimes_1.RuntimeList(databases);
+describe.each(allDucks.runtimeList)('duckdb:%s', (dbName, runtime) => {
+    it('can open tables with wildcards', async () => {
+        await expect(`
+      run: duckdb.table('test/data/duckdb/flights/part.*.parquet') -> {
+        top: 1
+        group_by: carrier;
+      }
+    `).malloyResultMatches(runtime, { carrier: 'AA' });
+    });
+    it('accepts all schema numbers', async () => {
+        const allNumeric = [
+            'BIGINT',
+            'DOUBLE',
+            'FLOAT',
+            'FLOAT4',
+            'FLOAT8',
+            'HUGEINT',
+            'INT',
+            'INT1',
+            'INT2',
+            'INT4',
+            'INT8',
+            'INTEGER',
+            'LONG',
+            'REAL',
+            'SHORT',
+            'SIGNED',
+            'SMALLINT',
+            'TINYINT',
+            'UBIGINT',
+            'UINTEGER',
+            'USMALLINT',
+            'UTINYINT',
+        ];
+        const allFields = allNumeric.map(numType => `a${numType.toLowerCase()}`);
+        const query = `
+      run: ${dbName}.sql("""
+        SELECT
+        ${allNumeric
+            .map(numType => `1::${numType} as a${numType.toLowerCase()}`)
+            .join(',\n')}
+      """) -> {
+        aggregate:
+        ${allFields
+            .map(fieldType => `sum_${fieldType} is sum(${fieldType})`)
+            .join('\n')}
+      }
+    `;
+        await expect(query).malloyResultMatches(runtime, allNumeric.reduce((building, ent) => {
+            building[`sum_a${ent.toLowerCase()}`] = 1;
+            return building;
+        }, {}));
+    });
+    it('can open json files', async () => {
+        await expect(`
+      run: duckdb.table('test/data/duckdb/test.json') -> {
+        select: *
+      }`).malloyResultMatches(runtime, { foo: 'bar' });
+    });
+    it('supports timezones', async () => {
+        await runtime.connection.runSQL("SET TimeZone='CET'");
+        const result = await runtime.connection.runSQL("SELECT current_setting('TimeZone')");
+        expect(result.rows[0]).toEqual({ "current_setting('TimeZone')": 'CET' });
+    });
+    it('supports varchars with parameters', async () => {
+        await expect("run: duckdb.sql(\"SELECT 'a'::VARCHAR as abc, 'a3'::VARCHAR(3) as abc3\")").malloyResultMatches(runtime, { abc: 'a', abc3: 'a3' });
+    });
+    describe('time', () => {
+        const zone = 'America/Mexico_City'; // -06:00 no DST
+        const zone_2020 = luxon_1.DateTime.fromObject({
+            year: 2020,
+            month: 2,
+            day: 20,
+            hour: 0,
+            minute: 0,
+            second: 0,
+        }, {
+            zone,
+        });
+        test('can cast TIMESTAMPTZ to timestamp', async () => {
+            await expect(`run: duckdb.sql("""
+              SELECT TIMESTAMPTZ '2020-02-20 00:00:00 ${zone}' as t_tstz
+          """) -> {
+            select: mex_220 is t_tstz::timestamp
+          }`).malloyResultMatches(runtime, { mex_220: zone_2020.toJSDate() });
+        });
+    });
+});
+afterAll(async () => {
+    await allDucks.closeAll();
+});
+//# sourceMappingURL=duckdb.spec.js.map

package/dist/databases/duckdb/duckdb.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"duckdb.spec.js","sourceRoot":"","sources":["../../../src/databases/duckdb/duckdb.spec.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;;;;;GAqBG;;AAEH,iCAA+B;AAC/B,6CAA2C;AAC3C,uCAAqC;AACrC,qCAAuD;AAEvD,kFAAkF;AAClF,MAAM,QAAQ,GAAG,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;AAE3C,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,GAAG,IAAA,kCAA2B,EAAC,QAAQ,CAAC,CAAC;AACpE,MAAM,QAAQ,GAAG,IAAI,sBAAW,CAAC,SAAS,CAAC,CAAC;AAE5C,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,WAAW,EAAE,CAAC,MAAM,EAAE,OAAO,EAAE,EAAE;IACnE,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,MAAM,CAAC;;;;;KAKZ,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,OAAO,EAAE,IAAI,EAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;QAC1C,MAAM,UAAU,GAAG;YACjB,QAAQ;YACR,QAAQ;YACR,OAAO;YACP,QAAQ;YACR,QAAQ;YACR,SAAS;YACT,KAAK;YACL,MAAM;YACN,MAAM;YACN,MAAM;YACN,MAAM;YACN,SAAS;YACT,MAAM;YACN,MAAM;YACN,OAAO;YACP,QAAQ;YACR,UAAU;YACV,SAAS;YACT,SAAS;YACT,UAAU;YACV,WAAW;YACX,UAAU;SACX,CAAC;QACF,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,IAAI,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;QACzE,MAAM,KAAK,GAAG;aACL,MAAM;;UAET,UAAU;aACT,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,OAAO,QAAQ,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC;aAC5D,IAAI,CAAC,KAAK,CAAC;;;UAGZ,SAAS;aACR,GAAG,CAAC,SAAS,CAAC,EAAE,CAAC,OAAO,SAAS,WAAW,SAAS,GAAG,CAAC;aACzD,IAAI,CAAC,IAAI,CAAC;;KAEhB,CAAC;QACF,MAAM,MAAM,CAAC,KAAK,CAAC,CAAC,mBAAmB,CACrC,OAAO,EACP,UAAU,CAAC,MAAM,CAAyB,CAAC,QAAQ,EAAE,GAAG,EAAE,EAAE;YAC1D,QAAQ,CAAC,QAAQ,GAAG,CAAC,WAAW,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;YAC1C,OAAO,QAAQ,CAAC;QAClB,CAAC,EAAE,EAAE,CAAC,CACP,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qBAAqB,EAAE,KAAK,IAAI,EAAE;QACnC,MAAM,MAAM,CAAC;;;QAGT,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,GAAG,EAAE,KAAK,EAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,MAAM,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,oBAAoB,CAAC,CAAC;QACtD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,MAAM,CAC5C,oCAAoC,CACrC,CAAC;QACF,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAC,6BAA6B,EAAE,KAAK,EAAC,CAAC,CAAC;IACzE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,MAAM,CACV,2EAA2E,CAC5E,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,MAAM,IAAI,GAAG,qBAAqB,CAAC,CAAC,gBAAgB;QACpD,MAAM,SAAS,GAAG,gBAAQ,CAAC,UAAU,CACnC;YACE,IAAI,EAAE,IAAI;YACV,KAAK,EAAE,CAAC;YACR,GAAG,EAAE,EAAE;YACP,IAAI,EAAE,CAAC;YACP,MAAM,EAAE,CAAC;YACT,MAAM,EAAE,CAAC;SACV,EACD;YACE,IAAI;SACL,CACF,CAAC;QACF,IAAI,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,MAAM,CACV;wDACgD,IAAI;;;YAGhD,CACL,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,OAAO,EAAE,SAAS,CAAC,QAAQ,EAAE,EAAC,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;IAClB,MAAM,QAAQ,CAAC,QAAQ,EAAE,CAAC;AAC5B,CAAC,CAAC,CAAC"}

package/dist/databases/duckdb/streaming.spec.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/databases/duckdb/streaming.spec.js

@@ -0,0 +1,142 @@
+"use strict";
+/*
+ * Copyright 2023 Google LLC
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files
+ * (the "Software"), to deal in the Software without restriction,
+ * including without limitation the rights to use, copy, modify, merge,
+ * publish, distribute, sublicense, and/or sell copies of the Software,
+ * and to permit persons to whom the Software is furnished to do so,
+ * subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+ * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+ * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+ * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+const malloy_1 = require("@malloydata/malloy");
+const util_1 = require("../../util");
+const runtimes_1 = require("../../runtimes");
+class StringAccumulator {
+    constructor() {
+        this.accumulatedValue = '';
+    }
+    write(text) {
+        this.accumulatedValue += text;
+    }
+    close() {
+        return;
+    }
+}
+const runtimes = ['duckdb'];
+const [_describe, databases] = (0, util_1.describeIfDatabaseAvailable)(runtimes);
+function modelText(databaseName) {
+    return `source: airports is ${databaseName}.table('test/data/duckdb/airports.parquet') extend {
+  rename: facility_type is fac_type
+
+  measure: airport_count is count()
+  measure: avg_elevation is avg(elevation)
+
+  view: higher_elevation is {
+    group_by: faa_region
+    aggregate: airport_count, avg_elevation
+    order_by: avg_elevation desc
+    limit: 5
+  }
+
+  view: by_county is {
+    where: county != null
+    group_by: county
+    aggregate: airport_count
+    limit: 2
+    order_by: airport_count desc, county desc
+  }
+
+  view: by_state is {
+    where: state != null
+    group_by: state
+    aggregate: airport_count
+    limit: 2
+    nest: by_county
+  }
+
+  view: by_facility_type is {
+    group_by: facility_type
+    aggregate:
+      airport_count
+    limit: 2
+  }
+
+  view: airports_by_region is {
+    group_by: faa_region
+    nest:
+      by_state
+      by_facility_type
+    limit: 2
+    aggregate: airport_count
+  }
+}`;
+}
+describe('Streaming tests', () => {
+    if (!databases.length) {
+        it.skip('skipped', () => { });
+    }
+    const runtimes = new runtimes_1.RuntimeList(databases);
+    afterAll(async () => {
+        await runtimes.closeAll();
+    });
+    runtimes.runtimeMap.forEach((runtime, databaseName) => {
+        it(`stream nested results to CSV - ${databaseName}`, async () => {
+            const stream = runtime
+                .loadModel(modelText(databaseName))
+                .loadQuery('run: airports -> airports_by_region')
+                .runStream();
+            const accummulator = new StringAccumulator();
+            const csvWriter = new malloy_1.CSVWriter(accummulator);
+            await csvWriter.process(stream);
+            const expectedCsv = `faa_region,by_state,,,,by_facility_type,,airport_count
+AGL,state,airport_count,by_county,,facility_type,airport_count,4437
+,IL,890,county,airport_count,AIRPORT,3443,
+,,,COOK,51,HELIPORT,826,
+,,,LA SALLE,39,,,
+,OH,749,county,airport_count,,,
+,,,FRANKLIN,27,,,
+,,,CUYAHOGA,27,,,
+ASW,state,airport_count,by_county,,facility_type,airport_count,3268
+,TX,1845,county,airport_count,AIRPORT,2341,
+,,,HARRIS,135,HELIPORT,861,
+,,,TARRANT,63,,,
+,LA,500,county,airport_count,,,
+,,,PLAQUEMINES,31,,,
+,,,VERMILION,29,,,
+`;
+            expect(accummulator.accumulatedValue).toBe(expectedCsv);
+        });
+        it(`stream simple results to CSV - ${databaseName}`, async () => {
+            const stream = runtime
+                .loadModel(modelText(databaseName))
+                .loadQuery('run: airports -> higher_elevation')
+                .runStream();
+            const accummulator = new StringAccumulator();
+            const csvWriter = new malloy_1.CSVWriter(accummulator);
+            await csvWriter.process(stream);
+            const expectedCsv = `faa_region,airport_count,avg_elevation
+ANM,2102,3284.3910561370126
+AWP,1503,1667.0991350632069
+ACE,1579,1339.0139328689045
+ASW,3268,1007.2873317013464
+AGL,4437,983.4800540906018
+`;
+            expect(accummulator.accumulatedValue).toBe(expectedCsv);
+        });
+    });
+});
+//# sourceMappingURL=streaming.spec.js.map

package/dist/databases/duckdb/streaming.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming.spec.js","sourceRoot":"","sources":["../../../src/databases/duckdb/streaming.spec.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;;;;;GAqBG;;AAEH,+CAA0D;AAC1D,qCAAuD;AACvD,6CAA2C;AAE3C,MAAM,iBAAiB;IAAvB;QACS,qBAAgB,GAAG,EAAE,CAAC;IAS/B,CAAC;IAPC,KAAK,CAAC,IAAY;QAChB,IAAI,CAAC,gBAAgB,IAAI,IAAI,CAAC;IAChC,CAAC;IAED,KAAK;QACH,OAAO;IACT,CAAC;CACF;AAED,MAAM,QAAQ,GAAG,CAAC,QAAQ,CAAC,CAAC;AAE5B,MAAM,CAAC,SAAS,EAAE,SAAS,CAAC,GAAG,IAAA,kCAA2B,EAAC,QAAQ,CAAC,CAAC;AAErE,SAAS,SAAS,CAAC,YAAoB;IACrC,OAAO,uBAAuB,YAAY;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA4C1C,CAAC;AACH,CAAC;AAED,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;IAC/B,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE;QACrB,EAAE,CAAC,IAAI,CAAC,SAAS,EAAE,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;KAC9B;IACD,MAAM,QAAQ,GAAG,IAAI,sBAAW,CAAC,SAAS,CAAC,CAAC;IAE5C,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,QAAQ,CAAC,QAAQ,EAAE,CAAC;IAC5B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE;QACpD,EAAE,CAAC,kCAAkC,YAAY,EAAE,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,MAAM,GAAG,OAAO;iBACnB,SAAS,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;iBAClC,SAAS,CAAC,qCAAqC,CAAC;iBAChD,SAAS,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,IAAI,iBAAiB,EAAE,CAAC;YAC7C,MAAM,SAAS,GAAG,IAAI,kBAAS,CAAC,YAAY,CAAC,CAAC;YAC9C,MAAM,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChC,MAAM,WAAW,GAAG;;;;;;;;;;;;;;;CAezB,CAAC;YACI,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kCAAkC,YAAY,EAAE,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,MAAM,GAAG,OAAO;iBACnB,SAAS,CAAC,SAAS,CAAC,YAAY,CAAC,CAAC;iBAClC,SAAS,CAAC,mCAAmC,CAAC;iBAC9C,SAAS,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,IAAI,iBAAiB,EAAE,CAAC;YAC7C,MAAM,SAAS,GAAG,IAAI,kBAAS,CAAC,YAAY,CAAC,CAAC;YAC9C,MAAM,SAAS,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAChC,MAAM,WAAW,GAAG;;;;;;CAMzB,CAAC;YACI,MAAM,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/databases/multi-connection/multi_connection.spec.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/databases/multi-connection/multi_connection.spec.js

@@ -0,0 +1,120 @@
+"use strict";
+/* eslint-disable no-console */
+/*
+ * Copyright 2023 Google LLC
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files
+ * (the "Software"), to deal in the Software without restriction,
+ * including without limitation the rights to use, copy, modify, merge,
+ * publish, distribute, sublicense, and/or sell copies of the Software,
+ * and to permit persons to whom the Software is furnished to do so,
+ * subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+ * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+ * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+ * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+const malloy = __importStar(require("@malloydata/malloy"));
+const malloy_1 = require("@malloydata/malloy");
+const runtimes_1 = require("../../runtimes");
+const util_1 = require("../../util");
+const [, databases] = (0, util_1.describeIfDatabaseAvailable)(['duckdb', 'postgres']);
+// *** NOTE ***
+// this is a special case (for now): a test that REQUIRES two databases - duckdb AND postgres
+const describe = databases.filter(d => ['postgres', 'duckdb'].includes(d)).length >= 2
+    ? globalThis.describe
+    : globalThis.describe.skip;
+describe('Multi-connection', () => {
+    const ddbConnection = new runtimes_1.DuckDBTestConnection('duckdb', 'test/data/duckdb/duckdb_test.db');
+    const postgresConnection = new runtimes_1.PostgresTestConnection('postgres');
+    const files = new malloy_1.EmptyURLReader();
+    const connectionMap = new malloy.FixedConnectionMap(new Map(Object.entries({
+        duckdb: ddbConnection,
+        postgres: postgresConnection,
+    })), 'duckdb');
+    const runtime = new malloy.Runtime(files, connectionMap);
+    afterAll(async () => {
+        await postgresConnection.close();
+    });
+    const expressionModelText = `
+source: duckdb_state_facts is duckdb.table('malloytest.state_facts') extend {
+  measure: state_count is count(state)+2
+}
+
+source: postgres_aircraft is postgres.table('malloytest.aircraft') extend {
+  measure: aircraft_count is count(tail_num)+4
+}
+`;
+    const expressionModel = runtime.loadModel(expressionModelText);
+    it('duckdb query', async () => {
+        const result = await expressionModel
+            .loadQuery(`
+      run: duckdb_state_facts-> {
+        aggregate: state_count
+      }
+    `)
+            .run();
+        // console.log(result.sql);
+        expect(result.data.path(0, 'state_count').value).toBe(53);
+    });
+    it('postgres query', async () => {
+        const result = await expressionModel
+            .loadQuery(`
+      run: postgres_aircraft-> {
+        aggregate: aircraft_count
+      }
+    `)
+            .run();
+        expect(result.data.path(0, 'aircraft_count').value).toBe(3603);
+    });
+    it('postgres raw query', async () => {
+        const result = await runtime
+            .loadQuery(`
+      run: postgres.table('malloytest.airports')->{
+        group_by:
+          version is version!()
+        aggregate:
+          code_count is count(code)
+          airport_count is count()
+      }
+    `)
+            .run();
+        expect(result.data.path(0, 'airport_count').value).toBe(19793);
+        expect(result.data.path(0, 'version').value).toMatch(/Postgre/);
+        expect(result.data.path(0, 'code_count').value).toBe(19793);
+    });
+});
+//# sourceMappingURL=multi_connection.spec.js.map

package/dist/databases/multi-connection/multi_connection.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"multi_connection.spec.js","sourceRoot":"","sources":["../../../src/databases/multi-connection/multi_connection.spec.ts"],"names":[],"mappings":";AAAA,+BAA+B;AAC/B;;;;;;;;;;;;;;;;;;;;;GAqBG;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,8DAA8D;AAE9D,2DAA6C;AAC7C,+CAAkD;AAClD,6CAA4E;AAC5E,qCAAuD;AAEvD,MAAM,CAAC,EAAE,SAAS,CAAC,GAAG,IAAA,kCAA2B,EAAC,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC,CAAC;AAE1E,eAAe;AACf,6FAA6F;AAC7F,MAAM,QAAQ,GACZ,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,IAAI,CAAC;IACnE,CAAC,CAAC,UAAU,CAAC,QAAQ;IACrB,CAAC,CAAC,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC;AAE/B,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,MAAM,aAAa,GAAG,IAAI,+BAAoB,CAC5C,QAAQ,EACR,iCAAiC,CAClC,CAAC;IACF,MAAM,kBAAkB,GAAG,IAAI,iCAAsB,CAAC,UAAU,CAAC,CAAC;IAClE,MAAM,KAAK,GAAG,IAAI,uBAAc,EAAE,CAAC;IAEnC,MAAM,aAAa,GAAG,IAAI,MAAM,CAAC,kBAAkB,CACjD,IAAI,GAAG,CACL,MAAM,CAAC,OAAO,CAAC;QACb,MAAM,EAAE,aAAa;QACrB,QAAQ,EAAE,kBAAkB;KAC7B,CAAC,CACH,EACD,QAAQ,CACT,CAAC;IAEF,MAAM,OAAO,GAAG,IAAI,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,aAAa,CAAC,CAAC;IAEzD,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,kBAAkB,CAAC,KAAK,EAAE,CAAC;IACnC,CAAC,CAAC,CAAC;IAEH,MAAM,mBAAmB,GAAG;;;;;;;;CAQ7B,CAAC;IAEA,MAAM,eAAe,GAAG,OAAO,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;IAE/D,EAAE,CAAC,cAAc,EAAE,KAAK,IAAI,EAAE;QAC5B,MAAM,MAAM,GAAG,MAAM,eAAe;aACjC,SAAS,CACR;;;;KAIH,CACE;aACA,GAAG,EAAE,CAAC;QACT,2BAA2B;QAC3B,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gBAAgB,EAAE,KAAK,IAAI,EAAE;QAC9B,MAAM,MAAM,GAAG,MAAM,eAAe;aACjC,SAAS,CACR;;;;KAIH,CACE;aACA,GAAG,EAAE,CAAC;QACT,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,gBAAgB,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACjE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oBAAoB,EAAE,KAAK,IAAI,EAAE;QAClC,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,SAAS,CACR;;;;;;;;KAQH,CACE;aACA,GAAG,EAAE,CAAC;QACT,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,eAAe,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC/D,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;QAChE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,YAAY,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/databases/postgres/postgres.spec.d.ts

@@ -0,0 +1 @@
+import '../../util/db-jest-matchers';

package/dist/databases/postgres/postgres.spec.js

@@ -0,0 +1,140 @@
+"use strict";
+/*
+ * Copyright 2023 Google LLC
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files
+ * (the "Software"), to deal in the Software without restriction,
+ * including without limitation the rights to use, copy, modify, merge,
+ * publish, distribute, sublicense, and/or sell copies of the Software,
+ * and to permit persons to whom the Software is furnished to do so,
+ * subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
+ * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
+ * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
+ * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
+ * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+Object.defineProperty(exports, "__esModule", { value: true });
+/* eslint-disable no-console */
+const runtimes_1 = require("../../runtimes");
+const util_1 = require("../../util");
+require("../../util/db-jest-matchers");
+const luxon_1 = require("luxon");
+const [describe] = (0, util_1.describeIfDatabaseAvailable)(['postgres']);
+describe('Postgres tests', () => {
+    const runtimeList = new runtimes_1.RuntimeList(['postgres']);
+    const runtime = runtimeList.runtimeMap.get('postgres');
+    if (runtime === undefined) {
+        throw new Error("Couldn't build runtime");
+    }
+    // Idempotently create schema and tables with capital letters to use in tests.
+    beforeAll(async () => {
+        await runtime.connection.runSQL('create schema if not exists "UpperSchema";');
+        await Promise.all([
+            runtime.connection.runSQL('create table if not exists "UpperSchema"."UpperSchemaUpperTable" as select 1 as one;'),
+            runtime.connection.runSQL('create table if not exists "UpperTablePublic" as select 1 as one;'),
+        ]);
+    });
+    afterAll(async () => {
+        await runtimeList.closeAll();
+    });
+    it('run an sql query', async () => {
+        await expect('run: postgres.sql("SELECT 1 as n") -> { select: n }').malloyResultMatches(runtime, { n: 1 });
+    });
+    it('mixed case col names are properly quoted so they retain case in results', async () => {
+        await expect(`
+      run: postgres.sql('SELECT 1 as "upperLower"') -> { select: upperLower }
+    `).malloyResultMatches(runtime, { upperLower: 1 });
+    });
+    it('fields which are sql keywords are quoted', async () => {
+        await expect(`
+    run: postgres.sql('SELECT 1 as "select"') -> {
+      select:
+        select
+        create is select + 1
+    }
+  `).malloyResultMatches(runtime, { select: 1, create: 2 });
+    });
+    async function oneExists(rt, tn) {
+        try {
+            const lookForOne = await rt
+                .loadQuery(`run: postgres.sql('SELECT one FROM ${tn}')`)
+                .run();
+            const one = lookForOne.data.path(0, 'one').value;
+            return one === 1;
+        }
+        catch (e) {
+            return false;
+        }
+    }
+    it('will quote to properly access mixed case table name', async () => {
+        if (await oneExists(runtime, 'public."UpperTablePublic"')) {
+            await expect(`
+        run: postgres.table('public.UpperTablePublic') -> { select: one }
+      `).malloyResultMatches(runtime, { one: 1 });
+        }
+    });
+    it('quote to properly access mixes case schema name', async () => {
+        if (await oneExists(runtime, '"UpperSchema"."UpperSchemaUpperTable"')) {
+            await expect(`
+        run: postgres.table('UpperSchema.UpperSchemaUpperTable') -> { select: one }
+      `).malloyResultMatches(runtime, { one: 1 });
+        }
+    });
+    it('passes unsupported data', async () => {
+        const result = await runtime
+            .loadQuery('run: postgres.sql("SELECT int4range(10, 20) as ranger")')
+            .run();
+        expect(result.data.value[0]['ranger']).toBeDefined();
+    });
+    it('supports varchars with parameters', async () => {
+        await expect("run: postgres.sql(\"SELECT 'a'::VARCHAR as abc, 'a3'::VARCHAR(3) as abc3\")").malloyResultMatches(runtime, { abc: 'a', abc3: 'a3' });
+    });
+    describe('time', () => {
+        const zone = 'America/Mexico_City'; // -06:00 no DST
+        const zone_2020 = luxon_1.DateTime.fromObject({
+            year: 2020,
+            month: 2,
+            day: 20,
+            hour: 0,
+            minute: 0,
+            second: 0,
+        }, {
+            zone,
+        });
+        test('can cast TIMESTAMPTZ to timestamp', async () => {
+            await expect(`run: postgres.sql("""
+              SELECT TIMESTAMPTZ '2020-02-20 00:00:00 ${zone}' as t_tstz
+          """) -> {
+            select: mex_220 is t_tstz::timestamp
+          }`).malloyResultMatches(runtime, { mex_220: zone_2020.toJSDate() });
+        });
+    });
+    describe('numbers', () => {
+        it.each([
+            'SMALLINT',
+            'INTEGER',
+            'BIGINT',
+            'DECIMAL',
+            'NUMERIC',
+            'REAL',
+            'DOUBLE PRECISION',
+        ])('supports %s', async (sqlType) => {
+            const result = await runtime
+                .loadQuery(`run: postgres.sql("SELECT 10::${sqlType} as d")`)
+                .run();
+            const field = result.data.field.allFields[0];
+            expect(field.isAtomicField()).toBe(true);
+            expect(field.isNumber()).toBe(true);
+            expect(result.data.value[0]['d']).toEqual(10);
+        });
+    });
+});
+//# sourceMappingURL=postgres.spec.js.map

package/dist/databases/postgres/postgres.spec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"postgres.spec.js","sourceRoot":"","sources":["../../../src/databases/postgres/postgres.spec.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;;;;;;;;;;GAqBG;;AAEH,+BAA+B;AAE/B,6CAA2C;AAE3C,qCAAuD;AACvD,uCAAqC;AACrC,iCAA+B;AAE/B,MAAM,CAAC,QAAQ,CAAC,GAAG,IAAA,kCAA2B,EAAC,CAAC,UAAU,CAAC,CAAC,CAAC;AAE7D,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,MAAM,WAAW,GAAG,IAAI,sBAAW,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;IAClD,MAAM,OAAO,GAAG,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACvD,IAAI,OAAO,KAAK,SAAS,EAAE;QACzB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;KAC3C;IAED,8EAA8E;IAC9E,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,OAAO,CAAC,UAAU,CAAC,MAAM,CAC7B,4CAA4C,CAC7C,CAAC;QACF,MAAM,OAAO,CAAC,GAAG,CAAC;YAChB,OAAO,CAAC,UAAU,CAAC,MAAM,CACvB,sFAAsF,CACvF;YACD,OAAO,CAAC,UAAU,CAAC,MAAM,CACvB,mEAAmE,CACpE;SACF,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,KAAK,IAAI,EAAE;QAClB,MAAM,WAAW,CAAC,QAAQ,EAAE,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kBAAkB,EAAE,KAAK,IAAI,EAAE;QAChC,MAAM,MAAM,CACV,qDAAqD,CACtD,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,CAAC,EAAE,CAAC,EAAC,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yEAAyE,EAAE,KAAK,IAAI,EAAE;QACvF,MAAM,MAAM,CAAC;;KAEZ,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,UAAU,EAAE,CAAC,EAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;QACxD,MAAM,MAAM,CAAC;;;;;;GAMd,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,MAAM,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAC,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,KAAK,UAAU,SAAS,CAAC,EAAW,EAAE,EAAU;QAC9C,IAAI;YACF,MAAM,UAAU,GAAG,MAAM,EAAE;iBACxB,SAAS,CAAC,sCAAsC,EAAE,IAAI,CAAC;iBACvD,GAAG,EAAE,CAAC;YACT,MAAM,GAAG,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,KAAK,CAAC;YACjD,OAAO,GAAG,KAAK,CAAC,CAAC;SAClB;QAAC,OAAO,CAAC,EAAE;YACV,OAAO,KAAK,CAAC;SACd;IACH,CAAC;IAED,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,IAAI,MAAM,SAAS,CAAC,OAAO,EAAE,2BAA2B,CAAC,EAAE;YACzD,MAAM,MAAM,CAAC;;OAEZ,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC;SAC3C;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,IAAI,MAAM,SAAS,CAAC,OAAO,EAAE,uCAAuC,CAAC,EAAE;YACrE,MAAM,MAAM,CAAC;;OAEZ,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,GAAG,EAAE,CAAC,EAAC,CAAC,CAAC;SAC3C;IACH,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;QACvC,MAAM,MAAM,GAAG,MAAM,OAAO;aACzB,SAAS,CAAC,yDAAyD,CAAC;aACpE,GAAG,EAAE,CAAC;QACT,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;IACvD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;QACjD,MAAM,MAAM,CACV,6EAA6E,CAC9E,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAC,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,MAAM,IAAI,GAAG,qBAAqB,CAAC,CAAC,gBAAgB;QACpD,MAAM,SAAS,GAAG,gBAAQ,CAAC,UAAU,CACnC;YACE,IAAI,EAAE,IAAI;YACV,KAAK,EAAE,CAAC;YACR,GAAG,EAAE,EAAE;YACP,IAAI,EAAE,CAAC;YACP,MAAM,EAAE,CAAC;YACT,MAAM,EAAE,CAAC;SACV,EACD;YACE,IAAI;SACL,CACF,CAAC;QACF,IAAI,CAAC,mCAAmC,EAAE,KAAK,IAAI,EAAE;YACnD,MAAM,MAAM,CACV;wDACgD,IAAI;;;YAGhD,CACL,CAAC,mBAAmB,CAAC,OAAO,EAAE,EAAC,OAAO,EAAE,SAAS,CAAC,QAAQ,EAAE,EAAC,CAAC,CAAC;QAClE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,SAAS,EAAE,GAAG,EAAE;QACvB,EAAE,CAAC,IAAI,CAAC;YACN,UAAU;YACV,SAAS;YACT,QAAQ;YACR,SAAS;YACT,SAAS;YACT,MAAM;YACN,kBAAkB;SACnB,CAAC,CAAC,aAAa,EAAE,KAAK,EAAC,OAAO,EAAC,EAAE;YAChC,MAAM,MAAM,GAAG,MAAM,OAAO;iBACzB,SAAS,CAAC,iCAAiC,OAAO,SAAS,CAAC;iBAC5D,GAAG,EAAE,CAAC;YACT,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;YAC7C,MAAM,CAAC,KAAK,CAAC,aAAa,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACzC,MAAM,CAAE,KAAqB,CAAC,QAAQ,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/dist/databases/shared/test_list.d.ts

@@ -0,0 +1,3 @@
+import { RuntimeList } from '../../runtimes';
+export type DatabaseTestSet = (runtimeList: RuntimeList) => void;
+export declare const allDatabaseTestSets: DatabaseTestSet[];

package/dist/databases/shared/test_list.js

@@ -0,0 +1,5 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.allDatabaseTestSets = void 0;
+exports.allDatabaseTestSets = [];
+//# sourceMappingURL=test_list.js.map

package/dist/databases/shared/test_list.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test_list.js","sourceRoot":"","sources":["../../../src/databases/shared/test_list.ts"],"names":[],"mappings":";;;AAIa,QAAA,mBAAmB,GAAsB,EAAE,CAAC"}

package/dist/databases/streaming/streaming.spec.d.ts

@@ -0,0 +1 @@
+export {};