Fix test "should validate complex code without errors"

Author: neuroevolutus

server/src/server.ts

@@ -513,33 +513,35 @@ function inferBasicType(node: SyntaxNode): string | null {
 
 function validateWhereClauseStructure(node: SyntaxNode): Diagnostic[] {
   return validateNodeBy(node, (currentNode) => {
-    let diagnostics: Diagnostic[] = [];
-    if (currentNode.type === "where") {
-      // Check for proper "; identifier = expression" structure
-      let hasProperStructure = false;
-
-      for (let i = 0; i < currentNode.childCount; i++) {
-        const child = currentNode.child(i);
-        if (child?.type === ";" && i + 1 < currentNode.childCount) {
-          const nextChild = currentNode.child(i + 1);
-
-          if (nextChild?.child(0)?.type === "pattern" && nextChild?.child(1)?.type === "=") {
-            hasProperStructure = true;
-            break;
-          }
-        }
-      }
+    // Lazily compute diagnostics
+    const getDiagnostics = () => {
+      return [{
+        severity: DiagnosticSeverity.Error,
+        range: nodeToRange(currentNode),
+        message: 'Invalid where clause.\n'
+          + 'Expected:\n'
+          + '\t"; pattern = expression"\n'
+          + '\tor\n'
+          + '\t"; id = data_type"',
+        source: "scrapscript",
+      }];
+    };
 
-      if (!hasProperStructure) {
-        diagnostics.push({
-          severity: DiagnosticSeverity.Error,
-          range: nodeToRange(currentNode),
-          message: 'Invalid where clause. Expected "; identifier = expression"',
-          source: "scrapscript",
-        });
+    if (currentNode.type === "where") {
+      // Node should have form "id; ((id = data_type)|(pattern = expression))"
+      if (currentNode.childCount !== 3) return getDiagnostics();
+      // Expect semicolon in middle
+      const semicolonPosition = 1;
+      if (currentNode.child(semicolonPosition)?.type !== ";") return getDiagnostics();
+      // Expect declaration or type declaration at the end
+      const declarationPosition = 2;
+      if (currentNode.child(declarationPosition)?.type !== "declaration" &&
+        currentNode.child(declarationPosition)?.type !== "type_declaration") {
+        return getDiagnostics();
       }
+      return [];
     }
-    return diagnostics;
+    return [];
   });
 }