wip: try reordering the operations on the test

Signed-off-by: Matt Peterson <[email protected]>
hashgraph · Aug 1, 2024 · a08c333 · a08c333
1 parent 2febc51
commit a08c333
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/server/src/test/java/com/hedera/block/server/mediator/LiveStreamMediatorImplTest.java b/server/src/test/java/com/hedera/block/server/mediator/LiveStreamMediatorImplTest.java
@@ -224,12 +224,12 @@ public void testOnCancelSubscriptionHandling() throws IOException {
         final List<BlockItem> blockItems = generateBlockItems(1);
         streamMediator.publishEvent(blockItems.getFirst());
 
-        // Verify the event made it to the consumer
-        verify(serverCallStreamObserver, timeout(50).times(1)).setOnCancelHandler(any());
-
         // Simulate the consumer cancelling the stream
         testConsumerBlockItemObserver.getOnCancel().run();
 
+        // Verify the event made it to the consumer
+        verify(serverCallStreamObserver, timeout(50).times(1)).setOnCancelHandler(any());
+
         // Confirm the mediator unsubscribed the consumer
         assertFalse(streamMediator.isSubscribed(testConsumerBlockItemObserver));
     }
@@ -258,12 +258,12 @@ public void testOnCloseSubscriptionHandling() throws IOException {
         final List<BlockItem> blockItems = generateBlockItems(1);
         streamMediator.publishEvent(blockItems.getFirst());
 
-        // Verify the event made it to the consumer
-        verify(serverCallStreamObserver, timeout(50).times(1)).setOnCancelHandler(any());
-
         // Simulate the consumer completing the stream
         testConsumerBlockItemObserver.getOnClose().run();
 
+        // Verify the event made it to the consumer
+        verify(serverCallStreamObserver, timeout(50).times(1)).setOnCancelHandler(any());
+
         // Confirm the mediator unsubscribed the consumer
         assertFalse(streamMediator.isSubscribed(testConsumerBlockItemObserver));
     }