Removing last statement with LexicalPreservingPrinter results in loss of indendation

[steve-thousand]

I am on javaparser-core 3.24.4 and am attempting to remove the last statement from a method, and it is resulting in a loss of indentation when using the LexicalPreservingPrinter.

@Test
void testRemoveLastStatement() {
  CompilationUnit cu = LexicalPreservingPrinter.setup(StaticJavaParser.parse(
      "class X {" + LineSeparator.SYSTEM +
          "  @Test" + LineSeparator.SYSTEM +
          "  public void testCase() {" + LineSeparator.SYSTEM +
          "    int x = 0;" + LineSeparator.SYSTEM +
          "    int y = 1;" + LineSeparator.SYSTEM +
          "  }" + LineSeparator.SYSTEM +
          "}" + LineSeparator.SYSTEM
  ));

  cu.getType(0).getMethods().get(0).getBody().get().getStatements().removeLast();

  assertEquals(
      "class X {" + LineSeparator.SYSTEM +
          "  @Test" + LineSeparator.SYSTEM +
          "  public void testCase() {" + LineSeparator.SYSTEM +
          "    int x = 0;" + LineSeparator.SYSTEM +
          "  }" + LineSeparator.SYSTEM +
          "}" + LineSeparator.SYSTEM, LexicalPreservingPrinter.print(cu.findCompilationUnit().get()));
}

This test fails, the actual output is

class X {
  @Test
  public void testCase() {
    int x = 0;
}
}

The same loss of indentation occurs when I attempt to replace the last statement

@Test
void testReplaceLastStatement() {
  CompilationUnit cu = LexicalPreservingPrinter.setup(StaticJavaParser.parse(
      "class X {" + LineSeparator.SYSTEM +
          "  @Test" + LineSeparator.SYSTEM +
          "  public void testCase() {" + LineSeparator.SYSTEM +
          "    int x = 0;" + LineSeparator.SYSTEM +
          "    int y = 1;" + LineSeparator.SYSTEM +
          "  }" + LineSeparator.SYSTEM +
          "}" + LineSeparator.SYSTEM
  ));

  cu.getType(0).getMethods().get(0).getBody().get().getStatements().removeLast();
  cu.getType(0).getMethods().get(0).getBody().get().getStatements().add(new ExpressionStmt(new VariableDeclarationExpr(
      PrimitiveType.intType(), "z")));

  assertEquals(
      "class X {" + LineSeparator.SYSTEM +
          "  @Test" + LineSeparator.SYSTEM +
          "  public void testCase() {" + LineSeparator.SYSTEM +
          "    int x = 0;" + LineSeparator.SYSTEM +
          "    int z;" + LineSeparator.SYSTEM +
          "  }" + LineSeparator.SYSTEM +
          "}" + LineSeparator.SYSTEM, LexicalPreservingPrinter.print(cu.findCompilationUnit().get()));
}

The above test fails, producing the actual output:

class X {
  @Test
  public void testCase() {
    int x = 0;
    int z;
}
}

Note that the same does not happen if I remove the first statement. The following test passes

@Test
void testRemoveFirstStatement() {
  CompilationUnit cu = LexicalPreservingPrinter.setup(StaticJavaParser.parse(
      "class X {" + LineSeparator.SYSTEM +
          "  @Test" + LineSeparator.SYSTEM +
          "  public void testCase() {" + LineSeparator.SYSTEM +
          "    int x = 0;" + LineSeparator.SYSTEM +
          "    int y = 1;" + LineSeparator.SYSTEM +
          "  }" + LineSeparator.SYSTEM +
          "}" + LineSeparator.SYSTEM
  ));

  cu.getType(0).getMethods().get(0).getBody().get().getStatements().removeFirst();

  assertEquals(
      "class X {" + LineSeparator.SYSTEM +
          "  @Test" + LineSeparator.SYSTEM +
          "  public void testCase() {" + LineSeparator.SYSTEM +
          "    int y = 1;" + LineSeparator.SYSTEM +
          "  }" + LineSeparator.SYSTEM +
          "}" + LineSeparator.SYSTEM, LexicalPreservingPrinter.print(cu.findCompilationUnit().get()));
}

[jlerbsc]

Hi @steve-thousand Thank for reporting this bug.