@@ -0,0 +1,1701 @@
+/*%0A * Licensed to the Apache Software Foundation (ASF) under one or more%0A * contributor license agreements.  See the NOTICE file distributed with%0A * this work for additional information regarding copyright ownership.%0A * The ASF licenses this file to You under the Apache License, Version 2.0%0A * (the %22License%22); you may not use this file except in compliance with%0A * the License.  You may obtain a copy of the License at%0A *%0A *    http://www.apache.org/licenses/LICENSE-2.0%0A *%0A * Unless required by applicable law or agreed to in writing, software%0A * distributed under the License is distributed on an %22AS IS%22 BASIS,%0A * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.%0A * See the License for the specific language governing permissions and%0A * limitations under the License.%0A */%0A%0Apackage org.apache.flink.runtime.io.network.partition;%0A%0Aimport org.apache.flink.util.TestLogger;%0A%0Aimport org.junit.Test;%0A%0Aimport static org.hamcrest.Matchers.is;%0Aimport static org.junit.Assert.assertThat;%0A%0A/**%0A * Tests for the %7B@link ReleaseOnConsumptionResultPartitionTest%7D.%0A */%0Apublic class ReleaseOnConsumptionResultPartitionTest extends TestLogger %7B%0A%0A%09@Test%0A%09public void testConsumptionBasedPartitionRelease() %7B%0A%09%09final ResultPartitionManager manager = new ResultPartitionManager();%0A%09%09final ResultPartition partition = new ResultPartitionBuilder()%0A%09%09%09.setNumberOfSubpartitions(2)%0A%09%09%09.isReleasedOnConsumption(true)%0A%09%09%09.setResultPartitionManager(manager)%0A%09%09%09.build();%0A%0A%09%09manager.registerResultPartition(partition);%0A%0A%09%09partition.onConsumedSubpartition(0);%0A%09%09assertThat(partition.isReleased(), is(false));%0A%0A%09%09partition.onConsumedSubpartition(1);%0A%09%09assertThat(partition.isReleased(), is(true));%0A%09%7D%0A%7D%0A

Add new unit tests for ReleaseOnConsumptionResultPartition