fix: test adjustments

Signed-off-by: Matt Peterson <matt.peterson@swirldslabs.com>

Author: mattp-swirldslabs

server/src/test/java/com/hedera/block/server/BlockStreamServiceIT.java

@@ -141,13 +141,13 @@ public void testPublishBlockStreamRegistrationAndExecution()
     }
 
     @Test
-    public void testSubscribeBlockStream() {
+    public void testSubscribeBlockStream() throws IOException {
 
         final ServiceStatus serviceStatus = new ServiceStatusImpl();
         serviceStatus.setWebServer(webServer);
 
         final var streamMediator =
-                new LiveStreamMediatorImpl(new HashMap<>(), blockWriter, serviceStatus);
+                new LiveStreamMediatorImpl(new ConcurrentHashMap<>(), blockWriter, serviceStatus);
 
         // Build the BlockStreamService
         final BlockStreamService blockStreamService =
@@ -171,6 +171,8 @@ public void testSubscribeBlockStream() {
         // Calling onNext() with a BlockItem
         streamObserver.onNext(publishStreamRequest);
 
+        verify(blockWriter, timeout(testTimeout).times(1)).write(blockItems.getFirst());
+
         final SubscribeStreamResponse subscribeStreamResponse =
                 SubscribeStreamResponse.newBuilder().setBlockItem(blockItems.getFirst()).build();
 

server/src/test/java/com/hedera/block/server/mediator/LiveStreamMediatorImplTest.java

@@ -56,6 +56,8 @@ public class LiveStreamMediatorImplTest {
     private final long TIMEOUT_THRESHOLD_MILLIS = 100L;
     private final long TEST_TIME = 1_719_427_664_950L;
 
+    private static final int testTimeout = 200;
+
     @Test
     public void testUnsubscribeEach() throws InterruptedException {
 
@@ -76,7 +78,7 @@ public void testUnsubscribeEach() throws InterruptedException {
                 streamMediator.isSubscribed(observer3),
                 "Expected the mediator to have observer3 subscribed");
 
-        Thread.sleep(50L);
+        Thread.sleep(testTimeout);
 
         streamMediator.unsubscribe(observer1);
         assertFalse(
@@ -152,9 +154,9 @@ public void testMediatorPublishEventToSubscribers() throws IOException, Interrup
         streamMediator.publish(blockItem);
 
         // Confirm each subscriber was notified of the new block
-        verify(streamObserver1, timeout(50).times(1)).onNext(subscribeStreamResponse);
-        verify(streamObserver2, timeout(50).times(1)).onNext(subscribeStreamResponse);
-        verify(streamObserver3, timeout(50).times(1)).onNext(subscribeStreamResponse);
+        verify(streamObserver1, timeout(testTimeout).times(1)).onNext(subscribeStreamResponse);
+        verify(streamObserver2, timeout(testTimeout).times(1)).onNext(subscribeStreamResponse);
+        verify(streamObserver3, timeout(testTimeout).times(1)).onNext(subscribeStreamResponse);
 
         // Confirm the BlockStorage write method was called
         verify(blockWriter).write(blockItem);
@@ -213,7 +215,7 @@ public void testOnCancelSubscriptionHandling() throws IOException {
         testConsumerBlockItemObserver.getOnCancel().run();
 
         // Verify the event made it to the consumer
-        verify(serverCallStreamObserver, timeout(50).times(1)).setOnCancelHandler(any());
+        verify(serverCallStreamObserver, timeout(testTimeout).times(1)).setOnCancelHandler(any());
 
         // Confirm the mediator unsubscribed the consumer
         assertFalse(streamMediator.isSubscribed(testConsumerBlockItemObserver));
@@ -245,7 +247,7 @@ public void testOnCloseSubscriptionHandling() throws IOException {
         testConsumerBlockItemObserver.getOnClose().run();
 
         // Verify the event made it to the consumer
-        verify(serverCallStreamObserver, timeout(50).times(1)).setOnCancelHandler(any());
+        verify(serverCallStreamObserver, timeout(testTimeout).times(1)).setOnCancelHandler(any());
 
         // Confirm the mediator unsubscribed the consumer
         assertFalse(streamMediator.isSubscribed(testConsumerBlockItemObserver));
@@ -274,7 +276,7 @@ public void testMediatorBlocksPublishAfterException() throws IOException, Interr
 
             // Confirm the BlockPersistenceHandler write method was only called
             // once despite the second block being published.
-            verify(blockWriter, timeout(50).times(1)).write(firstBlockItem);
+            verify(blockWriter, timeout(testTimeout).times(1)).write(firstBlockItem);
         }
     }