@malloydata/malloy-tests 0.0.147 → 0.0.148-dev240529172135

package/package.json

@@ -21,13 +21,13 @@
   },
   "dependencies": {
     "@jest/globals": "^29.4.3",
-    "@malloydata/db-bigquery": "^0.0.147",
-    "@malloydata/db-duckdb": "^0.0.147",
-    "@malloydata/db-postgres": "^0.0.147",
-    "@malloydata/db-snowflake": "^0.0.147",
-    "@malloydata/db-trino": "^0.0.147",
-    "@malloydata/malloy": "^0.0.147",
-    "@malloydata/render": "^0.0.147",
+    "@malloydata/db-bigquery": "^0.0.148-dev240529172135",
+    "@malloydata/db-duckdb": "^0.0.148-dev240529172135",
+    "@malloydata/db-postgres": "^0.0.148-dev240529172135",
+    "@malloydata/db-snowflake": "^0.0.148-dev240529172135",
+    "@malloydata/db-trino": "^0.0.148-dev240529172135",
+    "@malloydata/malloy": "^0.0.148-dev240529172135",
+    "@malloydata/render": "^0.0.148-dev240529172135",
     "jsdom": "^22.1.0",
     "luxon": "^2.4.0",
     "madge": "^6.0.0"
@@ -36,5 +36,5 @@
     "@types/jsdom": "^21.1.1",
     "@types/luxon": "^2.4.0"
   },
-  "version": "0.0.147"
+  "version": "0.0.148-dev240529172135"
 }

package/src/experimental-dialects.spec.ts

@@ -0,0 +1,116 @@
+/*
+ * 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.
+ */
+
+import {
+  DuckDBDialect,
+  FetchSchemaOptions,
+  MalloyError,
+  SQLBlock,
+  StructDef,
+  registerDialect,
+} from '@malloydata/malloy';
+import {testRuntimeFor} from './runtimes';
+import './util/db-jest-matchers';
+import {DuckDBConnection} from '@malloydata/db-duckdb';
+
+const envDatabases = (
+  process.env['MALLOY_DATABASES'] ||
+  process.env['MALLOY_DATABASE'] ||
+  'duckdb'
+).split(',');
+
+let describe = globalThis.describe;
+if (!envDatabases.includes('duckdb')) {
+  describe = describe.skip;
+  describe.skip = describe;
+}
+
+async function getError<T>(promise: Promise<T>): Promise<Error | undefined> {
+  try {
+    await promise;
+  } catch (error) {
+    return error;
+  }
+  return undefined;
+}
+
+describe('experimental dialects', () => {
+  const duckdbX = 'duckdb_experimental';
+  class DuckdbXConnection extends DuckDBConnection {
+    public async fetchSchemaForSQLBlock(
+      sqlRef: SQLBlock,
+      options: FetchSchemaOptions
+    ): Promise<
+      | {structDef: StructDef; error?: undefined}
+      | {error: string; structDef?: undefined}
+    > {
+      const result = await super.fetchSchemaForSQLBlock(sqlRef, options);
+      if (result.error === undefined) {
+        return {structDef: {...result.structDef, dialect: duckdbX}};
+      }
+      return result;
+    }
+  }
+
+  const connection = new DuckdbXConnection(
+    duckdbX,
+    'test/data/duckdb/duckdb_test.db'
+  );
+
+  class DuckdbXDialect extends DuckDBDialect {
+    experimental = true;
+    name = duckdbX;
+  }
+
+  registerDialect(new DuckdbXDialect());
+  const runtime = testRuntimeFor(connection);
+  runtime.isTestRuntime = false; // Enables checking experimental dialects
+
+  test('generate an error when used without experiment enabled', async () => {
+    const error = await getError(
+      runtime.getModel(`
+        source: s is ${duckdbX}.sql('SELECT 1 as one')
+      `)
+    );
+    expect(error).not.toBeUndefined();
+    if (error !== undefined) {
+      const problems = (error as MalloyError).problems;
+      expect(problems.length).toBe(1);
+      expect(problems[0].message).toContain(
+        `##! experimental.dialect.${duckdbX}`
+      );
+    }
+  });
+
+  test('does not generate an error when used with experiment enabled', async () => {
+    await runtime.getModel(`
+      ##! experimental.dialect.${duckdbX}
+      source: s is ${duckdbX}.sql('SELECT 1 as one')
+    `);
+  });
+
+  afterAll(async () => {
+    await runtime.connection.close();
+    registerDialect(new DuckDBDialect());
+  });
+});