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

Example 36 with MockProcessContext

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

the class AbstractHadoopTest method testErrorConditions.

@Test
public void testErrorConditions() {
    SimpleHadoopProcessor processor = new SimpleHadoopProcessor(kerberosProperties);
    TestRunner runner = TestRunners.newTestRunner(processor);
    Collection<ValidationResult> results;
    ProcessContext pc;
    results = new HashSet<>();
    runner.setProperty(AbstractHadoopProcessor.HADOOP_CONFIGURATION_RESOURCES, "target/classes");
    runner.enqueue(new byte[0]);
    pc = runner.getProcessContext();
    if (pc instanceof MockProcessContext) {
        results = ((MockProcessContext) pc).validate();
    }
    Assert.assertEquals(1, results.size());
    for (ValidationResult vr : results) {
        Assert.assertTrue(vr.toString().contains("is invalid because File target" + File.separator + "classes does not exist or is not a file"));
    }
    results = new HashSet<>();
    runner.setProperty(AbstractHadoopProcessor.HADOOP_CONFIGURATION_RESOURCES, "target/doesnotexist");
    runner.enqueue(new byte[0]);
    pc = runner.getProcessContext();
    if (pc instanceof MockProcessContext) {
        results = ((MockProcessContext) pc).validate();
    }
    Assert.assertEquals(1, results.size());
    for (ValidationResult vr : results) {
        Assert.assertTrue(vr.toString().contains("is invalid because File target" + File.separator + "doesnotexist does not exist or is not a file"));
    }
}

Example 37 with MockProcessContext

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

the class MoveHDFSTest method testOutputDirectoryValidator.

@Test
public void testOutputDirectoryValidator() {
    MoveHDFS proc = new TestableMoveHDFS(kerberosProperties);
    TestRunner runner = TestRunners.newTestRunner(proc);
    Collection<ValidationResult> results;
    ProcessContext pc;
    results = new HashSet<>();
    runner.setProperty(MoveHDFS.INPUT_DIRECTORY_OR_FILE, "/source");
    runner.enqueue(new byte[0]);
    pc = runner.getProcessContext();
    if (pc instanceof MockProcessContext) {
        results = ((MockProcessContext) pc).validate();
    }
    Assert.assertEquals(1, results.size());
    for (ValidationResult vr : results) {
        assertTrue(vr.toString().contains("Output Directory is required"));
    }
}

Example 38 with MockProcessContext

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

the class MoveHDFSTest method testBothInputAndOutputDirectoriesAreValid.

@Test
public void testBothInputAndOutputDirectoriesAreValid() {
    MoveHDFS proc = new TestableMoveHDFS(kerberosProperties);
    TestRunner runner = TestRunners.newTestRunner(proc);
    Collection<ValidationResult> results;
    ProcessContext pc;
    results = new HashSet<>();
    runner.setProperty(MoveHDFS.INPUT_DIRECTORY_OR_FILE, INPUT_DIRECTORY);
    runner.setProperty(MoveHDFS.OUTPUT_DIRECTORY, OUTPUT_DIRECTORY);
    runner.enqueue(new byte[0]);
    pc = runner.getProcessContext();
    if (pc instanceof MockProcessContext) {
        results = ((MockProcessContext) pc).validate();
    }
    Assert.assertEquals(0, results.size());
}

Example 39 with MockProcessContext

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

the class ExecuteGroovyScriptTest method setup.

@Before
public void setup() throws Exception {
    // init processor
    proc = new ExecuteGroovyScript();
    MockProcessContext context = new MockProcessContext(proc);
    MockProcessorInitializationContext initContext = new MockProcessorInitializationContext(proc, context);
    proc.initialize(initContext);
    assertNotNull(proc.getSupportedPropertyDescriptors());
    runner = TestRunners.newTestRunner(proc);
    runner.addControllerService("dbcp", dbcp, new HashMap<>());
    runner.enableControllerService(dbcp);
}