Examples with JMSConnectionFactoryProviderDefinition org.apache.nifi.jms.cf.JMSConnectionFactoryProviderDefinition used on opensource projects

Example 6 with JMSConnectionFactoryProviderDefinition

use of org.apache.nifi.jms.cf.JMSConnectionFactoryProviderDefinition in project nifi by apache.

the class PublishJMSIT method validateFailedPublishAndTransferToFailure.

@Test
public void validateFailedPublishAndTransferToFailure() throws Exception {
    ConnectionFactory cf = mock(ConnectionFactory.class);
    PublishJMS pubProc = new PublishJMS();
    TestRunner runner = TestRunners.newTestRunner(pubProc);
    JMSConnectionFactoryProviderDefinition cs = mock(JMSConnectionFactoryProviderDefinition.class);
    when(cs.getIdentifier()).thenReturn("cfProvider");
    when(cs.getConnectionFactory()).thenReturn(cf);
    runner.addControllerService("cfProvider", cs);
    runner.enableControllerService(cs);
    runner.setProperty(PublishJMS.CF_SERVICE, "cfProvider");
    runner.setProperty(PublishJMS.DESTINATION, "validateFailedPublishAndTransferToFailure");
    runner.enqueue("Hello Joe".getBytes());
    runner.run();
    Thread.sleep(200);
    assertTrue(runner.getFlowFilesForRelationship(PublishJMS.REL_SUCCESS).isEmpty());
    assertNotNull(runner.getFlowFilesForRelationship(PublishJMS.REL_FAILURE).get(0));
}