fix java test error

openmainframeproject · Jun 24, 2023 · 9ab9956 · 9ab9956
1 parent 9d1b53f
commit 9ab9956
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/...in/java/org/openmainframeproject/cobolcheck/features/testSuiteParser/TestSuiteParser.java b/...in/java/org/openmainframeproject/cobolcheck/features/testSuiteParser/TestSuiteParser.java
@@ -203,15 +203,20 @@ public List<String> getParsedTestSuiteLines(BufferedReader testSuiteReader,
                 continue;
             }
 
-            if (expectMockArguments
-            && CobolVerbs.isStartOrEndCobolVerb(testSuiteToken)) {
+            if (expectMockArguments && !expectUsing 
+            && (CobolVerbs.isCobolVerb(testSuiteToken)|| testSuiteToken.equals("END-MOCK"))) {
                 // NEW: In this case we expected cobol verbs and stop counting arguments
                 // update the keyword as fieldname was assumed
                 // keyword = Keywords.getKeywordFor(testSuiteToken, false);
                 expectMockArguments = false;
                 expectUsing = false;
-                handleEndOfMockStatement(testSuiteReader, testSuiteToken, false);
-                testSuiteToken = getNextTokenFromTestSuite(testSuiteReader);
+                if (!verifyInProgress) {
+                    ignoreCobolStatementAndFieldNameKeyAction = true;
+                    handleEndOfMockStatement(testSuiteReader, testSuiteToken, false);
+                }
+                if(testSuiteToken.equals("END-MOCK") ){
+                    testSuiteToken = getNextTokenFromTestSuite(testSuiteReader);
+                }
                 continue;
             }