From 361ac1f1e5c8ee943220b337c92cd49616a02194 Mon Sep 17 00:00:00 2001 From: danbress Date: Sun, 15 Feb 2015 09:08:22 -0500 Subject: [PATCH] NIFI-333 - Adding failure test for DecompressContent --- .../standard/TestCompressContent.java | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/nifi/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestCompressContent.java b/nifi/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestCompressContent.java index 71c85831d3..df1d506678 100644 --- a/nifi/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestCompressContent.java +++ b/nifi/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestCompressContent.java @@ -109,4 +109,22 @@ public class TestCompressContent { flowFile.assertAttributeEquals("filename", "SampleFile.txt.gz"); } + + @Test + public void testDecompressFailure() throws IOException { + final TestRunner runner = TestRunners.newTestRunner(CompressContent.class); + runner.setProperty(CompressContent.MODE, "decompress"); + runner.setProperty(CompressContent.COMPRESSION_FORMAT, "gzip"); + + byte[] data = new byte[]{1,2,3,4,5,6,7,8,9,10}; + runner.enqueue(data); + + + assertTrue(runner.setProperty(CompressContent.UPDATE_FILENAME, "true").isValid()); + runner.run(); + runner.assertQueueEmpty(); + runner.assertAllFlowFilesTransferred(CompressContent.REL_FAILURE, 1); + + runner.getFlowFilesForRelationship(CompressContent.REL_FAILURE).get(0).assertContentEquals(data); + } }