mirror of https://github.com/apache/nifi.git
NIFI-604: Fixed checkstyle violation
This commit is contained in:
Mark Payne
parent
24c080788f
commit
eb94dab00c
|
|
@ -42,212 +42,212 @@ import org.junit.Test;
|
|||
*
|
||||
*/
|
||||
public class TestExecuteStreamCommand {
|
||||
@BeforeClass
|
||||
public static void init() {
|
||||
System.setProperty("org.slf4j.simpleLogger.defaultLogLevel", "info");
|
||||
System.setProperty("org.slf4j.simpleLogger.showDateTime", "true");
|
||||
System.setProperty("org.slf4j.simpleLogger.log.nifi.processors.standard.ExecuteStreamCommand", "debug");
|
||||
System.setProperty("org.slf4j.simpleLogger.log.nifi.processors.standard.TestExecuteStreamCommand", "debug");
|
||||
}
|
||||
@BeforeClass
|
||||
public static void init() {
|
||||
System.setProperty("org.slf4j.simpleLogger.defaultLogLevel", "info");
|
||||
System.setProperty("org.slf4j.simpleLogger.showDateTime", "true");
|
||||
System.setProperty("org.slf4j.simpleLogger.log.nifi.processors.standard.ExecuteStreamCommand", "debug");
|
||||
System.setProperty("org.slf4j.simpleLogger.log.nifi.processors.standard.TestExecuteStreamCommand", "debug");
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testExecuteJar() throws Exception {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/TestSuccess.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
@Test
|
||||
public void testExecuteJar() throws Exception {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/TestSuccess.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
MockFlowFile outputFlowFile = flowFiles.get(0);
|
||||
byte[] byteArray = outputFlowFile.toByteArray();
|
||||
String result = new String(byteArray);
|
||||
assertTrue(Pattern.compile("Test was a success\r?\n").matcher(result).find());
|
||||
assertEquals("0", outputFlowFile.getAttribute("execution.status"));
|
||||
assertEquals("java", outputFlowFile.getAttribute("execution.command"));
|
||||
assertEquals("-jar;", outputFlowFile.getAttribute("execution.command.args").substring(0, 5));
|
||||
String attribute = outputFlowFile.getAttribute("execution.command.args");
|
||||
String expected = "src" + File.separator + "test" + File.separator + "resources" + File.separator + "ExecuteCommand" + File.separator + "TestSuccess.jar";
|
||||
assertEquals(expected, attribute.substring(attribute.length() - expected.length()));
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
MockFlowFile outputFlowFile = flowFiles.get(0);
|
||||
byte[] byteArray = outputFlowFile.toByteArray();
|
||||
String result = new String(byteArray);
|
||||
assertTrue(Pattern.compile("Test was a success\r?\n").matcher(result).find());
|
||||
assertEquals("0", outputFlowFile.getAttribute("execution.status"));
|
||||
assertEquals("java", outputFlowFile.getAttribute("execution.command"));
|
||||
assertEquals("-jar;", outputFlowFile.getAttribute("execution.command.args").substring(0, 5));
|
||||
String attribute = outputFlowFile.getAttribute("execution.command.args");
|
||||
String expected = "src" + File.separator + "test" + File.separator + "resources" + File.separator + "ExecuteCommand" + File.separator + "TestSuccess.jar";
|
||||
assertEquals(expected, attribute.substring(attribute.length() - expected.length()));
|
||||
|
||||
MockFlowFile originalFlowFile = controller.getFlowFilesForRelationship(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP).get(0);
|
||||
assertEquals(outputFlowFile.getAttribute("execution.status"), originalFlowFile.getAttribute("execution.status"));
|
||||
assertEquals(outputFlowFile.getAttribute("execution.command"), originalFlowFile.getAttribute("execution.command"));
|
||||
assertEquals(outputFlowFile.getAttribute("execution.command.args"), originalFlowFile.getAttribute("execution.command.args"));
|
||||
}
|
||||
MockFlowFile originalFlowFile = controller.getFlowFilesForRelationship(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP).get(0);
|
||||
assertEquals(outputFlowFile.getAttribute("execution.status"), originalFlowFile.getAttribute("execution.status"));
|
||||
assertEquals(outputFlowFile.getAttribute("execution.command"), originalFlowFile.getAttribute("execution.command"));
|
||||
assertEquals(outputFlowFile.getAttribute("execution.command.args"), originalFlowFile.getAttribute("execution.command.args"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testExecuteJarWithBadPath() throws Exception {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/noSuchFile.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
assertEquals(0, flowFiles.get(0).getSize());
|
||||
assertEquals("Error: Unable to access jarfile", flowFiles.get(0).getAttribute("execution.error").substring(0, 31));
|
||||
}
|
||||
@Test
|
||||
public void testExecuteJarWithBadPath() throws Exception {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/noSuchFile.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
assertEquals(0, flowFiles.get(0).getSize());
|
||||
assertEquals("Error: Unable to access jarfile", flowFiles.get(0).getAttribute("execution.error").substring(0, 31));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testExecuteIngestAndUpdate() throws IOException {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/TestIngestAndUpdate.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
File dummy100MBytes = new File("target/100MB.txt");
|
||||
FileInputStream fis = new FileInputStream(dummy);
|
||||
FileOutputStream fos = new FileOutputStream(dummy100MBytes);
|
||||
byte[] bytes = new byte[1024];
|
||||
assertEquals(1000, fis.read(bytes));
|
||||
fis.close();
|
||||
for (int i = 0; i < 100000; i++) {
|
||||
fos.write(bytes, 0, 1000);
|
||||
}
|
||||
fos.close();
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy100MBytes.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
byte[] byteArray = flowFiles.get(0).toByteArray();
|
||||
String result = new String(byteArray);
|
||||
@Test
|
||||
public void testExecuteIngestAndUpdate() throws IOException {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/TestIngestAndUpdate.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
File dummy100MBytes = new File("target/100MB.txt");
|
||||
FileInputStream fis = new FileInputStream(dummy);
|
||||
FileOutputStream fos = new FileOutputStream(dummy100MBytes);
|
||||
byte[] bytes = new byte[1024];
|
||||
assertEquals(1000, fis.read(bytes));
|
||||
fis.close();
|
||||
for (int i = 0; i < 100000; i++) {
|
||||
fos.write(bytes, 0, 1000);
|
||||
}
|
||||
fos.close();
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy100MBytes.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
byte[] byteArray = flowFiles.get(0).toByteArray();
|
||||
String result = new String(byteArray);
|
||||
|
||||
assertTrue(Pattern.compile("nifi-standard-processors:ModifiedResult\r?\n").matcher(result).find());
|
||||
}
|
||||
assertTrue(Pattern.compile("nifi-standard-processors:ModifiedResult\r?\n").matcher(result).find());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testExecuteIngestAndUpdateWithWorkingDir() throws IOException {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/TestIngestAndUpdate.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.WORKING_DIR, "target");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
byte[] byteArray = flowFiles.get(0).toByteArray();
|
||||
String result = new String(byteArray);
|
||||
@Test
|
||||
public void testExecuteIngestAndUpdateWithWorkingDir() throws IOException {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/TestIngestAndUpdate.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.WORKING_DIR, "target");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
byte[] byteArray = flowFiles.get(0).toByteArray();
|
||||
String result = new String(byteArray);
|
||||
|
||||
final String quotedSeparator = Pattern.quote(File.separator);
|
||||
assertTrue(Pattern.compile(quotedSeparator + "nifi-standard-processors" + quotedSeparator + "target:ModifiedResult\r?\n").matcher(result).find());
|
||||
}
|
||||
final String quotedSeparator = Pattern.quote(File.separator);
|
||||
assertTrue(Pattern.compile(quotedSeparator + "nifi-standard-processors" + quotedSeparator + "target:ModifiedResult\r?\n").matcher(result).find());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIgnoredStdin() throws IOException {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/TestIngestAndUpdate.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.WORKING_DIR, "target");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.setProperty(ExecuteStreamCommand.IGNORE_STDIN, "true");
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
byte[] byteArray = flowFiles.get(0).toByteArray();
|
||||
String result = new String(byteArray);
|
||||
assertTrue("TestIngestAndUpdate.jar should not have received anything to modify",
|
||||
Pattern.compile("target:ModifiedResult\r?\n$").matcher(result).find());
|
||||
}
|
||||
@Test
|
||||
public void testIgnoredStdin() throws IOException {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/TestIngestAndUpdate.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.WORKING_DIR, "target");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.setProperty(ExecuteStreamCommand.IGNORE_STDIN, "true");
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
byte[] byteArray = flowFiles.get(0).toByteArray();
|
||||
String result = new String(byteArray);
|
||||
assertTrue("TestIngestAndUpdate.jar should not have received anything to modify",
|
||||
Pattern.compile("target:ModifiedResult\r?\n$").matcher(result).find());
|
||||
}
|
||||
|
||||
// this is dependent on window with cygwin...so it's not enabled
|
||||
@Ignore
|
||||
@Test
|
||||
public void testExecuteTouch() throws Exception {
|
||||
File testFile = new File("target/test.txt");
|
||||
testFile.delete();
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.WORKING_DIR, "target/xx1");
|
||||
controller.setThreadCount(6);
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "c:\\cygwin\\bin\\touch");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "test.txt");
|
||||
controller.assertValid();
|
||||
controller.run(6);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
assertEquals(5, flowFiles.size());
|
||||
assertEquals(0, flowFiles.get(0).getSize());
|
||||
// this is dependent on window with cygwin...so it's not enabled
|
||||
@Ignore
|
||||
@Test
|
||||
public void testExecuteTouch() throws Exception {
|
||||
File testFile = new File("target/test.txt");
|
||||
testFile.delete();
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.WORKING_DIR, "target/xx1");
|
||||
controller.setThreadCount(6);
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "c:\\cygwin\\bin\\touch");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "test.txt");
|
||||
controller.assertValid();
|
||||
controller.run(6);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
assertEquals(5, flowFiles.size());
|
||||
assertEquals(0, flowFiles.get(0).getSize());
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testDynamicEnvironment() throws Exception {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/TestDynamicEnvironment.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setProperty("NIFI_TEST_1", "testvalue1");
|
||||
controller.setProperty("NIFI_TEST_2", "testvalue2");
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.WORKING_DIR, "target");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
byte[] byteArray = flowFiles.get(0).toByteArray();
|
||||
String result = new String(byteArray);
|
||||
Set<String> dynamicEnvironmentVariables = new HashSet<>(Arrays.asList(result.split("\r?\n")));
|
||||
assertFalse("Should contain at least two environment variables starting with NIFI", dynamicEnvironmentVariables.size() < 2);
|
||||
assertTrue("NIFI_TEST_1 environment variable is missing", dynamicEnvironmentVariables.contains("NIFI_TEST_1=testvalue1"));
|
||||
assertTrue("NIFI_TEST_2 environment variable is missing", dynamicEnvironmentVariables.contains("NIFI_TEST_2=testvalue2"));
|
||||
}
|
||||
@Test
|
||||
public void testDynamicEnvironment() throws Exception {
|
||||
File exJar = new File("src/test/resources/ExecuteCommand/TestDynamicEnvironment.jar");
|
||||
File dummy = new File("src/test/resources/ExecuteCommand/1000bytes.txt");
|
||||
String jarPath = exJar.getAbsolutePath();
|
||||
exJar.setExecutable(true);
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setProperty("NIFI_TEST_1", "testvalue1");
|
||||
controller.setProperty("NIFI_TEST_2", "testvalue2");
|
||||
controller.setValidateExpressionUsage(false);
|
||||
controller.enqueue(dummy.toPath());
|
||||
controller.setProperty(ExecuteStreamCommand.WORKING_DIR, "target");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "java");
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_ARGUMENTS, "-jar;" + jarPath);
|
||||
controller.run(1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.ORIGINAL_RELATIONSHIP, 1);
|
||||
controller.assertTransferCount(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP, 1);
|
||||
List<MockFlowFile> flowFiles = controller.getFlowFilesForRelationship(ExecuteStreamCommand.OUTPUT_STREAM_RELATIONSHIP);
|
||||
byte[] byteArray = flowFiles.get(0).toByteArray();
|
||||
String result = new String(byteArray);
|
||||
Set<String> dynamicEnvironmentVariables = new HashSet<>(Arrays.asList(result.split("\r?\n")));
|
||||
assertFalse("Should contain at least two environment variables starting with NIFI", dynamicEnvironmentVariables.size() < 2);
|
||||
assertTrue("NIFI_TEST_1 environment variable is missing", dynamicEnvironmentVariables.contains("NIFI_TEST_1=testvalue1"));
|
||||
assertTrue("NIFI_TEST_2 environment variable is missing", dynamicEnvironmentVariables.contains("NIFI_TEST_2=testvalue2"));
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testQuotedArguments() throws Exception {
|
||||
List<String> args = ArgumentUtils.splitArgs("echo -n \"arg1 arg2 arg3\"", ' ');
|
||||
assertEquals(3, args.size());
|
||||
args = ArgumentUtils.splitArgs("echo;-n;\"arg1 arg2 arg3\"", ';');
|
||||
assertEquals(3, args.size());
|
||||
}
|
||||
@Test
|
||||
public void testQuotedArguments() throws Exception {
|
||||
List<String> args = ArgumentUtils.splitArgs("echo -n \"arg1 arg2 arg3\"", ' ');
|
||||
assertEquals(3, args.size());
|
||||
args = ArgumentUtils.splitArgs("echo;-n;\"arg1 arg2 arg3\"", ';');
|
||||
assertEquals(3, args.size());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testInvalidDelimiter() throws Exception {
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "echo");
|
||||
controller.assertValid();
|
||||
controller.setProperty(ExecuteStreamCommand.ARG_DELIMITER, "foo");
|
||||
controller.assertNotValid();
|
||||
controller.setProperty(ExecuteStreamCommand.ARG_DELIMITER, "f");
|
||||
controller.assertValid();
|
||||
}
|
||||
@Test
|
||||
public void testInvalidDelimiter() throws Exception {
|
||||
final TestRunner controller = TestRunners.newTestRunner(ExecuteStreamCommand.class);
|
||||
controller.setProperty(ExecuteStreamCommand.EXECUTION_COMMAND, "echo");
|
||||
controller.assertValid();
|
||||
controller.setProperty(ExecuteStreamCommand.ARG_DELIMITER, "foo");
|
||||
controller.assertNotValid();
|
||||
controller.setProperty(ExecuteStreamCommand.ARG_DELIMITER, "f");
|
||||
controller.assertValid();
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue