191 wire simulator with test infra (#243)

Signed-off-by: georgi-l95 <[email protected]>

simulator/src/main/java/com/hedera/block/simulator/BlockStreamSimulator.java

@@ -18,7 +18,7 @@
 
 import static java.lang.System.Logger.Level.INFO;
 
-import com.hedera.pbj.runtime.ParseException;
+import com.hedera.block.simulator.exception.BlockSimulatorParsingException;
 import com.swirlds.config.api.Configuration;
 import com.swirlds.config.api.ConfigurationBuilder;
 import com.swirlds.config.extensions.sources.ClasspathFileConfigSource;
@@ -41,10 +41,10 @@ private BlockStreamSimulator() {}
      * @param args the arguments to be passed to the block stream simulator
      * @throws IOException if an I/O error occurs
      * @throws InterruptedException if the thread is interrupted
-     * @throws ParseException if a parse error occurs
+     * @throws BlockSimulatorParsingException if a parse error occurs
      */
     public static void main(String[] args)
-            throws IOException, InterruptedException, ParseException {
+            throws IOException, InterruptedException, BlockSimulatorParsingException {
 
         LOGGER.log(INFO, "Starting Block Stream Simulator");
 

simulator/src/main/java/com/hedera/block/simulator/BlockStreamSimulatorApp.java

@@ -17,13 +17,14 @@
 package com.hedera.block.simulator;
 
 import com.hedera.block.simulator.config.data.BlockStreamConfig;
+import com.hedera.block.simulator.exception.BlockSimulatorParsingException;
 import com.hedera.block.simulator.generator.BlockStreamManager;
 import com.hedera.block.simulator.grpc.PublishStreamGrpcClient;
 import com.hedera.hapi.block.stream.BlockItem;
-import com.hedera.pbj.runtime.ParseException;
 import com.swirlds.config.api.Configuration;
 import edu.umd.cs.findbugs.annotations.NonNull;
 import java.io.IOException;
+import java.util.concurrent.atomic.AtomicBoolean;
 import javax.inject.Inject;
 
 /** BlockStream Simulator App */
@@ -39,7 +40,7 @@ public class BlockStreamSimulatorApp {
 
     private final int delayBetweenBlockItems;
 
-    boolean isRunning = false;
+    private final AtomicBoolean isRunning = new AtomicBoolean(false);
 
     /**
      * Creates a new BlockStreamSimulatorApp instance.
@@ -66,14 +67,14 @@ public BlockStreamSimulatorApp(
      * Starts the block stream simulator.
      *
      * @throws InterruptedException if the thread is interrupted
-     * @throws ParseException if a parse error occurs
+     * @throws BlockSimulatorParsingException if a parse error occurs
      * @throws IOException if an I/O error occurs
      */
-    public void start() throws InterruptedException, ParseException, IOException {
+    public void start() throws InterruptedException, BlockSimulatorParsingException, IOException {
         int delayMSBetweenBlockItems = delayBetweenBlockItems / 1_000_000;
         int delayNSBetweenBlockItems = delayBetweenBlockItems % 1_000_000;
 
-        isRunning = true;
+        isRunning.set(true);
         LOGGER.log(System.Logger.Level.INFO, "Block Stream Simulator has started");
 
         boolean streamBlockItem = true;
@@ -113,12 +114,12 @@ public void start() throws InterruptedException, ParseException, IOException {
      * @return true if the block stream simulator is running, false otherwise
      */
     public boolean isRunning() {
-        return isRunning;
+        return isRunning.get();
     }
 
     /** Stops the block stream simulator. */
     public void stop() {
-        isRunning = false;
+        isRunning.set(false);
         LOGGER.log(System.Logger.Level.INFO, "Block Stream Simulator has stopped");
     }
 }

simulator/src/main/java/com/hedera/block/simulator/exception/BlockSimulatorParsingException.java

@@ -0,0 +1,29 @@
+/*
+ * Copyright (C) 2024 Hedera Hashgraph, LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.hedera.block.simulator.exception;
+
+/** Use this checked exception to represent a Block Simulator parsing exception. */
+public class BlockSimulatorParsingException extends Exception {
+    /**
+     * Constructs a new parsing exception with the specified detail message.
+     *
+     * @param message the detail message
+     */
+    public BlockSimulatorParsingException(String message) {
+        super(message);
+    }
+}

simulator/src/main/java/com/hedera/block/simulator/generator/BlockAsFileLargeDataSets.java

@@ -21,6 +21,7 @@
 
 import com.hedera.block.simulator.config.data.BlockStreamConfig;
 import com.hedera.block.simulator.config.types.GenerationMode;
+import com.hedera.block.simulator.exception.BlockSimulatorParsingException;
 import com.hedera.hapi.block.stream.Block;
 import com.hedera.hapi.block.stream.BlockItem;
 import com.hedera.pbj.runtime.ParseException;
@@ -59,7 +60,7 @@ public GenerationMode getGenerationMode() {
     }
 
     @Override
-    public BlockItem getNextBlockItem() throws IOException, ParseException {
+    public BlockItem getNextBlockItem() throws IOException, BlockSimulatorParsingException {
         if (currentBlock != null && currentBlock.items().size() > currentBlockItemIndex) {
             return currentBlock.items().get(currentBlockItemIndex++);
         } else {
@@ -74,22 +75,26 @@ public BlockItem getNextBlockItem() throws IOException, ParseException {
     }
 
     @Override
-    public Block getNextBlock() throws IOException, ParseException {
+    public Block getNextBlock() throws IOException, BlockSimulatorParsingException {
         currentBlockIndex++;
 
         String nextBlockFileName = String.format(formatString, currentBlockIndex);
         File blockFile = new File(blockstreamPath, nextBlockFileName);
 
-        if (blockFile.exists()) {
+        if (!blockFile.exists()) {
+            return null;
+        }
+
+        try {
             byte[] blockBytes = readFileBytes(blockFile.toPath());
 
             LOGGER.log(INFO, "Loading block: " + blockFile.getName());
 
             Block block = Block.PROTOBUF.parse(Bytes.wrap(blockBytes));
             LOGGER.log(INFO, "block loaded with items size= " + block.items().size());
             return block;
+        } catch (ParseException e) {
+            throw new BlockSimulatorParsingException(e.getMessage());
         }
-
-        return null; // No more blocks found
     }
 }

simulator/src/main/java/com/hedera/block/simulator/generator/BlockStreamManager.java

@@ -17,9 +17,9 @@
 package com.hedera.block.simulator.generator;
 
 import com.hedera.block.simulator.config.types.GenerationMode;
+import com.hedera.block.simulator.exception.BlockSimulatorParsingException;
 import com.hedera.hapi.block.stream.Block;
 import com.hedera.hapi.block.stream.BlockItem;
-import com.hedera.pbj.runtime.ParseException;
 import java.io.IOException;
 
 /** The block stream manager interface. */
@@ -36,17 +36,17 @@ public interface BlockStreamManager {
      * Get the next block item.
      *
      * @return the next block item
-     * @throws IOException if an I/O error occurs
-     * @throws ParseException if a parse error occurs
+     * @throws IOException if a I/O error occurs
+     * @throws BlockSimulatorParsingException if a parse error occurs
      */
-    BlockItem getNextBlockItem() throws IOException, ParseException;
+    BlockItem getNextBlockItem() throws IOException, BlockSimulatorParsingException;
 
     /**
      * Get the next block.
      *
      * @return the next block
-     * @throws IOException if an I/O error occurs
-     * @throws ParseException if a parse error occurs
+     * @throws IOException if a I/O error occurs
+     * @throws BlockSimulatorParsingException if a parse error occurs
      */
-    Block getNextBlock() throws IOException, ParseException;
+    Block getNextBlock() throws IOException, BlockSimulatorParsingException;
 }

simulator/src/main/java/module-info.java

@@ -3,6 +3,7 @@
 /** Runtime module of the simulator. */
 module com.hedera.block.simulator {
     exports com.hedera.block.simulator.config.data;
+    exports com.hedera.block.simulator.exception;
     exports com.hedera.block.simulator;
 
     requires static com.github.spotbugs.annotations;

simulator/src/test/java/com/hedera/block/simulator/BlockStreamSimulatorTest.java

@@ -19,10 +19,10 @@
 import static org.junit.jupiter.api.Assertions.*;
 import static org.mockito.Mockito.when;
 
+import com.hedera.block.simulator.exception.BlockSimulatorParsingException;
 import com.hedera.block.simulator.generator.BlockStreamManager;
 import com.hedera.block.simulator.grpc.PublishStreamGrpcClient;
 import com.hedera.hapi.block.stream.BlockItem;
-import com.hedera.pbj.runtime.ParseException;
 import com.swirlds.config.api.Configuration;
 import java.io.IOException;
 import java.nio.file.Paths;
@@ -63,13 +63,15 @@ void tearDown() {
     }
 
     @Test
-    void start_logsStartedMessage() throws InterruptedException, ParseException, IOException {
+    void start_logsStartedMessage()
+            throws InterruptedException, BlockSimulatorParsingException, IOException {
         blockStreamSimulator.start();
         assertTrue(blockStreamSimulator.isRunning());
     }
 
     @Test
-    void start_exitByBlockNull() throws InterruptedException, ParseException, IOException {
+    void start_exitByBlockNull()
+            throws InterruptedException, BlockSimulatorParsingException, IOException {
 
         BlockStreamManager blockStreamManager = Mockito.mock(BlockStreamManager.class);
         when(blockStreamManager.getNextBlockItem()).thenReturn(BlockItem.newBuilder().build());

simulator/src/test/java/com/hedera/block/simulator/generator/BlockAsDirBlockStreamManagerTest.java

@@ -22,7 +22,7 @@
 
 import com.hedera.block.simulator.config.data.BlockStreamConfig;
 import com.hedera.block.simulator.config.types.GenerationMode;
-import com.hedera.pbj.runtime.ParseException;
+import com.hedera.block.simulator.exception.BlockSimulatorParsingException;
 import java.io.IOException;
 import java.nio.file.Paths;
 import org.junit.jupiter.api.Test;
@@ -45,7 +45,7 @@ void getGenerationMode() {
     }
 
     @Test
-    void getNextBlockItem() throws IOException, ParseException {
+    void getNextBlockItem() throws IOException, BlockSimulatorParsingException {
         BlockStreamManager blockStreamManager =
                 getBlockAsDirBlockStreamManager(getAbsoluteFolder(rootFolder));
 
@@ -55,7 +55,7 @@ void getNextBlockItem() throws IOException, ParseException {
     }
 
     @Test
-    void getNextBlock() throws IOException, ParseException {
+    void getNextBlock() throws IOException, BlockSimulatorParsingException {
         BlockStreamManager blockStreamManager =
                 getBlockAsDirBlockStreamManager(getAbsoluteFolder(rootFolder));
 

simulator/src/test/java/com/hedera/block/simulator/generator/BlockAsFileBlockStreamManagerTest.java

@@ -20,7 +20,7 @@
 
 import com.hedera.block.simulator.config.data.BlockStreamConfig;
 import com.hedera.block.simulator.config.types.GenerationMode;
-import com.hedera.pbj.runtime.ParseException;
+import com.hedera.block.simulator.exception.BlockSimulatorParsingException;
 import java.io.IOException;
 import java.nio.file.Paths;
 import org.junit.jupiter.api.Test;
@@ -41,7 +41,7 @@ void getGenerationMode() {
     }
 
     @Test
-    void getNextBlock() throws IOException, ParseException {
+    void getNextBlock() throws IOException, BlockSimulatorParsingException {
         BlockStreamManager blockStreamManager =
                 getBlockAsFileBlockStreamManager(getAbsoluteFolder(gzRootFolder));
         for (int i = 0; i < 3000; i++) {
@@ -50,7 +50,7 @@ void getNextBlock() throws IOException, ParseException {
     }
 
     @Test
-    void getNextBlockItem() throws IOException, ParseException {
+    void getNextBlockItem() throws IOException, BlockSimulatorParsingException {
         BlockStreamManager blockStreamManager =
                 getBlockAsFileBlockStreamManager(getAbsoluteFolder(gzRootFolder));
         for (int i = 0; i < 35000; i++) {
@@ -59,7 +59,7 @@ void getNextBlockItem() throws IOException, ParseException {
     }
 
     @Test
-    void loadBlockBlk() throws IOException, ParseException {
+    void loadBlockBlk() throws IOException, BlockSimulatorParsingException {
         String blkRootFolder = "src/test/resources/block-0.0.3-blk/";
         BlockStreamManager blockStreamManager =
                 getBlockAsFileBlockStreamManager(getAbsoluteFolder(blkRootFolder));

simulator/src/test/java/com/hedera/block/simulator/generator/BlockAsFileLargeDataSetsTest.java

@@ -20,8 +20,8 @@
 
 import com.hedera.block.simulator.config.data.BlockStreamConfig;
 import com.hedera.block.simulator.config.types.GenerationMode;
+import com.hedera.block.simulator.exception.BlockSimulatorParsingException;
 import com.hedera.hapi.block.stream.BlockItem;
-import com.hedera.pbj.runtime.ParseException;
 import java.io.File;
 import java.io.IOException;
 import java.nio.file.Paths;
@@ -51,7 +51,7 @@ void getGenerationMode() {
     }
 
     @Test
-    void getNextBlock() throws IOException, ParseException {
+    void getNextBlock() throws IOException, BlockSimulatorParsingException {
         BlockStreamManager blockStreamManager =
                 getBlockAsFileLargeDatasetsBlockStreamManager(getAbsoluteFolder(rootFolder));
         for (int i = 0; i < filesInFolder; i++) {
@@ -62,7 +62,7 @@ void getNextBlock() throws IOException, ParseException {
     }
 
     @Test
-    void getNextBlockItem() throws IOException, ParseException {
+    void getNextBlockItem() throws IOException, BlockSimulatorParsingException {
         BlockStreamManager blockStreamManager =
                 getBlockAsFileLargeDatasetsBlockStreamManager(getAbsoluteFolder(rootFolder));
 

suites/src/main/java/com/hedera/block/suites/BaseSuite.java

@@ -127,7 +127,13 @@ public static GenericContainer<?> getConfiguration() {
         return blockNodeContainer;
     }
 
-    private static Configuration loadDefaultConfiguration() throws IOException {
+    /**
+     * Builds the default block simulator configuration
+     *
+     * @return default block simulator configuration
+     * @throws IOException if an I/O error occurs
+     */
+    protected static Configuration loadDefaultConfiguration() throws IOException {
         ConfigurationBuilder configurationBuilder =
                 ConfigurationBuilder.create()
                         .withSource(SystemEnvironmentConfigSource.getInstance())

suites/src/main/java/com/hedera/block/suites/persistence/DataPersistenceTestSuites.java

@@ -0,0 +1,39 @@
+/*
+ * Copyright (C) 2022-2024 Hedera Hashgraph, LLC
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package com.hedera.block.suites.persistence;
+
+import com.hedera.block.suites.persistence.positive.PositiveDataPersistenceTests;
+import org.junit.platform.suite.api.SelectClasses;
+import org.junit.platform.suite.api.Suite;
+
+/**
+ * Test suite for running data persistence tests, including both positive and negative test
+ * scenarios.
+ *
+ * <p>This suite aggregates the tests from {@link PositiveDataPersistenceTests}. The {@code @Suite}
+ * annotation allows running all selected classes in a single test run.
+ */
+@Suite
+@SelectClasses({PositiveDataPersistenceTests.class})
+public class DataPersistenceTestSuites {
+
+    /**
+     * Default constructor for the {@link DataPersistenceTestSuites} class. This constructor is
+     * empty as it does not need to perform any initialization.
+     */
+    public DataPersistenceTestSuites() {}
+}