RocMarshal commented on code in PR #21999:
URL: https://github.com/apache/flink/pull/21999#discussion_r1116555777


##########
flink-streaming-java/src/test/java/org/apache/flink/streaming/runtime/tasks/StreamTaskTest.java:
##########
@@ -362,30 +357,28 @@ private void testSyncSavepointWithEndInput(
                 "savepointResult");
         harness.processAll();
 
-        Assert.assertEquals(expectEndInput, 
TestBoundedOneInputStreamOperator.isInputEnded());
+        
assertThat(TestBoundedOneInputStreamOperator.isInputEnded()).isEqualTo(expectEndInput);
     }
 
     @Test
-    public void testCleanUpExceptionSuppressing() throws Exception {
+    void testCleanUpExceptionSuppressing() throws Exception {
         try (StreamTaskMailboxTestHarness<String> testHarness =
                 new 
StreamTaskMailboxTestHarnessBuilder<>(OneInputStreamTask::new, STRING_TYPE_INFO)
                         .addInput(STRING_TYPE_INFO)
                         .setupOutputForSingletonOperatorChain(new 
FailingTwiceOperator())
                         .build()) {
 
-            try {
-                testHarness.processElement(new StreamRecord<>("Doesn't 
matter", 0));
-                throw new RuntimeException("Expected an exception but ran 
successfully");
-            } catch (Exception ex) {
-                ExceptionUtils.assertThrowable(ex, 
ExpectedTestException.class);
-            }
+            assertThatThrownBy(
+                            () -> {
+                                testHarness.processElement(new 
StreamRecord<>("Doesn't matter", 0));
+                                throw new RuntimeException(

Review Comment:
   It's the old logic line from migration. but redundant.
   This lines rollback to the original lines.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]

Reply via email to