[hotfix][test] Simplify RecordWriterTest
Project: http://git-wip-us.apache.org/repos/asf/flink/repo Commit: http://git-wip-us.apache.org/repos/asf/flink/commit/ec7934ef Tree: http://git-wip-us.apache.org/repos/asf/flink/tree/ec7934ef Diff: http://git-wip-us.apache.org/repos/asf/flink/diff/ec7934ef Branch: refs/heads/master Commit: ec7934eff336299498cc79e857dce684c7126041 Parents: 306fd8e Author: Piotr Nowojski <piotr.nowoj...@gmail.com> Authored: Thu Jan 18 09:14:40 2018 +0100 Committer: Piotr Nowojski <piotr.nowoj...@gmail.com> Committed: Mon Feb 19 12:21:15 2018 +0100 ---------------------------------------------------------------------- .../flink/runtime/io/network/api/writer/RecordWriterTest.java | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/flink/blob/ec7934ef/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/api/writer/RecordWriterTest.java ---------------------------------------------------------------------- diff --git a/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/api/writer/RecordWriterTest.java b/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/api/writer/RecordWriterTest.java index 4fac5b5..d3bad47 100644 --- a/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/api/writer/RecordWriterTest.java +++ b/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/api/writer/RecordWriterTest.java @@ -129,7 +129,7 @@ public class RecordWriterTest { BufferProvider bufferProvider = mock(BufferProvider.class); when(bufferProvider.requestBufferBuilderBlocking()).thenAnswer(request); - ResultPartitionWriter partitionWriter = spy(new RecyclingPartitionWriter(bufferProvider)); + ResultPartitionWriter partitionWriter = new RecyclingPartitionWriter(bufferProvider); final RecordWriter<IntValue> recordWriter = new RecordWriter<IntValue>(partitionWriter); @@ -163,9 +163,8 @@ public class RecordWriterTest { recordWriter.clearBuffers(); - // Verify that buffer have been requested, but only one has been written out. + // Verify that buffer have been requested twice verify(bufferProvider, times(2)).requestBufferBuilderBlocking(); - verify(partitionWriter, times(1)).writeBuffer(any(Buffer.class), anyInt()); // Verify that the written out buffer has only been recycled once // (by the partition writer).