pnowojski commented on a change in pull request #6809: [FLINK-10491][network] Pass BufferPoolOwner in the constructor of LocalBufferPool URL: https://github.com/apache/flink/pull/6809#discussion_r226285789
########## File path: flink-runtime/src/test/java/org/apache/flink/runtime/io/network/partition/ResultPartitionTest.java ########## @@ -205,6 +216,66 @@ protected void testAddOnPartition(final ResultPartitionType pipelined) } } + @Test + public void testReleaseMemoryOnBlockingPartition() throws Exception { + testReleaseMemory(ResultPartitionType.BLOCKING); + } + + @Test + public void testReleaseMemoryOnPipelinedPartition() throws Exception { + testReleaseMemory(ResultPartitionType.PIPELINED); + } + + /** + * Tests {@link ResultPartition#releaseMemory(int)} on a working partition. + * + * @param resultPartitionType the result partition type to set up + */ + private void testReleaseMemory(final ResultPartitionType resultPartitionType) throws Exception { + final int numBuffers = 10; + final NetworkEnvironment network = new NetworkEnvironment( + new NetworkBufferPool(numBuffers, 128), + new LocalConnectionManager(), + new ResultPartitionManager(), + new TaskEventDispatcher(), + new KvStateRegistry(), + null, + null, + IOManager.IOMode.SYNC, + 0, + 0, + 2, + 8, + true); + final ResultPartitionConsumableNotifier notifier = mock(ResultPartitionConsumableNotifier.class); + final ResultPartition resultPartition = spy(createPartition(notifier, resultPartitionType, false)); Review comment: Same here. Do not use mockito for that kind of stuff. Instead of checking whether some exact method was called or not, and how many times with what arguments (those things can change while the code still remain valid while the mockito tests will start failing), check the state of the `NetworkBufferPool` - how many buffers are available after some kind of operation etc. ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services