Update CobolGenerator.java

Signed-off-by: Søren Andersen <[email protected]>
openmainframeproject · Apr 17, 2024 · 5feb40e · 5feb40e
1 parent 14f5165
commit 5feb40e
Showing 1 changed file with 16 additions and 5 deletions.
diff --git a/...ain/java/org/openmainframeproject/cobolcheck/features/testSuiteParser/CobolGenerator.java b/...ain/java/org/openmainframeproject/cobolcheck/features/testSuiteParser/CobolGenerator.java
@@ -75,17 +75,28 @@ static void addStartAndEndTags(List<String> lines){
             lines.add(getInjectEndTagComment());
     }
 
-    static List<String> generateWhenOtherLines(String identifier, String type, List<String> commentLines, List<String> bodyLines){
+    static List<String> generateWhenOtherLines(String identifier, String type, List<String> commentLines, List<String> bodyLines) {
         List<String> lines = new ArrayList<>();
-        if(type.equals(Constants.SECTION_TOKEN)) 
+        if (type.equals(Constants.SECTION_TOKEN))
             lines.add(String.format(WHEN_OTHER_SECTION_HEADER_FORMAT, identifier));
-        else lines.add(String.format(PARAGRAPH_HEADER_FORMAT, identifier));
+        else
+            lines.add(String.format(PARAGRAPH_HEADER_FORMAT, identifier));
+
         if (commentLines != null)
             lines.addAll(commentLines);
-        if (bodyLines != null)
+
+        if (bodyLines != null) {
             lines.addAll(bodyLines);
-        lines.add(ENDING_PERIOD);
+            // Check if the last line in bodyLines ends with a period
+            if (!bodyLines.get(bodyLines.size() - 1).endsWith(".")) {
+                lines.add(ENDING_PERIOD);
+            }
+        } else {
+            // If bodyLines is null, add ENDING_PERIOD
+            lines.add(ENDING_PERIOD);
+        }
         return lines;
     }
 
+
 }