@malloydata/malloy 0.0.80-dev230830211713 → 0.0.80-dev230903192621

package/dist/lang/test/locations.spec.js

@@ -0,0 +1,722 @@
+"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 test_translator_1 = require("./test-translator");
+require("./parse-expects");
+describe('source locations', () => {
+    test('renamed source location', () => {
+        const source = (0, test_translator_1.markSource) `source: ${'na is a'}`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toParse();
+        expect((0, test_translator_1.getExplore)(m.modelDef, 'na').location).toMatchObject(source.locations[0]);
+    });
+    test('refined source location', () => {
+        const source = (0, test_translator_1.markSource) `source: ${'na is a {}'}`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toParse();
+        expect((0, test_translator_1.getExplore)(m.modelDef, 'na').location).toMatchObject(source.locations[0]);
+    });
+    test('location of defined dimension', () => {
+        const source = (0, test_translator_1.markSource) `source: na is a { dimension: ${'x is 1'} }`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const x = (0, test_translator_1.getFieldDef)(na, 'x');
+        expect(x.location).toMatchObject(source.locations[0]);
+    });
+    test('location of defined measure', () => {
+        const source = (0, test_translator_1.markSource) `source: na is a { measure: ${'x is count()'} }`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const x = (0, test_translator_1.getFieldDef)(na, 'x');
+        expect(x.location).toMatchObject(source.locations[0]);
+    });
+    test('location of defined query', () => {
+        const source = (0, test_translator_1.markSource) `source: na is a { query: ${'x is { group_by: y is 1 }'} }`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const x = (0, test_translator_1.getFieldDef)(na, 'x');
+        expect(x.location).toMatchObject(source.locations[0]);
+    });
+    test('location of defined field inside a query', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        query: x is {
+          group_by: ${'y is 1'}
+        }
+      }`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const x = (0, test_translator_1.getQueryField)(na, 'x');
+        const y = (0, test_translator_1.getFieldDef)(x.pipeline[0], 'y');
+        expect(y.location).toMatchObject(source.locations[0]);
+    });
+    test('location of filtered field inside a query', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        measure: y is count()
+        query: x is {
+          aggregate: ${'z is y { where: true }'}
+        }
+      }`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const x = (0, test_translator_1.getQueryField)(na, 'x');
+        const z = (0, test_translator_1.getFieldDef)(x.pipeline[0], 'z');
+        expect(z.location).toMatchObject(source.locations[0]);
+    });
+    test('location of field inherited from table', () => {
+        const source = (0, test_translator_1.markSource) `source: na is ${"table('aTable')"}`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const abool = (0, test_translator_1.getFieldDef)(na, 'abool');
+        expect(abool.location).toMatchObject(source.locations[0]);
+    });
+    test('location of field inherited from sql block', () => {
+        const source = (0, test_translator_1.markSource) `--- comment
+      sql: s is { select: ${'"""SELECT 1 as one """'} }
+      source: na is from_sql(s)
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toParse();
+        const compileSql = m.translate().compileSQL;
+        expect(compileSql).toBeDefined();
+        if (compileSql) {
+            m.update({
+                compileSQL: { [compileSql.name]: (0, test_translator_1.getSelectOneStruct)(compileSql) },
+            });
+            expect(m).toTranslate();
+            const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+            const one = (0, test_translator_1.getFieldDef)(na, 'one');
+            expect(one.location).isLocationIn(source.locations[0], source.code);
+        }
+    });
+    test('location of fields inherited from a query', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is from(
+        ${"table('aTable')"} -> {
+          group_by:
+            abool
+            ${'y is 1'}
+        }
+      )
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const abool = (0, test_translator_1.getFieldDef)(na, 'abool');
+        expect(abool.location).toMatchObject(source.locations[0]);
+        const y = (0, test_translator_1.getFieldDef)(na, 'y');
+        expect(y.location).toMatchObject(source.locations[1]);
+    });
+    test('location of named query', () => {
+        const source = (0, test_translator_1.markSource) `query: ${'q is a -> { project: * }'}`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const q = (0, test_translator_1.getExplore)(m.modelDef, 'q');
+        expect(q.location).toMatchObject(source.locations[0]);
+    });
+    test('location of field in named query', () => {
+        const source = (0, test_translator_1.markSource) `query: q is a -> { group_by: ${'b is 1'} }`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const q = (0, test_translator_1.getModelQuery)(m.modelDef, 'q');
+        const a = (0, test_translator_1.getFieldDef)(q.pipeline[0], 'b');
+        expect(a.location).toMatchObject(source.locations[0]);
+    });
+    test('location of named SQL block', () => {
+        const source = (0, test_translator_1.markSource) `${'sql: s is { select: """SELECT 1 as one""" }'}`;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toParse();
+        const compileSql = m.translate().compileSQL;
+        expect(compileSql).toBeDefined();
+        if (compileSql) {
+            m.update({
+                compileSQL: { [compileSql.name]: (0, test_translator_1.getSelectOneStruct)(compileSql) },
+            });
+            expect(m).toTranslate();
+            const s = m.sqlBlocks[0];
+            expect(s.location).isLocationIn(source.locations[0], source.code);
+        }
+    });
+    test('location of renamed field', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        rename: ${'bbool is abool'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const bbool = (0, test_translator_1.getFieldDef)(na, 'bbool');
+        expect(bbool.location).toMatchObject(source.locations[0]);
+    });
+    test('location of join on', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        join_one: ${'x is a { primary_key: abool } on abool'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const x = (0, test_translator_1.getFieldDef)(na, 'x');
+        expect(x.location).toMatchObject(source.locations[0]);
+    });
+    test('location of join with', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        join_one: ${'x is a { primary_key: astr } with astr'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const x = (0, test_translator_1.getFieldDef)(na, 'x');
+        expect(x.location).toMatchObject(source.locations[0]);
+    });
+    test('location of field in join', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        join_one: x is a {
+          primary_key: abool
+          dimension: ${'y is 1'}
+        } on abool
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        const na = (0, test_translator_1.getExplore)(m.modelDef, 'na');
+        const x = (0, test_translator_1.getJoinField)(na, 'x');
+        const y = (0, test_translator_1.getFieldDef)(x, 'y');
+        expect(y.location).toMatchObject(source.locations[0]);
+    });
+    // Since """ strings are not single tokens, I don't know how to do this.
+    // test("multi line sql block token span is correct", () => {
+    //   const sqlSource = `sql: { select: """// line 0\n//line 1\n// line 2""" }`;
+    //   const m = new TestTranslator(sqlSource);
+    //   expect(m).not.toParse();
+    //   const errList = m.errors().errors;
+    //   expect(errList[0].at?.range.end).toEqual({ line: 2, character: 11 });
+    // });
+    test('undefined query location', () => {
+        expect((0, test_translator_1.model) `query: ${'-> xyz'}`).translationToFailWith("Reference to undefined query 'xyz'");
+    });
+    test('undefined field reference', () => {
+        expect((0, test_translator_1.model) `query: a -> { group_by: ${'xyz'} }`).translationToFailWith("'xyz' is not defined");
+    });
+    test('bad query', () => {
+        expect((0, test_translator_1.model) `query: a -> { group_by: astr; ${'project: *'} }`).translationToFailWith(/Not legal in grouping query/);
+    });
+    test.skip('undefined field reference in top', () => {
+        expect((0, test_translator_1.model) `query: a -> { group_by: one is 1; top: 1 by ${'xyz'} }`).translationToFailWith("'xyz' is not defined");
+    });
+    test.skip('undefined field reference in order_by', () => {
+        expect((0, test_translator_1.model) `query: a -> { group_by: one is 1; order_by: ${'xyz'} }`).translationToFailWith("'xyz' is not defined");
+    });
+});
+describe('source references', () => {
+    test('reference to explore', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: ${'na is a'}
+      query: ${'na'} -> { project: * }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'exploreReference',
+            text: 'na',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to query in query', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: t is a {
+        query: ${'q is { project: * }'}
+      }
+      query: t -> ${'q'}
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'q',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to query in query (version 2)', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a { query: ${'x is { group_by: y is 1 }'} }
+      query: na -> ${'x'}
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'x',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to sql block', () => {
+        const source = (0, test_translator_1.markSource) `
+      ${'sql: s is {select:"""SELECT 1 as one"""}'}
+      source: na is from_sql(${'s'})
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toParse();
+        const compileSql = m.translate().compileSQL;
+        expect(compileSql).toBeDefined();
+        if (compileSql) {
+            m.update({
+                compileSQL: { [compileSql.name]: (0, test_translator_1.getSelectOneStruct)(compileSql) },
+            });
+            expect(m).toTranslate();
+            const ref = m.referenceAt(pos(source.locations[1]));
+            expect(ref).toMatchObject({
+                location: source.locations[1],
+                type: 'sqlBlockReference',
+                text: 's',
+                definition: {
+                    ...(0, test_translator_1.getSelectOneStruct)(compileSql),
+                    location: source.locations[0],
+                },
+            });
+        }
+    });
+    test('reference to query in from', () => {
+        const source = (0, test_translator_1.markSource) `
+      query: ${'q is a -> { project: * }'}
+      source: na is from(-> ${'q'})
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'queryReference',
+            text: 'q',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to query in query head', () => {
+        const source = (0, test_translator_1.markSource) `
+      query: ${'q is a -> { project: * }'}
+      query: q2 is -> ${'q'} -> { project: * }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'queryReference',
+            text: 'q',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to query in refined query', () => {
+        const source = (0, test_translator_1.markSource) `
+      query: ${'q is a -> { project: * }'}
+      query: q2 is -> ${'q'} { limit: 10 }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'queryReference',
+            text: 'q',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to field in expression', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is ${"table('aTable')"}
+      query: na -> { project: bbool is not ${'abool'} }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'abool',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to quoted field in expression', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        dimension: ${"`name` is 'name'"}
+      }
+      query: na -> { project: ${'`name`'} }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'name',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to joined field in expression', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        join_one: self is ${"table('aTable')"}
+          on astr = self.astr
+      }
+      query: na -> { project: bstr is self.${'astr'} }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'astr',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to joined join in expression', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        join_one: ${'self is a on astr = self.astr'}
+      }
+      query: na -> { project: bstr is ${'self'}.astr }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'joinReference',
+            text: 'self',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to field not in expression (group by)', () => {
+        const source = (0, test_translator_1.markSource) `
+      query: ${"table('aTable')"} -> { group_by: ${'abool'} }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'abool',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to field not in expression (project)', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is ${"table('aTable')"}
+      query: na -> { project: ${'abool'} }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'abool',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test.skip('reference to field in order by', () => {
+        const source = (0, test_translator_1.markSource) `
+      query: ${"table('aTable')"} -> {
+        group_by: abool
+        order_by: ${'abool'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'abool',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test.skip('reference to field in order by (output space)', () => {
+        const source = (0, test_translator_1.markSource) `
+      query: a -> {
+        group_by: ${'one is 1'}
+        order_by: ${'one'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'abool',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to field in aggregate', () => {
+        const source = (0, test_translator_1.markSource) `
+      query: a { measure: ${'c is count()'} } -> {
+        group_by: abool
+        aggregate: ${'c'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'c',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to field in measure', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: e is a {
+        measure: ${'c is count()'}
+        measure: c2 is ${'c'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'c',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test.skip('reference to field in top', () => {
+        const source = (0, test_translator_1.markSource) `
+      query: ${"table('aTable')"} -> {
+        group_by: abool
+        top: 10 by ${'abool'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'abool',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test.skip('reference to field in top (output space)', () => {
+        const source = (0, test_translator_1.markSource) `
+      query: a -> {
+        group_by: ${'one is 1'}
+        top: 10 by ${'one'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'abool',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to field in filter', () => {
+        const source = (0, test_translator_1.markSource) `
+      query: ${"table('aTable')"} -> {
+        group_by: abool
+        where: ${'abool'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'abool',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to field in aggregate source', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is ${"table('aTable')"}
+      query: na -> { aggregate: ai_sum is ${'ai'}.sum() }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'ai',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    function pos(location) {
+        return location.range.start;
+    }
+    test('reference to join in aggregate source', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        join_one: ${'self is a on astr = self.astr'}
+      }
+      query: na -> { aggregate: ai_sum is ${'self'}.sum(self.ai) }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'joinReference',
+            text: 'self',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to join in aggregate in expr', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is a {
+        join_one: ${'self is a on astr = self.astr'}
+      }
+      query: na -> { aggregate: ai_sum is self.sum(${'self'}.ai) }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'joinReference',
+            text: 'self',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to sourcein join', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: ${'exp1 is a'}
+      source: exp2 is a {
+        join_one: ${'exp1'} on astr = exp1.astr
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'exploreReference',
+            text: 'exp1',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to field in aggregate (in expr)', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is ${"table('aTable')"}
+      query: na -> { aggregate: ai_sum is sum(${'ai'}) }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'ai',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to field in rename', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: na is ${"table('aTable')"} {
+        rename: bbool is ${'abool'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'abool',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+    test('reference to field in join with', () => {
+        const source = (0, test_translator_1.markSource) `
+      source: exp1 is a { primary_key: astr }
+      source: exp2 is ${"table('aTable')"} {
+        join_one: exp1 with ${'astr'}
+      }
+    `;
+        const m = new test_translator_1.TestTranslator(source.code);
+        expect(m).toTranslate();
+        expect(m.referenceAt(pos(source.locations[1]))).toMatchObject({
+            location: source.locations[1],
+            type: 'fieldReference',
+            text: 'astr',
+            definition: {
+                location: source.locations[0],
+            },
+        });
+    });
+});
+//# sourceMappingURL=locations.spec.js.map