Tests for READ

server/engine/src/test/java/org/eclipse/lsp/cobol/usecases/TestCicsReadStatement.java

@@ -0,0 +1,86 @@
+/*
+ * Copyright (c) 2024 Broadcom.
+ * The term "Broadcom" refers to Broadcom Inc. and/or its subsidiaries.
+ *
+ * This program and the accompanying materials are made
+ * available under the terms of the Eclipse Public License 2.0
+ * which is available at https://www.eclipse.org/legal/epl-2.0/
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ *
+ * Contributors:
+ *    Broadcom, Inc. - initial API and implementation
+ *
+ */
+
+package org.eclipse.lsp.cobol.usecases;
+
+import com.google.common.collect.ImmutableList;
+import com.google.common.collect.ImmutableMap;
+import org.eclipse.lsp.cobol.common.error.ErrorSource;
+import org.eclipse.lsp.cobol.test.engine.UseCaseEngine;
+import org.eclipse.lsp.cobol.usecases.common.CICSTestUtils;
+import org.eclipse.lsp4j.Diagnostic;
+import org.eclipse.lsp4j.DiagnosticSeverity;
+import org.eclipse.lsp4j.Range;
+import org.junit.jupiter.api.Test;
+
+import java.util.*;
+
+/**
+ * Test CICS WEB command. Documentation link: <a
+ * href="https://www.ibm.com/docs/en/cics-ts/6.x?topic=summary-receive-zos-communications-server-default">WEB
+ * Command</a>
+ *
+ * <p>This class tests all variations of the WEB command found in the link above.
+ */
+
+public class TestCicsReadStatement {
+    // Main Building Blocks
+    private static final String BASE_TEXT =
+            "       IDENTIFICATION DIVISION.\n"
+                    + "       PROGRAM-ID. ABCDEF.\n"
+                    + "       DATA DIVISION.\n"
+                    + "       WORKING-STORAGE SECTION.\n"
+                    + "       PROCEDURE DIVISION.\n"
+                    + "            EXEC CICS \n"
+                    + "            END-EXEC.";
+
+    // Test Strings
+    private static final String READ_VALID_1 = "READ FILE(123) UNCOMMITTED INTO(123) RIDFLD(123) KEYLENGTH(123) GENERIC SYSID(123) LENGTH(123) DEBKEY EQUAL NOSUSPEND";
+    private static final String READ_VALID_2 = "READ FILE(123) INTO(123) RIDFLD(123)";
+
+
+    private static final String READ_INVALID_1 = "READ {RIDFLD(123)|errorOne}";
+
+    // Utility Functions
+    private static void noErrorTest(String newCommand) {
+        UseCaseEngine.runTest(getTestString(newCommand), ImmutableList.of(), ImmutableMap.of());
+    }
+
+    private static String getTestString(String newCommand) {
+        List<String> instances = Arrays.asList(newCommand.split("\\s"));
+        instances.replaceAll(String.join("", Collections.nCopies(12, " "))::concat);
+        ArrayList<String> base = new ArrayList<String>(Arrays.asList(BASE_TEXT.split("\n")));
+        base.addAll(base.size() - 1, instances);
+        return String.join("\n", base);
+    }
+
+    // Test Functions
+    @Test
+    void testReadValid1() {
+        noErrorTest(READ_VALID_1);
+    }
+
+    @Test
+    void testReadValid2() {
+        noErrorTest(READ_VALID_2);
+    }
+
+    @Test
+    void testReadInvalid1() {
+        HashMap<String, Diagnostic> expectedDiagnostics = new HashMap<>();
+        expectedDiagnostics.put("errorOne", new Diagnostic(new Range(), "Missing required option: FILE", DiagnosticSeverity.Error, ErrorSource.PARSING.getText()));
+        CICSTestUtils.errorTest(READ_INVALID_1, expectedDiagnostics);
+    }
+}