Examples with TestProcessor org.elasticsearch.ingest.TestProcessor used on opensource projects

Example 16 with TestProcessor

use of org.elasticsearch.ingest.TestProcessor in project elasticsearch by elastic.

the class TrackingResultProcessorTests method testActualCompoundProcessorWithoutOnFailure.

public void testActualCompoundProcessorWithoutOnFailure() throws Exception {
    RuntimeException exception = new RuntimeException("processor failed");
    TestProcessor testProcessor = new TestProcessor(ingestDocument -> {
        throw exception;
    });
    CompoundProcessor actualProcessor = new CompoundProcessor(testProcessor);
    CompoundProcessor trackingProcessor = decorate(actualProcessor, resultList);
    try {
        trackingProcessor.execute(ingestDocument);
        fail("processor should throw exception");
    } catch (ElasticsearchException e) {
        assertThat(e.getRootCause().getMessage(), equalTo(exception.getMessage()));
    }
    SimulateProcessorResult expectedFirstResult = new SimulateProcessorResult(testProcessor.getTag(), ingestDocument);
    assertThat(testProcessor.getInvokedCounter(), equalTo(1));
    assertThat(resultList.size(), equalTo(1));
    assertThat(resultList.get(0).getIngestDocument(), nullValue());
    assertThat(resultList.get(0).getFailure(), equalTo(exception));
    assertThat(resultList.get(0).getProcessorTag(), equalTo(expectedFirstResult.getProcessorTag()));
}

Example 17 with TestProcessor

use of org.elasticsearch.ingest.TestProcessor in project elasticsearch by elastic.

the class ForEachProcessorTests method testNestedForEach.

public void testNestedForEach() throws Exception {
    List<Map<String, Object>> values = new ArrayList<>();
    List<Object> innerValues = new ArrayList<>();
    innerValues.add("abc");
    innerValues.add("def");
    Map<String, Object> value = new HashMap<>();
    value.put("values2", innerValues);
    values.add(value);
    innerValues = new ArrayList<>();
    innerValues.add("ghi");
    innerValues.add("jkl");
    value = new HashMap<>();
    value.put("values2", innerValues);
    values.add(value);
    IngestDocument ingestDocument = new IngestDocument("_index", "_type", "_id", null, null, Collections.singletonMap("values1", values));
    TestProcessor testProcessor = new TestProcessor(doc -> doc.setFieldValue("_ingest._value", doc.getFieldValue("_ingest._value", String.class).toUpperCase(Locale.ENGLISH)));
    ForEachProcessor processor = new ForEachProcessor("_tag", "values1", new ForEachProcessor("_tag", "_ingest._value.values2", testProcessor));
    processor.execute(ingestDocument);
    List result = ingestDocument.getFieldValue("values1.0.values2", List.class);
    assertThat(result.get(0), equalTo("ABC"));
    assertThat(result.get(1), equalTo("DEF"));
    result = ingestDocument.getFieldValue("values1.1.values2", List.class);
    assertThat(result.get(0), equalTo("GHI"));
    assertThat(result.get(1), equalTo("JKL"));
}

Example 18 with TestProcessor

use of org.elasticsearch.ingest.TestProcessor in project elasticsearch by elastic.

the class ForEachProcessorTests method testScalarValueAllowsUnderscoreValueFieldToRemainAccessible.

public void testScalarValueAllowsUnderscoreValueFieldToRemainAccessible() throws Exception {
    List<Object> values = new ArrayList<>();
    values.add("please");
    values.add("change");
    values.add("me");
    Map<String, Object> source = new HashMap<>();
    source.put("_value", "new_value");
    source.put("values", values);
    IngestDocument ingestDocument = new IngestDocument("_index", "_type", "_id", null, null, source);
    TestProcessor processor = new TestProcessor(doc -> doc.setFieldValue("_ingest._value", doc.getFieldValue("_source._value", String.class)));
    ForEachProcessor forEachProcessor = new ForEachProcessor("_tag", "values", processor);
    forEachProcessor.execute(ingestDocument);
    List result = ingestDocument.getFieldValue("values", List.class);
    assertThat(result.get(0), equalTo("new_value"));
    assertThat(result.get(1), equalTo("new_value"));
    assertThat(result.get(2), equalTo("new_value"));
}

Example 19 with TestProcessor

use of org.elasticsearch.ingest.TestProcessor in project elasticsearch by elastic.

the class ForEachProcessorFactoryTests method testCreate.

public void testCreate() throws Exception {
    Processor processor = new TestProcessor(ingestDocument -> {
    });
    Map<String, Processor.Factory> registry = new HashMap<>();
    registry.put("_name", (r, t, c) -> processor);
    ForEachProcessor.Factory forEachFactory = new ForEachProcessor.Factory();
    Map<String, Object> config = new HashMap<>();
    config.put("field", "_field");
    config.put("processor", Collections.singletonMap("_name", Collections.emptyMap()));
    ForEachProcessor forEachProcessor = forEachFactory.create(registry, null, config);
    assertThat(forEachProcessor, Matchers.notNullValue());
    assertThat(forEachProcessor.getField(), equalTo("_field"));
    assertThat(forEachProcessor.getProcessor(), Matchers.sameInstance(processor));
}