Examples with AWSCredentialsProviderControllerService org.apache.nifi.processors.aws.credentials.provider.service.AWSCredentialsProviderControllerService used on opensource projects

Example 6 with AWSCredentialsProviderControllerService

use of org.apache.nifi.processors.aws.credentials.provider.service.AWSCredentialsProviderControllerService in project nifi by apache.

the class ITPutCloudWatchMetric method testPublishWithCredentialsProviderService.

@Test
public void testPublishWithCredentialsProviderService() throws Throwable {
    final TestRunner runner = TestRunners.newTestRunner(new PutCloudWatchMetric());
    runner.setValidateExpressionUsage(false);
    final AWSCredentialsProviderControllerService serviceImpl = new AWSCredentialsProviderControllerService();
    runner.addControllerService("awsCredentialsProvider", serviceImpl);
    runner.setProperty(serviceImpl, AbstractAWSCredentialsProviderProcessor.CREDENTIALS_FILE, System.getProperty("user.home") + "/aws-credentials.properties");
    runner.enableControllerService(serviceImpl);
    runner.assertValid(serviceImpl);
    runner.setProperty(PutCloudWatchMetric.NAMESPACE, "Test");
    runner.setProperty(PutCloudWatchMetric.METRIC_NAME, "Test");
    runner.setProperty(PutCloudWatchMetric.VALUE, "1.0");
    runner.setProperty(PutSNS.AWS_CREDENTIALS_PROVIDER_SERVICE, "awsCredentialsProvider");
    runner.enqueue(new byte[] {});
    runner.run();
    runner.assertAllFlowFilesTransferred(PutCloudWatchMetric.REL_SUCCESS, 1);
}

Example 7 with AWSCredentialsProviderControllerService

use of org.apache.nifi.processors.aws.credentials.provider.service.AWSCredentialsProviderControllerService in project nifi by apache.

the class ITPutS3Object method testPutS3ObjectUsingCredentialsProviderService.

@Test
public void testPutS3ObjectUsingCredentialsProviderService() throws Throwable {
    final TestRunner runner = TestRunners.newTestRunner(new PutS3Object());
    final AWSCredentialsProviderControllerService serviceImpl = new AWSCredentialsProviderControllerService();
    runner.addControllerService("awsCredentialsProvider", serviceImpl);
    runner.setProperty(serviceImpl, AbstractAWSCredentialsProviderProcessor.CREDENTIALS_FILE, System.getProperty("user.home") + "/aws-credentials.properties");
    runner.enableControllerService(serviceImpl);
    runner.assertValid(serviceImpl);
    runner.setProperty(PutS3Object.AWS_CREDENTIALS_PROVIDER_SERVICE, "awsCredentialsProvider");
    runner.setProperty(PutS3Object.REGION, REGION);
    runner.setProperty(PutS3Object.BUCKET, BUCKET_NAME);
    Assert.assertTrue(runner.setProperty("x-custom-prop", "hello").isValid());
    for (int i = 0; i < 3; i++) {
        final Map<String, String> attrs = new HashMap<>();
        attrs.put("filename", String.valueOf(i) + ".txt");
        runner.enqueue(getResourcePath(SAMPLE_FILE_RESOURCE_NAME), attrs);
    }
    runner.run(3);
    runner.assertAllFlowFilesTransferred(PutS3Object.REL_SUCCESS, 3);
}

Example 8 with AWSCredentialsProviderControllerService

use of org.apache.nifi.processors.aws.credentials.provider.service.AWSCredentialsProviderControllerService in project nifi by apache.

the class ITPutSNS method testPublishWithCredentialsProviderService.

@Test
public void testPublishWithCredentialsProviderService() throws Throwable {
    final TestRunner runner = TestRunners.newTestRunner(new PutSNS());
    String snsArn = "Add Sns arn here";
    runner.setValidateExpressionUsage(false);
    runner.setProperty(PutSNS.ARN, snsArn);
    assertTrue(runner.setProperty("DynamicProperty", "hello!").isValid());
    final AWSCredentialsProviderControllerService serviceImpl = new AWSCredentialsProviderControllerService();
    runner.addControllerService("awsCredentialsProvider", serviceImpl);
    runner.setProperty(serviceImpl, AbstractAWSCredentialsProviderProcessor.CREDENTIALS_FILE, System.getProperty("user.home") + "/aws-credentials.properties");
    runner.enableControllerService(serviceImpl);
    runner.assertValid(serviceImpl);
    runner.setProperty(PutSNS.AWS_CREDENTIALS_PROVIDER_SERVICE, "awsCredentialsProvider");
    runner.run(1);
    final Map<String, String> attrs = new HashMap<>();
    attrs.put("filename", "1.txt");
    runner.enqueue(Paths.get("src/test/resources/hello.txt"), attrs);
    runner.run();
    runner.assertAllFlowFilesTransferred(PutSNS.REL_SUCCESS, 1);
}

Example 9 with AWSCredentialsProviderControllerService

use of org.apache.nifi.processors.aws.credentials.provider.service.AWSCredentialsProviderControllerService in project nifi by apache.

the class ITPutSQS method testSimplePutUsingCredentialsProviderService.

@Test
public void testSimplePutUsingCredentialsProviderService() 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()));
    }
}