You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@flink.apache.org by ch...@apache.org on 2022/10/20 17:30:30 UTC
[flink] 01/02: [hotfix][tests] Port SavepointWriterTest to JUnit 5
This is an automated email from the ASF dual-hosted git repository.
chesnay pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/flink.git
commit 7941052c1e9bb8b2c7da010c2990ecfdbdaba941
Author: Chesnay Schepler <ch...@apache.org>
AuthorDate: Mon Oct 17 12:29:46 2022 +0200
[hotfix][tests] Port SavepointWriterTest to JUnit 5
---
.../flink/state/api/SavepointWriterTest.java | 34 +++++-----------------
1 file changed, 8 insertions(+), 26 deletions(-)
diff --git a/flink-libraries/flink-state-processing-api/src/test/java/org/apache/flink/state/api/SavepointWriterTest.java b/flink-libraries/flink-state-processing-api/src/test/java/org/apache/flink/state/api/SavepointWriterTest.java
index fb0372dd559..41bc033bf87 100644
--- a/flink-libraries/flink-state-processing-api/src/test/java/org/apache/flink/state/api/SavepointWriterTest.java
+++ b/flink-libraries/flink-state-processing-api/src/test/java/org/apache/flink/state/api/SavepointWriterTest.java
@@ -22,43 +22,25 @@ import org.apache.flink.api.common.RuntimeExecutionMode;
import org.apache.flink.configuration.Configuration;
import org.apache.flink.configuration.ExecutionOptions;
import org.apache.flink.configuration.StateBackendOptions;
-import org.apache.flink.state.api.functions.KeyedStateBootstrapFunction;
import org.apache.flink.state.api.utils.CustomStateBackendFactory;
-import org.apache.flink.streaming.api.datastream.DataStream;
import org.apache.flink.streaming.api.environment.StreamExecutionEnvironment;
-import org.junit.Test;
+import org.junit.jupiter.api.Test;
+
+import static org.assertj.core.api.Assertions.assertThatThrownBy;
/** Tests for the savepoint writer. */
-public class SavepointWriterTest {
+class SavepointWriterTest {
- @Test(expected = CustomStateBackendFactory.ExpectedException.class)
- public void testCustomStateBackend() throws Exception {
+ @Test
+ void testCustomStateBackend() {
StreamExecutionEnvironment env = StreamExecutionEnvironment.getExecutionEnvironment();
Configuration configuration = new Configuration();
configuration.set(
StateBackendOptions.STATE_BACKEND,
CustomStateBackendFactory.class.getCanonicalName());
configuration.set(ExecutionOptions.RUNTIME_MODE, RuntimeExecutionMode.BATCH);
- env.configure(configuration);
-
- DataStream<String> input = env.fromElements("");
-
- StateBootstrapTransformation<String> transformation =
- OperatorTransformation.bootstrapWith(input)
- .keyBy(element -> element)
- .transform(new Bootstrapper());
-
- SavepointWriter.newSavepoint(128)
- .withOperator(OperatorIdentifier.forUid("uid"), transformation)
- .write("file:///tmp/path");
-
- env.execute();
- }
-
- private static class Bootstrapper extends KeyedStateBootstrapFunction<String, String> {
-
- @Override
- public void processElement(String value, Context ctx) throws Exception {}
+ assertThatThrownBy(() -> env.configure(configuration))
+ .isInstanceOf(CustomStateBackendFactory.ExpectedException.class);
}
}