refactor: code

Signed-off-by: Mustafa Uzun <mustafa.uzun@limechain.tech>

Author: mustafauzunn

tools-and-tests/simulator/src/main/java/org/hiero/block/simulator/mode/impl/PublisherClientModeHandler.java

@@ -120,6 +120,8 @@ private void millisPerBlockStreaming() throws IOException, InterruptedException,
             Consumer<PublishStreamResponse> publishStreamResponseConsumer = publishStreamResponseAtomicReference::set;
 
             if (!publishStreamGrpcClient.streamBlock(nextBlock, publishStreamResponseConsumer)) {
+                // TODO: how we would simulate starting a new stream? Creating a new instance of publishStreamGrpcClient?
+                //  The same for blockStreamManager because we would want to start from before the failed block for example?
                 publishStreamGrpcClient.shutdown();
                 if (publishStreamResponseAtomicReference.get().getEndStream().getStatus().equals(Code.SUCCESS)) {
                     publishStreamGrpcClient.init();

tools-and-tests/simulator/src/test/java/org/hiero/block/simulator/mode/PublisherClientModeHandlerTest.java

@@ -3,7 +3,6 @@
 
 import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.mockito.ArgumentMatchers.any;
-import static org.mockito.ArgumentMatchers.eq;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.never;
 import static org.mockito.Mockito.times;
@@ -19,7 +18,6 @@
 import java.util.Collections;
 import java.util.Map;
 import java.util.function.Consumer;
-import org.hiero.block.api.protoc.PublishStreamResponse;
 import org.hiero.block.simulator.TestUtils;
 import org.hiero.block.simulator.config.data.BlockStreamConfig;
 import org.hiero.block.simulator.config.types.StreamingMode;