Java Code Examples for org.apache.nifi.util.MockFlowFile#putAttributes()
The following examples show how to use
org.apache.nifi.util.MockFlowFile#putAttributes() .
You can vote up the ones you like or vote down the ones you don't like,
and go to the original project or source file by following the links above each example. You may check out the related API usage on the sidebar.
Example 1
| Source File: TestDBCPConnectionPoolLookup.java From nifi with Apache License 2.0 | 6 votes |
|
@Test
public void testFlowFileFiltering() {
final FlowFileFilter filter = dbcpLookupService.getFlowFileFilter();
assertNotNull(filter);
final MockFlowFile ff0 = new MockFlowFile(0);
final MockFlowFile ff1 = new MockFlowFile(1);
final MockFlowFile ff2 = new MockFlowFile(2);
final MockFlowFile ff3 = new MockFlowFile(3);
final MockFlowFile ff4 = new MockFlowFile(4);
ff0.putAttributes(Collections.singletonMap(DBCPConnectionPoolLookup.DATABASE_NAME_ATTRIBUTE, "db-A"));
ff1.putAttributes(Collections.singletonMap(DBCPConnectionPoolLookup.DATABASE_NAME_ATTRIBUTE, "db-B"));
ff2.putAttributes(Collections.singletonMap(DBCPConnectionPoolLookup.DATABASE_NAME_ATTRIBUTE, "db-A"));
ff3.putAttributes(Collections.singletonMap(DBCPConnectionPoolLookup.DATABASE_NAME_ATTRIBUTE, "db-B"));
ff4.putAttributes(Collections.singletonMap(DBCPConnectionPoolLookup.DATABASE_NAME_ATTRIBUTE, "db-A"));
assertEquals(FlowFileFilter.FlowFileFilterResult.ACCEPT_AND_CONTINUE, filter.filter(ff0));
assertEquals(FlowFileFilter.FlowFileFilterResult.REJECT_AND_CONTINUE, filter.filter(ff1));
assertEquals(FlowFileFilter.FlowFileFilterResult.ACCEPT_AND_CONTINUE, filter.filter(ff2));
assertEquals(FlowFileFilter.FlowFileFilterResult.REJECT_AND_CONTINUE, filter.filter(ff3));
assertEquals(FlowFileFilter.FlowFileFilterResult.ACCEPT_AND_CONTINUE, filter.filter(ff4));
}
Example 2
| Source File: ExtractTextProcessorTest.java From nifi-extracttext-processor with Apache License 2.0 | 6 votes |
|
@Test
public void when_running_processor_mime_type_should_be_discovered_for_pdf_input() {
try {
final String filename = "simple.pdf";
MockFlowFile flowFile = testRunner.enqueue(new FileInputStream(new File("src/test/resources/" + filename)));
Map<String, String> attrs = new HashMap<String, String>() {{ put("filename", filename); }};
flowFile.putAttributes(attrs);
} catch (FileNotFoundException e) {
e.printStackTrace();
}
testRunner.assertValid();
testRunner.run();
testRunner.assertAllFlowFilesTransferred(ExtractTextProcessor.REL_SUCCESS);
List<MockFlowFile> successFiles = testRunner.getFlowFilesForRelationship(ExtractTextProcessor.REL_SUCCESS);
for (MockFlowFile mockFile : successFiles) {
mockFile.assertAttributeExists("mime.type");
mockFile.assertAttributeEquals("mime.type", "text/plain");
mockFile.assertAttributeExists("orig.mime.type");
mockFile.assertAttributeEquals("orig.mime.type", "application/pdf");
}
}
Example 3
| Source File: PutAzureEventHubTest.java From nifi with Apache License 2.0 | 6 votes |
|
@Test
public void testAllAttributesAreLiftedToProperties() {
MockedEventhubClientMockPutAzureEventHub processor = new PutAzureEventHubTest.MockedEventhubClientMockPutAzureEventHub();
MockitoAnnotations.initMocks(processor);
EventHubClient eventHubClient = processor.getEventHubClient();
when(eventHubClient.send(any(EventData.class)))
.thenReturn(CompletableFuture.completedFuture(null));
testRunner = TestRunners.newTestRunner(processor);
setUpStandardTestConfig();
MockFlowFile flowFile = new MockFlowFile(1234);
ImmutableMap<String, String> demoAttributes = ImmutableMap.of("A", "a", "B", "b", "D", "d", "C", "c");
flowFile.putAttributes(demoAttributes);
testRunner.enqueue(flowFile);
testRunner.run(1, true);
ArgumentCaptor<EventData> eventDataCaptor = ArgumentCaptor.forClass(EventData.class);
Mockito.verify(eventHubClient).send(eventDataCaptor.capture());
EventData event = eventDataCaptor.getValue();
assertTrue(event.getProperties().entrySet().containsAll(demoAttributes.entrySet()));
}
Example 4
| Source File: ExtractTextProcessorTest.java From nifi-extracttext-processor with Apache License 2.0 | 6 votes |
|
@Test
public void when_running_processor_mime_type_should_be_discovered_for_doc_input() {
try {
final String filename = "simple.doc";
MockFlowFile flowFile = testRunner.enqueue(new FileInputStream(new File("src/test/resources/" + filename)));
Map<String, String> attrs = new HashMap<String, String>() {{ put("filename", filename); }};
flowFile.putAttributes(attrs);
} catch (FileNotFoundException e) {
e.printStackTrace();
}
testRunner.assertValid();
testRunner.run();
testRunner.assertAllFlowFilesTransferred(ExtractTextProcessor.REL_SUCCESS);
List<MockFlowFile> successFiles = testRunner.getFlowFilesForRelationship(ExtractTextProcessor.REL_SUCCESS);
for (MockFlowFile mockFile : successFiles) {
mockFile.assertAttributeExists("mime.type");
mockFile.assertAttributeEquals("mime.type", "text/plain");
mockFile.assertAttributeExists("orig.mime.type");
mockFile.assertAttributeEquals("orig.mime.type", "application/msword");
}
}
Example 5
| Source File: PutAzureEventHubTest.java From nifi with Apache License 2.0 | 6 votes |
|
@Test
public void testMessageIsSentWithPartitioningKeyIfSpecifiedAndPopulated() {
MockedEventhubClientMockPutAzureEventHub processor = new PutAzureEventHubTest.MockedEventhubClientMockPutAzureEventHub();
MockitoAnnotations.initMocks(processor);
EventHubClient eventHubClient = processor.getEventHubClient();
when(eventHubClient.send(any(EventData.class), anyString()))
.thenReturn(CompletableFuture.completedFuture(null));
when(eventHubClient.send(any(EventData.class)))
.thenThrow(new RuntimeException("Partition-key-less method called despite key is defined and required."));
testRunner = TestRunners.newTestRunner(processor);
setUpStandardTestConfig();
testRunner.setProperty(PutAzureEventHub.PARTITIONING_KEY_ATTRIBUTE_NAME, TEST_PARTITIONING_KEY_ATTRIBUTE_NAME);
MockFlowFile flowFile = new MockFlowFile(1234);
flowFile.putAttributes(ImmutableMap.of(TEST_PARTITIONING_KEY_ATTRIBUTE_NAME, TEST_PARTITIONING_KEY));
testRunner.enqueue(flowFile);
testRunner.run(1, true);
Mockito.verify(eventHubClient).send(any(EventData.class), eq(TEST_PARTITIONING_KEY));
}
Example 6
| Source File: TestPutRiemann.java From nifi with Apache License 2.0 | 5 votes |
|
@Test(expected = AssertionError.class)
public void testFailedDeref() {
TestRunner runner = getTestRunner(true);
MockFlowFile flowFile = new MockFlowFile(1);
Map<String, String> attributes = new HashMap<>();
attributes.put("riemann.metric", "5");
flowFile.putAttributes(attributes);
runner.enqueue(flowFile);
try {
runner.run();
} catch (ProcessException e) {
runner.assertQueueNotEmpty();
throw e;
}
}
Example 7
| Source File: TestPutRiemann.java From nifi with Apache License 2.0 | 5 votes |
|
@Test
public void testBatchedEvents() {
// (2 batches) + (1 remaining event)
int iterations = Integer.parseInt(PutRiemann.BATCH_SIZE.getDefaultValue()) * 2 + 1;
TestRunner runner = getTestRunner();
for (int i = 0; i < iterations; i++) {
Map<String, String> attributes = new HashMap<>();
attributes.put("riemann.metric", Float.toString(i));
attributes.put("riemann.host", "batch-host");
attributes.put("custom.attribute.1", "attr1");
attributes.put("custom.attribute.2", "attr2");
attributes.put("custom.attribute.3", "attr3");
MockFlowFile flowFile = new MockFlowFile(i);
flowFile.putAttributes(attributes);
runner.enqueue(flowFile);
}
runner.run(3);
runner.assertAllFlowFilesTransferred(PutRiemann.REL_SUCCESS);
for (int i = 0; i < iterations; i++) {
Proto.Event event = eventStream.remove();
assertEquals("nifi-test-service", event.getService());
assertTrue(5.0 == event.getTtl());
assertTrue(i == event.getMetricF());
assertEquals("batch-host", event.getHost());
assertEquals("test", event.getDescription());
assertEquals(3, event.getTagsCount());
assertEquals(3, event.getAttributesCount());
assertTrue(event.getTagsList().contains("tag1"));
assertTrue(event.getTagsList().contains("tag2"));
assertTrue(event.getTagsList().contains("tag3"));
}
}
Example 8
| Source File: TestVisibilityUtil.java From nifi with Apache License 2.0 | 5 votes |
|
@Test
public void testOnlyColumnFamilyOnFlowfile() {
runner.setProperty("visibility.test", "U&PII");
MockFlowFile ff = new MockFlowFile(System.currentTimeMillis());
ff.putAttributes(new HashMap<String, String>(){{
put("visibility.test", "U&PII&PHI");
}});
ProcessContext context = runner.getProcessContext();
String label = VisibilityUtil.pickVisibilityString("test", "test", ff, context);
Assert.assertNotNull(label);
Assert.assertEquals("U&PII&PHI", label);
}
Example 9
| Source File: TestVisibilityUtil.java From nifi with Apache License 2.0 | 5 votes |
|
@Test
public void testAllPresentOnFlowfile() {
runner.setProperty("visibility.test.test", "U&PII");
MockFlowFile ff = new MockFlowFile(System.currentTimeMillis());
ff.putAttributes(new HashMap<String, String>(){{
put("visibility.test.test", "U&PII&PHI");
}});
ProcessContext context = runner.getProcessContext();
String label = VisibilityUtil.pickVisibilityString("test", "test", ff, context);
Assert.assertNotNull(label);
Assert.assertEquals("U&PII&PHI", label);
}
Example 10
| Source File: TestVisibilityUtil.java From nifi with Apache License 2.0 | 5 votes |
|
@Test
public void testColumnFamilyAttributeOnly() {
MockFlowFile ff = new MockFlowFile(System.currentTimeMillis());
ff.putAttributes(new HashMap<String, String>(){{
put("visibility.test", "U&PII");
}});
ProcessContext context = runner.getProcessContext();
String label = VisibilityUtil.pickVisibilityString("test", "test", ff, context);
Assert.assertNotNull(label);
Assert.assertEquals("U&PII", label);
}
Example 11
| Source File: PutAzureEventHubTest.java From nifi with Apache License 2.0 | 5 votes |
|
@Test
public void testMessageIsSentWithoutPartitioningKeyIfNotSpecifiedOrNotPopulated() {
MockedEventhubClientMockPutAzureEventHub processor = new PutAzureEventHubTest.MockedEventhubClientMockPutAzureEventHub();
MockitoAnnotations.initMocks(processor);
EventHubClient eventHubClient = processor.getEventHubClient();
when(eventHubClient.send(any(EventData.class), anyString()))
.thenThrow(new RuntimeException("Partition-key-full method called despite key is Not required or not populated."));
when(eventHubClient.send(any(EventData.class)))
.thenReturn(CompletableFuture.completedFuture(null));
testRunner = TestRunners.newTestRunner(processor);
setUpStandardTestConfig();
MockFlowFile flowFile = new MockFlowFile(1234);
flowFile.putAttributes(ImmutableMap.of(TEST_PARTITIONING_KEY_ATTRIBUTE_NAME, TEST_PARTITIONING_KEY));
// Key not specified
testRunner.enqueue(flowFile);
testRunner.run(1, true);
Mockito.verify(eventHubClient, never()).send(any(EventData.class), eq(TEST_PARTITIONING_KEY));
Mockito.verify(eventHubClient).send(any(EventData.class));
// Key wanted but not available
testRunner.setProperty(PutAzureEventHub.PARTITIONING_KEY_ATTRIBUTE_NAME, "Non-existing-attribute");
testRunner.enqueue(flowFile);
testRunner.run(1, true);
Mockito.verify(eventHubClient, never()).send(any(EventData.class), eq(TEST_PARTITIONING_KEY));
Mockito.verify(eventHubClient, times(2)).send(any(EventData.class));
}
Example 12
| Source File: TestVisibilityUtil.java From nifi with Apache License 2.0 | 5 votes |
|
@Test
public void testInvalidAttributes() {
runner.setProperty("visibility.test", "U&PII");
MockFlowFile ff = new MockFlowFile(System.currentTimeMillis());
ff.putAttributes(new HashMap<String, String>(){{
put("visibility..test", "U&PII&PHI");
}});
ProcessContext context = runner.getProcessContext();
String label = VisibilityUtil.pickVisibilityString("test", "test", ff, context);
Assert.assertNotNull(label);
Assert.assertEquals("U&PII", label);
}
Example 13
| Source File: ExtractTextProcessorTest.java From nifi-extracttext-processor with Apache License 2.0 | 5 votes |
|
@Test
public void when_running_processor_mime_type_should_be_discovered_for_pdf_input_html() {
try {
final String filename = "simple.pdf";
MockFlowFile flowFile = testRunner.enqueue(new FileInputStream(new File("src/test/resources/" + filename)));
Map<String, String> attrs = new HashMap<String, String>() {{ put("filename", filename); }};
testRunner.setProperty(ExtractTextProcessor.FIELD_HTML_OUTPUT, ExtractTextProcessor.HTML_FORMAT);
flowFile.putAttributes(attrs);
} catch (FileNotFoundException e) {
e.printStackTrace();
}
testRunner.assertValid();
testRunner.run();
testRunner.assertAllFlowFilesTransferred(ExtractTextProcessor.REL_SUCCESS);
List<MockFlowFile> successFiles = testRunner.getFlowFilesForRelationship(ExtractTextProcessor.REL_SUCCESS);
for (MockFlowFile mockFile : successFiles) {
// for ( String attribute : mockFile.getAttributes().keySet() ) {
// System.out.println("Attribute:" + attribute + "=" + mockFile.getAttribute(attribute));
// }
mockFile.assertAttributeExists("mime.type");
mockFile.assertAttributeEquals("mime.type", "text/html");
mockFile.assertAttributeExists("orig.mime.type");
mockFile.assertAttributeEquals("orig.mime.type", "application/pdf");
}
}
Example 14
| Source File: TestPutRiemann.java From localization_nifi with Apache License 2.0 | 5 votes |
|
@Test(expected = AssertionError.class)
public void testFailedDeref() {
TestRunner runner = getTestRunner(true);
MockFlowFile flowFile = new MockFlowFile(1);
Map<String, String> attributes = new HashMap<>();
attributes.put("riemann.metric", "5");
flowFile.putAttributes(attributes);
runner.enqueue(flowFile);
try {
runner.run();
} catch (ProcessException e) {
runner.assertQueueNotEmpty();
throw e;
}
}
Example 15
| Source File: TestPutRiemann.java From localization_nifi with Apache License 2.0 | 5 votes |
|
@Test
public void testInvalidEvents() {
TestRunner runner = getTestRunner();
MockFlowFile flowFile = new MockFlowFile(1);
Map<String, String> attributes = new HashMap<>();
attributes.put("riemann.metric", "NOT A NUMBER");
flowFile.putAttributes(attributes);
runner.enqueue(flowFile);
runner.run();
runner.assertAllFlowFilesTransferred(PutRiemann.REL_FAILURE);
}
Example 16
| Source File: TestPutRiemann.java From localization_nifi with Apache License 2.0 | 5 votes |
|
@Test
public void testBatchedEvents() {
// (2 batches) + (1 remaining event)
int iterations = Integer.parseInt(PutRiemann.BATCH_SIZE.getDefaultValue()) * 2 + 1;
TestRunner runner = getTestRunner();
for (int i = 0; i < iterations; i++) {
Map<String, String> attributes = new HashMap<>();
attributes.put("riemann.metric", Float.toString(i));
attributes.put("riemann.host", "batch-host");
attributes.put("custom.attribute.1", "attr1");
attributes.put("custom.attribute.2", "attr2");
attributes.put("custom.attribute.3", "attr3");
MockFlowFile flowFile = new MockFlowFile(i);
flowFile.putAttributes(attributes);
runner.enqueue(flowFile);
}
runner.run(3);
runner.assertAllFlowFilesTransferred(PutRiemann.REL_SUCCESS);
for (int i = 0; i < iterations; i++) {
Proto.Event event = eventStream.remove();
assertEquals("nifi-test-service", event.getService());
assertTrue(5.0 == event.getTtl());
assertTrue(i == event.getMetricF());
assertEquals("batch-host", event.getHost());
assertEquals("test", event.getDescription());
assertEquals(3, event.getTagsCount());
assertEquals(3, event.getAttributesCount());
assertTrue(event.getTagsList().contains("tag1"));
assertTrue(event.getTagsList().contains("tag2"));
assertTrue(event.getTagsList().contains("tag3"));
}
}
Example 17
| Source File: TestParseUserAgent.java From yauaa with Apache License 2.0 | 4 votes |
|
@Test
public void testParserPartial() {
// Content to be mock a json file
String content = "CONTENT:>>" + TEST_USER_AGENT + "<<";
// Add properties
runner.setProperty(PROPERTY_PREFIX + "DeviceClass", "true");
runner.setProperty(PROPERTY_PREFIX + "OperatingSystemName", "true");
runner.setProperty(PROPERTY_PREFIX + "OperatingSystemVersion", "true");
runner.setProperty(PROPERTY_PREFIX + "AgentClass", "true");
runner.setProperty(PROPERTY_PREFIX + "AgentName", "true");
runner.setProperty(PROPERTY_PREFIX + "AgentNameVersionMajor", "true");
// Add the content to the runner (just because we 'should' have some content).
MockFlowFile flowfile = runner.enqueue(content);
Map<String, String> attributes = new HashMap<>();
attributes.put(ParseUserAgent.USERAGENTSTRING_ATTRIBUTENAME, TEST_USER_AGENT);
flowfile.putAttributes(attributes);
// Run the enqueued content, it also takes an int = number of contents queued
runner.run(1);
// All results were processed with out failure
runner.assertQueueEmpty();
// If you need to read or do additional tests on results you can access the content
List<MockFlowFile> results = runner.getFlowFilesForRelationship(ParseUserAgent.SUCCESS);
assertEquals(1, results.size(), "Must be 1 match");
MockFlowFile result = results.get(0);
result.assertAttributeEquals(ATTRIBUTE_PREFIX + "DeviceClass", "Desktop" );
result.assertAttributeEquals(ATTRIBUTE_PREFIX + "OperatingSystemName", "Linux" );
result.assertAttributeEquals(ATTRIBUTE_PREFIX + "OperatingSystemVersion", "Intel x86_64" );
result.assertAttributeEquals(ATTRIBUTE_PREFIX + "AgentClass", "Browser" );
result.assertAttributeEquals(ATTRIBUTE_PREFIX + "AgentName", "Chrome" );
result.assertAttributeEquals(ATTRIBUTE_PREFIX + "AgentNameVersionMajor", "Chrome 48" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "DeviceName" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "OperatingSystemClass" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "LayoutEngineClass" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "LayoutEngineName" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "LayoutEngineVersion" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "AgentVersion" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "AgentVersionMajor" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "AgentNameVersion" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "AgentBuild" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "AgentInformationEmail" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "AgentInformationUrl" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "AgentLanguage" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "AgentSecurity" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "AgentUuid" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "Anonymized" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "DeviceBrand" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "DeviceCpu" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "DeviceFirmwareVersion" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "DeviceVersion" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "FacebookCarrier" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "FacebookDeviceClass" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "FacebookDeviceName" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "FacebookDeviceVersion" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "FacebookFBOP" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "FacebookFBSS" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "FacebookOperatingSystemName" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "FacebookOperatingSystemVersion" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "HackerAttackVector" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "HackerToolkit" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "KoboAffiliate" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "KoboPlatformId" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "LayoutEngineBuild" );
result.assertAttributeNotExists(ATTRIBUTE_PREFIX + "OperatingSystemVersionBuild" );
// Test attributes and content
result.assertContentEquals(content);
}
Example 18
| Source File: TestSiteToSiteStatusReportingTask.java From nifi with Apache License 2.0 | 4 votes |
|
public static FlowFile createFlowFile(final long id, final Map<String, String> attributes) {
MockFlowFile mockFlowFile = new MockFlowFile(id);
mockFlowFile.putAttributes(attributes);
return mockFlowFile;
}
Example 19
| Source File: PriorityAttributePrioritizerTest.java From nifi with Apache License 2.0 | 4 votes |
|
@Test
public void testPrioritizer() throws InstantiationException, IllegalAccessException {
final Processor processor = new SimpleProcessor();
final AtomicLong idGenerator = new AtomicLong(0L);
final MockProcessSession session = new MockProcessSession(new SharedSessionState(processor, idGenerator), Mockito.mock(Processor.class));
final MockFlowFile ffNoPriority = session.create();
final MockFlowFile ffPri1 = session.create();
ffPri1.putAttributes(attrsPri1);
final MockFlowFile ffPri2 = session.create();
ffPri2.putAttributes(attrsPri2);
final MockFlowFile ffPrin1 = session.create();
ffPrin1.putAttributes(attrsPrin1);
final MockFlowFile ffPriA = session.create();
ffPriA.putAttributes(attrsPriA);
final MockFlowFile ffPriB = session.create();
ffPriB.putAttributes(attrsPriB);
final MockFlowFile ffPriLP = session.create();
ffPriLP.putAttributes(attrsPriLP);
final MockFlowFile ffPriLN = session.create();
ffPriLN.putAttributes(attrsPriLN);
final PriorityAttributePrioritizer prioritizer = new PriorityAttributePrioritizer();
assertEquals(0, prioritizer.compare(null, null));
assertEquals(-1, prioritizer.compare(ffNoPriority, null));
assertEquals(1, prioritizer.compare(null, ffNoPriority));
assertEquals(0, prioritizer.compare(ffNoPriority, ffNoPriority));
assertEquals(-1, prioritizer.compare(ffPri1, ffNoPriority));
assertEquals(1, prioritizer.compare(ffNoPriority, ffPri1));
assertEquals(0, prioritizer.compare(ffPri1, ffPri1));
assertEquals(-1, prioritizer.compare(ffPri1, ffPri2));
assertEquals(1, prioritizer.compare(ffPri2, ffPri1));
assertEquals(-1, prioritizer.compare(ffPrin1, ffPri1));
assertEquals(1, prioritizer.compare(ffPri1, ffPrin1));
assertEquals(-1, prioritizer.compare(ffPri1, ffPriA));
assertEquals(1, prioritizer.compare(ffPriA, ffPri1));
assertEquals(0, prioritizer.compare(ffPriA, ffPriA));
assertEquals(-1, prioritizer.compare(ffPriA, ffPriB));
assertEquals(1, prioritizer.compare(ffPriB, ffPriA));
assertEquals(1, prioritizer.compare(ffPriLP, ffPri1));
assertEquals(-1, prioritizer.compare(ffPri1, ffPriLP));
assertEquals(-1, prioritizer.compare(ffPriLN, ffPri1));
assertEquals(1, prioritizer.compare(ffPri1, ffPriLN));
}
Example 20
| Source File: TestSiteToSiteProvenanceReportingTask.java From localization_nifi with Apache License 2.0 | 4 votes |
|
public static FlowFile createFlowFile(final long id, final Map<String, String> attributes) {
MockFlowFile mockFlowFile = new MockFlowFile(id);
mockFlowFile.putAttributes(attributes);
return mockFlowFile;
}
