Github user NicoK commented on a diff in the pull request:
https://github.com/apache/flink/pull/4552#discussion_r161485403
--- Diff:
flink-runtime/src/test/java/org/apache/flink/runtime/io/network/netty/PartitionRequestClientHandlerTest.java
---
@@ -434,6 +443,29 @@ private RemoteInputChannel
createRemoteInputChannel(SingleInputGate inputGate) t
UnregisteredMetricGroups.createUnregisteredTaskMetricGroup().getIOMetricGroup());
}
+ /**
+ * Creates and returns a remote input channel for the specific input
gate with specific partition request client.
+ *
+ * @param inputGate The input gate owns the created input channel.
+ * @param client The client is used to send partition request.
+ * @return The new created remote input channel.
+ */
+ private RemoteInputChannel createRemoteInputChannel(SingleInputGate
inputGate, PartitionRequestClient client) throws Exception {
--- End diff --
could you modify
`PartitionRequestClientHandlerTest#createRemoteInputChannel(SingleInputGate)`
to rely on this method, i.e. `return createRemoteInputChannel(inputGate,
mock(PartitionRequestClient.class));`?
---