Examples with SimulatePipelineResponse org.opensearch.action.ingest.SimulatePipelineResponse used on opensource projects

Example 1 with SimulatePipelineResponse

use of org.opensearch.action.ingest.SimulatePipelineResponse in project OpenSearch by opensearch-project.

the class IngestClientIT method testSimulate.

public void testSimulate() throws Exception {
    BytesReference pipelineSource = BytesReference.bytes(jsonBuilder().startObject().field("description", "my_pipeline").startArray("processors").startObject().startObject("test").endObject().endObject().endArray().endObject());
    client().admin().cluster().preparePutPipeline("_id", pipelineSource, XContentType.JSON).get();
    GetPipelineResponse getResponse = client().admin().cluster().prepareGetPipeline("_id").get();
    assertThat(getResponse.isFound(), is(true));
    assertThat(getResponse.pipelines().size(), equalTo(1));
    assertThat(getResponse.pipelines().get(0).getId(), equalTo("_id"));
    BytesReference bytes = BytesReference.bytes(jsonBuilder().startObject().startArray("docs").startObject().field("_index", "index").field("_type", "type").field("_id", "id").startObject("_source").field("foo", "bar").field("fail", false).endObject().endObject().endArray().endObject());
    SimulatePipelineResponse response;
    if (randomBoolean()) {
        response = client().admin().cluster().prepareSimulatePipeline(bytes, XContentType.JSON).setId("_id").get();
    } else {
        SimulatePipelineRequest request = new SimulatePipelineRequest(bytes, XContentType.JSON);
        request.setId("_id");
        response = client().admin().cluster().simulatePipeline(request).get();
    }
    assertThat(response.isVerbose(), equalTo(false));
    assertThat(response.getPipelineId(), equalTo("_id"));
    assertThat(response.getResults().size(), equalTo(1));
    assertThat(response.getResults().get(0), instanceOf(SimulateDocumentBaseResult.class));
    SimulateDocumentBaseResult simulateDocumentBaseResult = (SimulateDocumentBaseResult) response.getResults().get(0);
    Map<String, Object> source = new HashMap<>();
    source.put("foo", "bar");
    source.put("fail", false);
    source.put("processed", true);
    IngestDocument ingestDocument = new IngestDocument("index", "id", null, null, null, source);
    assertThat(simulateDocumentBaseResult.getIngestDocument().getSourceAndMetadata(), equalTo(ingestDocument.getSourceAndMetadata()));
    assertThat(simulateDocumentBaseResult.getFailure(), nullValue());
    // cleanup
    AcknowledgedResponse deletePipelineResponse = client().admin().cluster().prepareDeletePipeline("_id").get();
    assertTrue(deletePipelineResponse.isAcknowledged());
}

Example 2 with SimulatePipelineResponse

use of org.opensearch.action.ingest.SimulatePipelineResponse in project OpenSearch by opensearch-project.

the class IngestClientIT method testSimulatePipeline.

private void testSimulatePipeline(boolean isVerbose, boolean isFailure) throws IOException {
    XContentType xContentType = randomFrom(XContentType.values());
    XContentBuilder builder = XContentBuilder.builder(xContentType.xContent());
    String rankValue = isFailure ? "non-int" : Integer.toString(1234);
    builder.startObject();
    {
        builder.field("pipeline");
        buildRandomXContentPipeline(builder);
        builder.startArray("docs");
        {
            builder.startObject().field("_index", "index").field("_id", "doc_" + 1).startObject("_source").field("foo", "rab_" + 1).field("rank", rankValue).endObject().endObject();
        }
        builder.endArray();
    }
    builder.endObject();
    SimulatePipelineRequest request = new SimulatePipelineRequest(BytesReference.bytes(builder), builder.contentType());
    request.setVerbose(isVerbose);
    SimulatePipelineResponse response = execute(request, highLevelClient().ingest()::simulate, highLevelClient().ingest()::simulateAsync);
    List<SimulateDocumentResult> results = response.getResults();
    assertEquals(1, results.size());
    if (isVerbose) {
        assertThat(results.get(0), instanceOf(SimulateDocumentVerboseResult.class));
        SimulateDocumentVerboseResult verboseResult = (SimulateDocumentVerboseResult) results.get(0);
        assertEquals(2, verboseResult.getProcessorResults().size());
        if (isFailure) {
            assertNotNull(verboseResult.getProcessorResults().get(1).getFailure());
            assertThat(verboseResult.getProcessorResults().get(1).getFailure().getMessage(), containsString("unable to convert [non-int] to integer"));
        } else {
            assertEquals(verboseResult.getProcessorResults().get(0).getIngestDocument().getFieldValue("foo", String.class), "bar");
            assertEquals(Integer.valueOf(1234), verboseResult.getProcessorResults().get(1).getIngestDocument().getFieldValue("rank", Integer.class));
        }
    } else {
        assertThat(results.get(0), instanceOf(SimulateDocumentBaseResult.class));
        SimulateDocumentBaseResult baseResult = (SimulateDocumentBaseResult) results.get(0);
        if (isFailure) {
            assertNotNull(baseResult.getFailure());
            assertThat(baseResult.getFailure().getMessage(), containsString("unable to convert [non-int] to integer"));
        } else {
            assertNotNull(baseResult.getIngestDocument());
            assertEquals(baseResult.getIngestDocument().getFieldValue("foo", String.class), "bar");
            assertEquals(Integer.valueOf(1234), baseResult.getIngestDocument().getFieldValue("rank", Integer.class));
        }
    }
}