Repository: nifi Updated Branches: refs/heads/master 9e9182aa6 -> 704c333b0
http://git-wip-us.apache.org/repos/asf/nifi/blob/704c333b/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/test/java/org/apache/nifi/processors/aws/sqs/TestPutSQS.java ---------------------------------------------------------------------- diff --git a/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/test/java/org/apache/nifi/processors/aws/sqs/TestPutSQS.java b/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/test/java/org/apache/nifi/processors/aws/sqs/TestPutSQS.java index 712f706..85bd6e8 100644 --- a/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/test/java/org/apache/nifi/processors/aws/sqs/TestPutSQS.java +++ b/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/test/java/org/apache/nifi/processors/aws/sqs/TestPutSQS.java @@ -19,9 +19,13 @@ package org.apache.nifi.processors.aws.sqs; import java.io.IOException; import java.nio.file.Paths; import java.util.HashMap; +import java.util.List; import java.util.Map; +import org.apache.nifi.processors.aws.AbstractAWSProcessor; +import org.apache.nifi.processors.aws.credentials.provider.service.AWSCredentialsProviderControllerService; import org.apache.nifi.processors.aws.sns.PutSNS; +import org.apache.nifi.util.MockFlowFile; import org.apache.nifi.util.TestRunner; import org.apache.nifi.util.TestRunners; import org.junit.Assert; @@ -49,4 +53,34 @@ public class TestPutSQS { runner.assertAllFlowFilesTransferred(PutSQS.REL_SUCCESS, 1); } + @Test + public void testSimplePutUsingCredentailsProviderService() throws Throwable { + final TestRunner runner = TestRunners.newTestRunner(new PutSQS()); + + runner.setProperty(PutSQS.TIMEOUT, "30 secs"); + String queueUrl = "Add queue url here"; + runner.setProperty(PutSQS.QUEUE_URL, queueUrl); + runner.setValidateExpressionUsage(false); + final AWSCredentialsProviderControllerService serviceImpl = new AWSCredentialsProviderControllerService(); + + runner.addControllerService("awsCredentialsProvider", serviceImpl); + + runner.setProperty(serviceImpl, AbstractAWSProcessor.CREDENTIALS_FILE, System.getProperty("user.home") + "/aws-credentials.properties"); + runner.enableControllerService(serviceImpl); + + runner.assertValid(serviceImpl); + + final Map<String, String> attrs = new HashMap<>(); + attrs.put("filename", "1.txt"); + runner.enqueue(Paths.get("src/test/resources/hello.txt"), attrs); + runner.setProperty(PutSQS.AWS_CREDENTIALS_PROVIDER_SERVICE, "awsCredentialsProvider"); + runner.run(1); + + final List<MockFlowFile> flowFiles = runner.getFlowFilesForRelationship(PutSQS.REL_SUCCESS); + for (final MockFlowFile mff : flowFiles) { + System.out.println(mff.getAttributes()); + System.out.println(new String(mff.toByteArray())); + } + + } } http://git-wip-us.apache.org/repos/asf/nifi/blob/704c333b/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/test/resources/mock-aws-credentials.properties ---------------------------------------------------------------------- diff --git a/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/test/resources/mock-aws-credentials.properties b/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/test/resources/mock-aws-credentials.properties new file mode 100644 index 0000000..11d1225 --- /dev/null +++ b/nifi-nar-bundles/nifi-aws-bundle/nifi-aws-processors/src/test/resources/mock-aws-credentials.properties @@ -0,0 +1,2 @@ +accessKey = awsAccessKey +secretKey = awsSecretKeyId \ No newline at end of file
