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

Example 86 with TestRunner

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

the class TestPutCloudWatchMetric method testValueLiteralDoubleInvalid.

@Test
public void testValueLiteralDoubleInvalid() throws Exception {
    MockPutCloudWatchMetric mockPutCloudWatchMetric = new MockPutCloudWatchMetric();
    final TestRunner runner = TestRunners.newTestRunner(mockPutCloudWatchMetric);
    runner.setProperty(PutCloudWatchMetric.NAMESPACE, "TestNamespace");
    runner.setProperty(PutCloudWatchMetric.METRIC_NAME, "TestMetric");
    runner.setProperty(PutCloudWatchMetric.VALUE, "nan");
    runner.assertNotValid();
}

Example 87 with TestRunner

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

the class TestPutCloudWatchMetric method testMissingBothValueAndStatisticSetInvalid.

@Test
public void testMissingBothValueAndStatisticSetInvalid() throws Exception {
    MockPutCloudWatchMetric mockPutCloudWatchMetric = new MockPutCloudWatchMetric();
    final TestRunner runner = TestRunners.newTestRunner(mockPutCloudWatchMetric);
    runner.setProperty(PutCloudWatchMetric.NAMESPACE, "TestNamespace");
    runner.setProperty(PutCloudWatchMetric.METRIC_NAME, "TestMetric");
    runner.assertNotValid();
}

Example 88 with TestRunner

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

the class TestPutCloudWatchMetric method testDimensions.

@Test
public void testDimensions() throws Exception {
    MockPutCloudWatchMetric mockPutCloudWatchMetric = new MockPutCloudWatchMetric();
    final TestRunner runner = TestRunners.newTestRunner(mockPutCloudWatchMetric);
    runner.setProperty(PutCloudWatchMetric.NAMESPACE, "TestNamespace");
    runner.setProperty(PutCloudWatchMetric.METRIC_NAME, "TestMetric");
    runner.setProperty(PutCloudWatchMetric.VALUE, "1.0");
    runner.setProperty(PutCloudWatchMetric.UNIT, "Count");
    runner.setProperty(PutCloudWatchMetric.TIMESTAMP, "1476296132575");
    runner.setProperty(new PropertyDescriptor.Builder().dynamic(true).name("dim1").build(), "${metric.dim1}");
    runner.setProperty(new PropertyDescriptor.Builder().dynamic(true).name("dim2").build(), "val2");
    runner.assertValid();
    final Map<String, String> attributes = new HashMap<>();
    attributes.put("metric.dim1", "1");
    runner.enqueue(new byte[] {}, attributes);
    runner.run();
    runner.assertAllFlowFilesTransferred(PutCloudWatchMetric.REL_SUCCESS, 1);
    Assert.assertEquals(1, mockPutCloudWatchMetric.putMetricDataCallCount);
    Assert.assertEquals("TestNamespace", mockPutCloudWatchMetric.actualNamespace);
    MetricDatum datum = mockPutCloudWatchMetric.actualMetricData.get(0);
    Assert.assertEquals("TestMetric", datum.getMetricName());
    Assert.assertEquals(1d, datum.getValue(), 0.0001d);
    List<Dimension> dimensions = datum.getDimensions();
    Collections.sort(dimensions, (d1, d2) -> d1.getName().compareTo(d2.getName()));
    Assert.assertEquals(2, dimensions.size());
    Assert.assertEquals("dim1", dimensions.get(0).getName());
    Assert.assertEquals("1", dimensions.get(0).getValue());
    Assert.assertEquals("dim2", dimensions.get(1).getName());
    Assert.assertEquals("val2", dimensions.get(1).getValue());
}

Example 89 with TestRunner

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

the class TestCredentialsProviderFactory method testAssumeRoleInvalidProxyPort.

@Test
public void testAssumeRoleInvalidProxyPort() throws Throwable {
    final TestRunner runner = TestRunners.newTestRunner(MockAWSProcessor.class);
    runner.setProperty(CredentialPropertyDescriptors.CREDENTIALS_FILE, "src/test/resources/mock-aws-credentials.properties");
    runner.setProperty(CredentialPropertyDescriptors.ASSUME_ROLE_PROXY_HOST, "proxy.company.com");
    runner.setProperty(CredentialPropertyDescriptors.ASSUME_ROLE_PROXY_PORT, "notIntPort");
    runner.assertNotValid();
}

Example 90 with TestRunner

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

the class TestCredentialsProviderFactory method testAssumeRoleCredentials.

@Test
public void testAssumeRoleCredentials() throws Throwable {
    final TestRunner runner = TestRunners.newTestRunner(MockAWSProcessor.class);
    runner.setProperty(CredentialPropertyDescriptors.CREDENTIALS_FILE, "src/test/resources/mock-aws-credentials.properties");
    runner.setProperty(CredentialPropertyDescriptors.ASSUME_ROLE_ARN, "BogusArn");
    runner.setProperty(CredentialPropertyDescriptors.ASSUME_ROLE_NAME, "BogusSession");
    runner.assertValid();
    Map<PropertyDescriptor, String> properties = runner.getProcessContext().getProperties();
    final CredentialsProviderFactory factory = new CredentialsProviderFactory();
    final AWSCredentialsProvider credentialsProvider = factory.getCredentialsProvider(properties);
    Assert.assertNotNull(credentialsProvider);
    assertEquals("credentials provider should be equal", STSAssumeRoleSessionCredentialsProvider.class, credentialsProvider.getClass());
}