diff --git a/server/src/test/java/com/hedera/block/server/BlockStreamServiceTest.java b/server/src/test/java/com/hedera/block/server/BlockStreamServiceTest.java index b2593e65e..305293bea 100644 --- a/server/src/test/java/com/hedera/block/server/BlockStreamServiceTest.java +++ b/server/src/test/java/com/hedera/block/server/BlockStreamServiceTest.java @@ -42,6 +42,7 @@ import io.helidon.config.Config; import io.helidon.config.MapConfigSource; import io.helidon.config.spi.ConfigSource; +import io.helidon.webserver.grpc.GrpcService; import java.io.IOException; import java.nio.file.Files; import java.nio.file.Path; @@ -49,10 +50,6 @@ import java.util.List; import java.util.Map; import java.util.Optional; -import java.util.function.BiConsumer; - -import io.helidon.webserver.Routing; -import io.helidon.webserver.grpc.GrpcService; import org.junit.jupiter.api.AfterEach; import org.junit.jupiter.api.BeforeEach; import org.junit.jupiter.api.Test; @@ -266,9 +263,14 @@ public void testUpdateInvokesRoutingWithLambdas() { GrpcService.Routing routing = mock(GrpcService.Routing.class); blockStreamService.update(routing); - verify(routing, timeout(50).times(1)).bidi(eq(CLIENT_STREAMING_METHOD_NAME), any(ServerCalls.BidiStreamingMethod.class)); - verify(routing, timeout(50).times(1)).serverStream(eq(SERVER_STREAMING_METHOD_NAME), any(ServerCalls.ServerStreamingMethod.class)); - verify(routing, timeout(50).times(1)).unary(eq(SINGLE_BLOCK_METHOD_NAME), any(ServerCalls.UnaryMethod.class)); + verify(routing, timeout(50).times(1)) + .bidi(eq(CLIENT_STREAMING_METHOD_NAME), any(ServerCalls.BidiStreamingMethod.class)); + verify(routing, timeout(50).times(1)) + .serverStream( + eq(SERVER_STREAMING_METHOD_NAME), + any(ServerCalls.ServerStreamingMethod.class)); + verify(routing, timeout(50).times(1)) + .unary(eq(SINGLE_BLOCK_METHOD_NAME), any(ServerCalls.UnaryMethod.class)); } private BlockPersistenceHandler buildBlockPersistenceHandler()