Examples with TestRunner org.apache.nifi.util.TestRunner used on opensource projects

Example 71 with TestRunner

use of org.apache.nifi.util.TestRunner in project nifi by apache.

the class TestExtractEmailHeaders method testValidEmailWithAttachments.

@Test
public void testValidEmailWithAttachments() throws Exception {
    final TestRunner runner = TestRunners.newTestRunner(new ExtractEmailHeaders());
    // Create the message dynamically
    byte[] withAttachment = attachmentGenerator.WithAttachments(1);
    runner.enqueue(withAttachment);
    runner.run();
    runner.assertTransferCount(ExtractEmailHeaders.REL_SUCCESS, 1);
    runner.assertTransferCount(ExtractEmailHeaders.REL_FAILURE, 0);
    runner.assertQueueEmpty();
    final List<MockFlowFile> splits = runner.getFlowFilesForRelationship(ExtractEmailHeaders.REL_SUCCESS);
    splits.get(0).assertAttributeEquals("email.headers.from.0", from);
    splits.get(0).assertAttributeEquals("email.headers.to.0", to);
    splits.get(0).assertAttributeEquals("email.headers.subject", subject);
    splits.get(0).assertAttributeEquals("email.attachment_count", "1");
}

Example 72 with TestRunner

use of org.apache.nifi.util.TestRunner in project nifi by apache.

the class TestExtractEmailHeaders method testValidEmailWithNoRecipients.

/**
 * Test case added for NIFI-4326 for a potential NPE bug
 * if the email message contains no recipient header fields, ie,
 * TO, CC, BCC.
 */
@Test
public void testValidEmailWithNoRecipients() throws Exception {
    final TestRunner runner = TestRunners.newTestRunner(new ExtractEmailHeaders());
    runner.setProperty(ExtractEmailHeaders.CAPTURED_HEADERS, "MIME-Version");
    MimeMessage simpleEmailMimeMessage = attachmentGenerator.SimpleEmailMimeMessage();
    simpleEmailMimeMessage.removeHeader("To");
    simpleEmailMimeMessage.removeHeader("Cc");
    simpleEmailMimeMessage.removeHeader("Bcc");
    ByteArrayOutputStream messageBytes = new ByteArrayOutputStream();
    try {
        simpleEmailMimeMessage.writeTo(messageBytes);
    } catch (IOException | MessagingException e) {
        e.printStackTrace();
    }
    runner.enqueue(messageBytes.toByteArray());
    runner.run();
    runner.assertTransferCount(ExtractEmailHeaders.REL_SUCCESS, 1);
    runner.assertTransferCount(ExtractEmailHeaders.REL_FAILURE, 0);
    runner.assertQueueEmpty();
    final List<MockFlowFile> splits = runner.getFlowFilesForRelationship(ExtractEmailHeaders.REL_SUCCESS);
    splits.get(0).assertAttributeEquals("email.headers.from.0", from);
    splits.get(0).assertAttributeExists("email.headers.mime-version");
    splits.get(0).assertAttributeNotExists("email.headers.to");
    splits.get(0).assertAttributeNotExists("email.headers.cc");
    splits.get(0).assertAttributeNotExists("email.headers.bcc");
}

Example 73 with TestRunner

use of org.apache.nifi.util.TestRunner in project nifi by apache.

the class TestExtractEmailHeaders method testNonStrictParsingPassesForInvalidAddresses.

/**
 * NIFI-4326 adds a new feature to disable strict address parsing for
 * mailbox list header fields. This is a test case that asserts that
 * lax address parsing passes (when set to "strict=false") for malformed
 * addresses.
 */
@Test
public void testNonStrictParsingPassesForInvalidAddresses() throws Exception {
    final TestRunner runner = TestRunners.newTestRunner(new ExtractEmailHeaders());
    runner.setProperty(ExtractEmailHeaders.STRICT_PARSING, "false");
    MimeMessage simpleEmailMimeMessage = attachmentGenerator.SimpleEmailMimeMessage();
    simpleEmailMimeMessage.setHeader("From", "<bad_email>");
    simpleEmailMimeMessage.setHeader("To", "<>, Joe, \"\" <>");
    ByteArrayOutputStream messageBytes = new ByteArrayOutputStream();
    try {
        simpleEmailMimeMessage.writeTo(messageBytes);
    } catch (IOException | MessagingException e) {
        e.printStackTrace();
    }
    runner.enqueue(messageBytes.toByteArray());
    runner.run();
    runner.assertTransferCount(ExtractEmailHeaders.REL_SUCCESS, 1);
    runner.assertTransferCount(ExtractEmailHeaders.REL_FAILURE, 0);
    runner.assertQueueEmpty();
    final List<MockFlowFile> splits = runner.getFlowFilesForRelationship(ExtractEmailHeaders.REL_SUCCESS);
    splits.get(0).assertAttributeEquals("email.headers.from.0", "bad_email");
    splits.get(0).assertAttributeEquals("email.headers.to.0", "");
    splits.get(0).assertAttributeEquals("email.headers.to.1", "Joe");
    splits.get(0).assertAttributeEquals("email.headers.to.2", "");
}

Example 74 with TestRunner

use of org.apache.nifi.util.TestRunner in project nifi by apache.

the class TestExtractTNEFAttachments method testValidTNEFWithoutAttachment.

@Test
public void testValidTNEFWithoutAttachment() throws Exception {
    final TestRunner runner = TestRunners.newTestRunner(new ExtractTNEFAttachments());
    runner.enqueue(Paths.get("src/test/resources/winmail-simple.dat"));
    runner.run();
    runner.assertTransferCount(ExtractEmailAttachments.REL_ORIGINAL, 1);
    runner.assertTransferCount(ExtractEmailAttachments.REL_FAILURE, 0);
    runner.assertTransferCount(ExtractEmailAttachments.REL_ATTACHMENTS, 0);
    // Have a look at the attachments...
    final List<MockFlowFile> splits = runner.getFlowFilesForRelationship(ExtractEmailAttachments.REL_ATTACHMENTS);
    Assert.assertTrue(splits.size() == 0);
}

Example 75 with TestRunner

use of org.apache.nifi.util.TestRunner in project nifi by apache.

the class TestExtractTNEFAttachments method testValidTNEFWithAttachment.

@Test
public void testValidTNEFWithAttachment() throws Exception {
    Random rnd = new Random();
    final TestRunner runner = TestRunners.newTestRunner(new ExtractTNEFAttachments());
    runner.enqueue(Paths.get("src/test/resources/winmail-with-attachment.dat"));
    runner.run();
    runner.assertTransferCount(ExtractEmailAttachments.REL_ORIGINAL, 1);
    runner.assertTransferCount(ExtractEmailAttachments.REL_FAILURE, 0);
    runner.assertTransferCount(ExtractEmailAttachments.REL_ATTACHMENTS, 1);
    // Have a look at the attachments...
    final List<MockFlowFile> splits = runner.getFlowFilesForRelationship(ExtractTNEFAttachments.REL_ATTACHMENTS);
    List<String> filenames = new ArrayList<>();
    for (int a = 0; a < splits.size(); a++) {
        filenames.add(splits.get(a).getAttribute("filename").toString());
    }
    Assert.assertTrue(filenames.containsAll(Arrays.asList("nifiDrop.svg")));
}