Fix bug to include parameters for call statements

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

Author: worksofliam

src/language/sql/statement.ts

@@ -550,7 +550,7 @@ export default class Statement {
 
 		let endIndex = i;
 
-		const isSubSelect = tokenIs(nextToken, `function`, `TABLE`) || tokenIs(nextToken, `function`, `LATERAL`) || (options.includeParameters && tokenIs(nextToken, `function`));
+		const isSubSelect = (tokenIs(nextToken, `function`, `TABLE`) || tokenIs(nextToken, `function`, `LATERAL`) || (options.includeParameters && tokenIs(nextToken, `function`)) && this.type !== StatementType.Call);
 
 		if (isSubSelect) {
 			sqlObj = this.getRefAtToken(i+2);
@@ -598,7 +598,6 @@ export default class Statement {
 		}
 
 		if (sqlObj) {
-
 			if (options.withSystemName !== true) {
 				// If the next token is not a clause.. we might have the alias
 				if (nextToken && this.tokens[nextIndex+1]) {

src/language/sql/tests/statements.test.ts

@@ -480,9 +480,13 @@ parserScenarios(`Object references`, ({newDoc}) => {
 
     const refsA = talksStatement.getObjectReferences();
     expect(refsA.length).toBe(1);
-    expect(refsA[0].tokens.length).toBe(1);
+
     expect(refsA[0].object.name).toBe(`create_Sql_sample`);
     expect(refsA[0].object.schema).toBeUndefined();
+
+    const tokens = refsA[0].tokens;
+    expect(tokens.length).toBe(4); // Includes the parameter tokens since it's a call
+    expect(tokens[tokens.length-1].type).toBe(`closebracket`);
   });
 
   test(`CALL: simple qualified`, () => {
@@ -498,9 +502,13 @@ parserScenarios(`Object references`, ({newDoc}) => {
 
     const refsA = talksStatement.getObjectReferences();
     expect(refsA.length).toBe(1);
-    expect(refsA[0].tokens.length).toBe(3);
+
     expect(refsA[0].object.name).toBe(`create_Sql_sample`);
     expect(refsA[0].object.schema).toBe(`"QSYS"`);
+
+    const tokens = refsA[0].tokens;
+    expect(tokens.length).toBe(6); // Includes the parameter tokens since it's a call
+    expect(tokens[tokens.length-1].type).toBe(`closebracket`);
   });
 
   test(`ALTER: with reference`, () => {