From 7796f6e9601cdec3e1377acbfa212e6cfb36f793 Mon Sep 17 00:00:00 2001
From: Zhijiang <wangzhijiang999@aliyun.com>
Date: Sat, 16 May 2020 01:00:37 +0800
Subject: [PATCH] [FLINK-17739][network][tests] Fix unstable
 ResultPartitionTest#testInitializeMoreStateThanBuffer

---
 .../flink/runtime/io/network/partition/ResultPartitionTest.java | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/partition/ResultPartitionTest.java b/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/partition/ResultPartitionTest.java
index 65bdeb5871b85..8c2df5c3bf4f7 100644
--- a/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/partition/ResultPartitionTest.java
+++ b/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/partition/ResultPartitionTest.java
@@ -61,6 +61,7 @@
 import static org.hamcrest.Matchers.contains;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertNull;
 import static org.junit.Assert.assertTrue;
 import static org.mockito.Matchers.eq;
 import static org.mockito.Mockito.mock;
@@ -509,6 +510,7 @@ public void testInitializeMoreStateThanBuffer() throws Exception {
 							Thread.sleep(5);
 						}
 					}
+					assertNull(view.getNextBuffer());
 				}
 				return null;
 			};