[FLINK-11904][tests]Improve MemoryStateBackendTest by using JUnit's P…

…arameterized This closes apache#7984.
static-max · Mar 14, 2019 · 6c7ddf3 · 6c7ddf3
1 parent b6b4baf
commit 6c7ddf3
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 32 deletions.
diff --git a/flink-runtime/src/test/java/org/apache/flink/runtime/state/AsyncMemoryStateBackendTest.java b/flink-runtime/src/test/java/org/apache/flink/runtime/state/AsyncMemoryStateBackendTest.java
diff --git a/flink-runtime/src/test/java/org/apache/flink/runtime/state/MemoryStateBackendTest.java b/flink-runtime/src/test/java/org/apache/flink/runtime/state/MemoryStateBackendTest.java
@@ -22,19 +22,29 @@
 
 import org.junit.Ignore;
 import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.Parameterized;
+
+import java.util.Arrays;
+import java.util.List;
 
 /**
  * Tests for the {@link org.apache.flink.runtime.state.memory.MemoryStateBackend}.
  */
+@RunWith(Parameterized.class)
 public class MemoryStateBackendTest extends StateBackendTestBase<MemoryStateBackend> {
 
- @Override
- protected MemoryStateBackend getStateBackend() throws Exception {
- return new MemoryStateBackend(useAsyncMode());
+ @Parameterized.Parameters(name = "useAsyncmode")
+ public static List<Boolean> modes() {
+ return Arrays.asList(true, false);
  }
 
- protected boolean useAsyncMode() {
- return false;
+ @Parameterized.Parameter
+ public boolean useAsyncmode;
+
+ @Override
+ protected MemoryStateBackend getStateBackend() {
+ return new MemoryStateBackend(useAsyncmode);
  }
 
  @Override