[FLINK-10364][tests] Fix instability in NonHAQueryableStateFsBackendI…

…TCase#testMapState This closes apache#6975.
mbalassi · Nov 5, 2018 · b9351fc · b9351fc
1 parent 31e0098
commit b9351fc
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/...src/test/java/org/apache/flink/queryablestate/itcases/AbstractQueryableStateTestBase.java b/...src/test/java/org/apache/flink/queryablestate/itcases/AbstractQueryableStateTestBase.java
@@ -836,9 +836,11 @@ public void processElement(Tuple2<Integer, Long> value, Context ctx, Collector<O
  false,
  executor);
 
- Tuple2<Integer, Long> value = future.get(deadline.timeLeft().toMillis(), TimeUnit.MILLISECONDS).get(key);
- assertEquals("Key mismatch", key, value.f0.intValue());
- if (expected == value.f1) {
+ Tuple2<Integer, Long> value =
+ future.get(deadline.timeLeft().toMillis(), TimeUnit.MILLISECONDS).get(key);
+
+ if (value != null && value.f0 != null && expected == value.f1) {
+ assertEquals("Key mismatch", key, value.f0.intValue());
  success = true;
  } else {
  // Retry