[FLINK-11280] [state backends, tests] Do not set current key before r…

…estore in TtlStateTestBase tests
tongcheng-elong · Jan 8, 2019 · 31685a3 · 31685a3
1 parent f39fd29
commit 31685a3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/flink-runtime/src/test/java/org/apache/flink/runtime/state/ttl/StateBackendTestContext.java b/flink-runtime/src/test/java/org/apache/flink/runtime/state/ttl/StateBackendTestContext.java
@@ -76,9 +76,8 @@ void createAndRestoreKeyedStateBackend() {
  keyedStateBackend = stateBackend.createKeyedStateBackend(
  env, new JobID(), "test", StringSerializer.INSTANCE, 10,
  new KeyGroupRange(0, 9), env.getTaskKvStateRegistry(), timeProvider);
- keyedStateBackend.setCurrentKey("defaultKey");
  } catch (Exception e) {
- throw new RuntimeException("unexpected");
+ throw new RuntimeException("unexpected", e);
  }
  }
 

diff --git a/flink-runtime/src/test/java/org/apache/flink/runtime/state/ttl/TtlStateTestBase.java b/flink-runtime/src/test/java/org/apache/flink/runtime/state/ttl/TtlStateTestBase.java
@@ -113,6 +113,7 @@ private void initTest(StateTtlConfig ttlConfig) throws Exception {
  this.ttlConfig = ttlConfig;
  sbetc.createAndRestoreKeyedStateBackend();
  sbetc.restoreSnapshot(null);
+ sbetc.setCurrentKey("defaultKey");
  createState();
  ctx().initTestValues();
  }
@@ -129,6 +130,7 @@ private void takeAndRestoreSnapshot() throws Exception {
  KeyedStateHandle snapshot = sbetc.takeSnapshot();
  sbetc.createAndRestoreKeyedStateBackend();
  sbetc.restoreSnapshot(snapshot);
+ sbetc.setCurrentKey("defaultKey");
  createState();
  }
 
@@ -397,6 +399,7 @@ public void testRestoreTtlAndRegisterNonTtlStateCompatFailure() throws Exception
  sbetc.createAndRestoreKeyedStateBackend();
 
  sbetc.restoreSnapshot(snapshot);
+ sbetc.setCurrentKey("defaultKey");
  sbetc.createState(ctx().createStateDescriptor(), "");
  }