Tests to prove references to template (#320)

Signed-off-by: worksofliam <[email protected]>

Author: worksofliam

tests/suite/files.test.ts

@@ -92,14 +92,14 @@ test("ds_extname", async () => {
   expect(structDef.subItems.length).toBe(14);
 });
 
-test("ds_extname_no_alias", async () => {
+test("ds_extname", async () => {
   const lines = [
     `**free`,
     ``,
-    `Dcl-Ds dept ExtName('department') Qualified;`,
+    `Dcl-Ds Employee ExtName('EMPLOYEE') Qualified;`,
     `end-ds;`,
     ``,
-    `Dsply dept.deptname;`,
+    `Dsply Employee.empno;`,
     ``,
     `return;`
   ].join(`\n`);
@@ -109,11 +109,31 @@ test("ds_extname_no_alias", async () => {
   expect(cache.files.length).toBe(0);
   expect(cache.structs.length).toBe(1);
 
-  const dept = cache.find(`dept`);
-  expect(dept.subItems.length).toBe(5);
+  const structDef = cache.find(`employee`);
+  expect(structDef.name).toBe(`Employee`);
+  expect(structDef.subItems.length).toBe(14);
+});
 
-  expect(dept.subItems[0].name).toBe(`DEPTNO`);
-  expect(dept.subItems[1].name).toBe(`DEPTNAME`);
+test("ds_extname_template", async () => {
+  const lines = [
+    `**free`,
+    ``,
+    `Dcl-Ds dept ExtName('department') Qualified template end-ds;`,
+    ``,
+    `Dcl-DS dsExample qualified inz;`,
+    `  Field1 like(tmpDS.Field1) inz;`,
+    `  Field2 like(tmpDS.Field2) inz;`,
+    `END-DS`,
+    ``,
+    `return;`
+  ].join(`\n`);
+
+  const cache = await parser.getDocs(uri, lines, {withIncludes: true, ignoreCache: true});
+
+  expect(cache.structs.length).toBe(2);
+
+  const dept = cache.find(`dsExample`);
+  expect(dept.subItems.length).toBe(2);
 });
 
 test("ds_extname_alias", async () => {
@@ -187,4 +207,5 @@ test('file DS in a copy book', async () => {
   expect(someStruct.subItems.length).toBeGreaterThan(0);
 
   expect(someStruct.subItems.map(s => ({name: s.name, keyword: s.keyword}))).toMatchObject(globalStruct.subItems.map(s => ({name: s.name, keyword: s.keyword})));
-})
+});
+

tests/suite/linter.test.ts

@@ -3687,4 +3687,41 @@ test('allow special global subroutines', async () => {
   }, cache);
 
   expect(errors.length).toBe(2);
+});
+
+test("DS template references", async () => {
+  const lines = [
+    `**free`,
+    ``,
+    `Dcl-Ds dept ExtName('department') Qualified template end-ds;`,
+    ``,
+    `Dcl-DS dsExample qualified inz;`,
+    `  Field1 like(dept.DEPTNO) inz;`,
+    `  Field2 like(dept.DEPTNAME) inz;`,
+    `END-DS`,
+    ``,
+    `return;`
+  ].join(`\n`);
+
+  const cache = await parser.getDocs(uri, lines, {withIncludes: true, ignoreCache: true, collectReferences: true});
+  const { errors } = Linter.getErrors({ uri, content: lines }, {
+    NoUnreferenced: true,
+  }, cache);
+
+  const dept = cache.find(`dept`);
+  const deptno = dept.subItems.find(s => s.name === `DEPTNO`);
+  const deptname = dept.subItems.find(s => s.name === `DEPTNAME`);
+
+  expect(deptno).toBeDefined();
+  expect(deptname).toBeDefined();
+
+  expect(deptno.references.length).toBe(1);
+  expect(deptname.references.length).toBe(1);
+
+  expect(errors.length).toBe(3);
+
+  for (const err of errors) {
+    expect(lines.substring(err.offset.start, err.offset.end)).toContain(`inz`);
+  }
+
 });