add test code for pruning enable

Signed-off-by: philosup <philosup@gmail.com>
hyperledger · Jan 20, 2025 · 86764b9 · 86764b9
1 parent bbf2065
commit 86764b9
Showing 1 changed file with 39 additions and 0 deletions.
diff --git a/besu/src/test/java/org/hyperledger/besu/cli/BesuCommandTest.java b/besu/src/test/java/org/hyperledger/besu/cli/BesuCommandTest.java
@@ -2705,4 +2705,43 @@ void helpOutputShouldDisplayCorrectDefaultValues() {
 
     assertThat(errorOutputString).isEmpty();
   }
+
+  @Test
+  void chainPruningEnabledWithQBFTShouldFailWhenChainPruningBlocksRetainedValueLessThanEpochLength()
+      throws IOException {
+
+    JsonObject genesis = GENESIS_VALID_JSON;
+    genesis.getJsonObject("config").put("qbft", new JsonObject().put("epochlength", 30000));
+
+    final Path genesisFile = createFakeGenesisFile(genesis);
+
+    parseCommand(
+        "--genesis-file",
+        genesisFile.toString(),
+        "--Xchain-pruning-enabled=true",
+        "--Xchain-pruning-blocks-retained=7200",
+        "--version-compatibility-protection=false");
+
+    assertThat(commandErrorOutput.toString(UTF_8))
+        .contains(
+            "--Xchain-pruning-blocks-retained must be >= epochlength(30000) for QBFT networks");
+  }
+
+  @Test
+  void chainPruningEnabledWithQBFT() throws IOException {
+
+    JsonObject genesis = GENESIS_VALID_JSON;
+    genesis.getJsonObject("config").put("qbft", new JsonObject().put("epochlength", 30000));
+
+    final Path genesisFile = createFakeGenesisFile(genesis);
+
+    parseCommand(
+        "--genesis-file",
+        genesisFile.toString(),
+        "--Xchain-pruning-enabled=true",
+        "--Xchain-pruning-blocks-retained=30000",
+        "--version-compatibility-protection=false");
+
+    assertThat(commandErrorOutput.toString(UTF_8)).isEmpty();
+  }
 }