NIFI-1016 Naive fix so org.apache.nifi.prioritizer.PriorityAttributePrioritizerTest builds

This commit is contained in:
Tony Kurc 2015-10-18 20:34:00 -04:00
parent 88b1b844fb
commit 9200542298
1 changed files with 8 additions and 9 deletions

View File

@ -16,14 +16,13 @@
*/ */
package org.apache.nifi.prioritizer; package org.apache.nifi.prioritizer;
import org.apache.nifi.prioritizer.PriorityAttributePrioritizer;
import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
import java.util.HashMap; import java.util.HashMap;
import java.util.Map; import java.util.Map;
import java.util.concurrent.atomic.AtomicLong; import java.util.concurrent.atomic.AtomicLong;
import org.apache.nifi.flowfile.attributes.CoreAttributes;
import org.apache.nifi.processor.AbstractProcessor; import org.apache.nifi.processor.AbstractProcessor;
import org.apache.nifi.processor.ProcessContext; import org.apache.nifi.processor.ProcessContext;
import org.apache.nifi.processor.ProcessSession; import org.apache.nifi.processor.ProcessSession;
@ -48,13 +47,13 @@ public class PriorityAttributePrioritizerTest {
@BeforeClass @BeforeClass
public static void init() { public static void init() {
attrsPri1.put(PriorityAttributePrioritizer.PRIORITY_ATTR, "1"); attrsPri1.put(CoreAttributes.PRIORITY.key(), "1");
attrsPri2.put(PriorityAttributePrioritizer.PRIORITY_ATTR, "2"); attrsPri2.put(CoreAttributes.PRIORITY.key(), "2");
attrsPrin1.put(PriorityAttributePrioritizer.PRIORITY_ATTR, "-1"); attrsPrin1.put(CoreAttributes.PRIORITY.key(), "-1");
attrsPriA.put(PriorityAttributePrioritizer.PRIORITY_ATTR, "A"); attrsPriA.put(CoreAttributes.PRIORITY.key(), "A");
attrsPriB.put(PriorityAttributePrioritizer.PRIORITY_ATTR, "B"); attrsPriB.put(CoreAttributes.PRIORITY.key(), "B");
attrsPriLP.put(PriorityAttributePrioritizer.PRIORITY_ATTR, "5432123456789"); attrsPriLP.put(CoreAttributes.PRIORITY.key(), "5432123456789");
attrsPriLN.put(PriorityAttributePrioritizer.PRIORITY_ATTR, "-5432123456789"); attrsPriLN.put(CoreAttributes.PRIORITY.key(), "-5432123456789");
} }
@Test @Test