Examples with HttpRemoteSiteListener org.apache.nifi.remote.HttpRemoteSiteListener used on opensource projects

Example 6 with HttpRemoteSiteListener

use of org.apache.nifi.remote.HttpRemoteSiteListener in project nifi by apache.

the class TestDataTransferResource method testCommitInputPortTransaction.

@Test
public void testCommitInputPortTransaction() throws Exception {
    final HttpServletRequest req = createCommonHttpServletRequest();
    final DataTransferResource resource = getDataTransferResource();
    final ServletContext context = null;
    final InputStream inputStream = null;
    final HttpRemoteSiteListener transactionManager = HttpRemoteSiteListener.getInstance(NiFiProperties.createBasicNiFiProperties(null, null));
    final String transactionId = transactionManager.createTransaction();
    final Response response = resource.commitInputPortTransaction(ResponseCode.CONFIRM_TRANSACTION.getCode(), "port-id", transactionId, req, context, inputStream);
    transactionManager.cancelTransaction(transactionId);
    TransactionResultEntity resultEntity = (TransactionResultEntity) response.getEntity();
    assertEquals(200, response.getStatus());
    assertEquals(ResponseCode.CONFIRM_TRANSACTION.getCode(), resultEntity.getResponseCode());
}

Example 7 with HttpRemoteSiteListener

use of org.apache.nifi.remote.HttpRemoteSiteListener in project nifi by apache.

the class TestDataTransferResource method testExtendTransaction.

@Test
public void testExtendTransaction() throws Exception {
    final HttpServletRequest req = createCommonHttpServletRequest();
    final DataTransferResource resource = getDataTransferResource();
    final String locationUriStr = "http://localhost:8080/nifi-api/data-transfer/input-ports/port-id/transactions/transaction-id";
    final ServletContext context = null;
    final HttpServletResponse res = null;
    final UriInfo uriInfo = mockUriInfo(locationUriStr);
    final InputStream inputStream = null;
    final HttpRemoteSiteListener transactionManager = HttpRemoteSiteListener.getInstance(NiFiProperties.createBasicNiFiProperties(null, null));
    final String transactionId = transactionManager.createTransaction();
    final Response response = resource.extendPortTransactionTTL("input-ports", "port-id", transactionId, req, res, context, uriInfo, inputStream);
    transactionManager.cancelTransaction(transactionId);
    TransactionResultEntity resultEntity = (TransactionResultEntity) response.getEntity();
    assertEquals(200, response.getStatus());
    assertEquals(ResponseCode.CONTINUE_TRANSACTION.getCode(), resultEntity.getResponseCode());
}

Example 8 with HttpRemoteSiteListener

use of org.apache.nifi.remote.HttpRemoteSiteListener in project nifi by apache.

the class TestDataTransferResource method testReceiveZeroFlowFiles.

@Test
public void testReceiveZeroFlowFiles() throws Exception {
    final HttpServletRequest req = createCommonHttpServletRequest();
    final DataTransferResource resource = getDataTransferResource();
    final HttpFlowFileServerProtocol serverProtocol = resource.getHttpFlowFileServerProtocol(null);
    final RootGroupPort port = mock(RootGroupPort.class);
    doReturn(port).when(serverProtocol).getPort();
    doAnswer(invocation -> 0).when(port).receiveFlowFiles(any(Peer.class), any());
    final ServletContext context = null;
    final InputStream inputStream = null;
    final HttpRemoteSiteListener transactionManager = HttpRemoteSiteListener.getInstance(NiFiProperties.createBasicNiFiProperties(null, null));
    final String transactionId = transactionManager.createTransaction();
    final Response response = resource.receiveFlowFiles("port-id", transactionId, req, context, inputStream);
    transactionManager.cancelTransaction(transactionId);
    assertEquals(400, response.getStatus());
}

Example 9 with HttpRemoteSiteListener

use of org.apache.nifi.remote.HttpRemoteSiteListener in project nifi by apache.

the class TestDataTransferResource method testReceiveFlowFiles.

@Test
public void testReceiveFlowFiles() throws Exception {
    final HttpServletRequest req = createCommonHttpServletRequest();
    final DataTransferResource resource = getDataTransferResource();
    final HttpFlowFileServerProtocol serverProtocol = resource.getHttpFlowFileServerProtocol(null);
    final RootGroupPort port = mock(RootGroupPort.class);
    doReturn(port).when(serverProtocol).getPort();
    doAnswer(invocation -> {
        Peer peer = (Peer) invocation.getArguments()[0];
        ((HttpServerCommunicationsSession) peer.getCommunicationsSession()).setChecksum("server-checksum");
        return 7;
    }).when(port).receiveFlowFiles(any(Peer.class), any());
    final ServletContext context = null;
    final InputStream inputStream = null;
    final HttpRemoteSiteListener transactionManager = HttpRemoteSiteListener.getInstance(NiFiProperties.createBasicNiFiProperties(null, null));
    final String transactionId = transactionManager.createTransaction();
    final Response response = resource.receiveFlowFiles("port-id", transactionId, req, context, inputStream);
    transactionManager.cancelTransaction(transactionId);
    final Object entity = response.getEntity();
    assertEquals(202, response.getStatus());
    assertEquals("server-checksum", entity);
}