1 files changed, 2 insertions, 0 deletions
diff --git a/streaming/src/test/java/JavaAPISuite.java b/streaming/src/test/java/JavaAPISuite.java
index 8c94e13e65..c84e7331c7 100644
--- a/streaming/src/test/java/JavaAPISuite.java
+++ b/streaming/src/test/java/JavaAPISuite.java
@@ -34,12 +34,14 @@ public class JavaAPISuite implements Serializable {
   @Before
   public void setUp() {
     ssc = new JavaStreamingContext("local[2]", "test", new Duration(1000));
+    ssc.checkpoint("checkpoint", new Duration(1000));
   }
 
   @After
   public void tearDown() {
     ssc.stop();
     ssc = null;
+
     // To avoid Akka rebinding to the same port, since it doesn't unbind immediately on shutdown
     System.clearProperty("spark.master.port");
   }