Repository: flink
Updated Branches:
  refs/heads/master fdf436099 -> d353895ba


http://git-wip-us.apache.org/repos/asf/flink/blob/d353895b/flink-tests/src/test/java/org/apache/flink/test/checkpointing/StreamFaultToleranceTestBase.java
----------------------------------------------------------------------
diff --git 
a/flink-tests/src/test/java/org/apache/flink/test/checkpointing/StreamFaultToleranceTestBase.java
 
b/flink-tests/src/test/java/org/apache/flink/test/checkpointing/StreamFaultToleranceTestBase.java
index 530bae9..67c05e5 100644
--- 
a/flink-tests/src/test/java/org/apache/flink/test/checkpointing/StreamFaultToleranceTestBase.java
+++ 
b/flink-tests/src/test/java/org/apache/flink/test/checkpointing/StreamFaultToleranceTestBase.java
@@ -23,8 +23,10 @@ import org.apache.flink.configuration.Configuration;
 import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;
 import org.apache.flink.test.util.ForkableFlinkMiniCluster;
 
+import org.apache.flink.test.util.TestUtils;
 import org.apache.flink.util.TestLogger;
 import org.junit.AfterClass;
+import org.junit.Assert;
 import org.junit.BeforeClass;
 import org.junit.Test;
 
@@ -89,7 +91,7 @@ public abstract class StreamFaultToleranceTestBase extends 
TestLogger {
         * followed by the checks in {@link #postSubmit}.
         */
        @Test
-       public void runCheckpointedProgram() {
+       public void runCheckpointedProgram() throws Exception {
                try {
                        StreamExecutionEnvironment env = 
StreamExecutionEnvironment.createRemoteEnvironment(
                                        "localhost", 
cluster.getLeaderRPCPort());
@@ -99,13 +101,13 @@ public abstract class StreamFaultToleranceTestBase extends 
TestLogger {
 
                        testProgram(env);
 
-                       env.execute();
+                       TestUtils.tryExecute(env, "Fault Tolerance Test");
 
                        postSubmit();
                }
                catch (Exception e) {
                        e.printStackTrace();
-                       fail(e.getMessage());
+                       Assert.fail(e.getMessage());
                }
        }
 

Reply via email to