Fix broken tests not expecting new exceptions

Signed-off-by: Richard Bair <rbair23@users.noreply.github.com>
hashgraph · Nov 22, 2024 · b45068b · b45068b
1 parent 69a20cd
commit b45068b
Showing 1 changed file with 10 additions and 5 deletions.
diff --git a/pbj-core/pbj-runtime/src/test/java/com/hedera/pbj/runtime/grpc/PipelinesTest.java b/pbj-core/pbj-runtime/src/test/java/com/hedera/pbj/runtime/grpc/PipelinesTest.java
@@ -177,7 +177,8 @@ void exceptionDuring_onNext_IsHandled() {
                     .build();
 
             pipeline.onSubscribe(mock(Flow.Subscription.class));
-            pipeline.onNext(Bytes.wrap("hello"));
+            final var data = Bytes.wrap("hello");
+            assertThatThrownBy(() -> pipeline.onNext(data)).isInstanceOf(RuntimeException.class);
             verify(replies).onError(any(RuntimeException.class));
         }
 
@@ -309,7 +310,8 @@ void exceptionDuring_onNext_IsHandled() {
                     .build();
 
             pipeline.onSubscribe(mock(Flow.Subscription.class));
-            pipeline.onNext(Bytes.wrap("hello"));
+            final var data = Bytes.wrap("hello");
+            assertThatThrownBy(() -> pipeline.onNext(data)).isInstanceOf(RuntimeException.class);
             verify(replies).onError(any(RuntimeException.class));
         }
 
@@ -448,7 +450,8 @@ void badRequestMapperCallsOnError() {
                     .build();
 
             pipeline.onSubscribe(mock(Flow.Subscription.class));
-            pipeline.onNext(Bytes.wrap("hello"));
+            final var data = Bytes.wrap("hello");
+            assertThatThrownBy(() -> pipeline.onNext(data)).isInstanceOf(RuntimeException.class);
             verify(replies).onError(ex);
         }
 
@@ -463,7 +466,8 @@ void badMethodCallsOnError() {
                     .build();
 
             pipeline.onSubscribe(mock(Flow.Subscription.class));
-            pipeline.onNext(Bytes.wrap("hello"));
+            final var data = Bytes.wrap("hello");
+            assertThatThrownBy(() -> pipeline.onNext(data)).isInstanceOf(RuntimeException.class);
             verify(replies).onError(ex);
         }
 
@@ -577,7 +581,8 @@ void badRequestMapperCallsOnError() {
                     .build();
 
             pipeline.onSubscribe(mock(Flow.Subscription.class));
-            pipeline.onNext(Bytes.wrap("hello"));
+            final var data = Bytes.wrap("hello");
+            assertThatThrownBy(() -> pipeline.onNext(data)).isInstanceOf(RuntimeException.class);
             verify(replies).onError(ex);
         }