public void onFailedPartitionRequest() {
        inputGate.triggerPartitionStateCheck(partitionId, channelInfo);
    }
@Test
    void testOnFailedPartitionRequest() {
        final ResultPartitionID partitionId = new ResultPartitionID();
        final TestPartitionProducerStateProvider provider =
                new TestPartitionProducerStateProvider(partitionId);
        final SingleInputGate inputGate =
                new SingleInputGateBuilder().setPartitionProducerStateProvider(provider).build();
        final RemoteInputChannel ch =
                InputChannelBuilder.newBuilder()
                        .setPartitionId(partitionId)
                        .buildRemoteChannel(inputGate);

        ch.onFailedPartitionRequest();

        assertThat(provider.isInvoked()).isTrue();
    }