[FLINK-5292] Add savepointToFile in OneInputStreamOperatorTestHarness.
This will be used mainly for migration testing purposes between
Flink 1.1 and Flink 1.2.
diff --git a/flink-streaming-java/src/test/java/org/apache/flink/streaming/util/OneInputStreamOperatorTestHarness.java b/flink-streaming-java/src/test/java/org/apache/flink/streaming/util/OneInputStreamOperatorTestHarness.java
index 12acfb6..b97b2cf 100644
--- a/flink-streaming-java/src/test/java/org/apache/flink/streaming/util/OneInputStreamOperatorTestHarness.java
+++ b/flink-streaming-java/src/test/java/org/apache/flink/streaming/util/OneInputStreamOperatorTestHarness.java
@@ -46,6 +46,8 @@
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
+import java.io.FileOutputStream;
+import java.io.ObjectOutputStream;
import java.io.Serializable;
import java.util.Collection;
import java.util.HashMap;
@@ -250,6 +252,13 @@
}
return snapshot;
}
+
+ public void snaphotToFile(StreamTaskState snapshot, String filename) throws Exception {
+ FileOutputStream out = new FileOutputStream(filename);
+ ObjectOutputStream oos = new ObjectOutputStream(out);
+ oos.writeObject(snapshot);
+ out.close();
+ }
/**
* Calls {@link org.apache.flink.streaming.api.operators.StreamOperator#restoreState(StreamTaskState)} ()}
