Search in sources :

Example 1 with SimulatePipelineResponse

use of org.elasticsearch.action.ingest.SimulatePipelineResponse in project elasticsearch by elastic.

the class IngestClientIT method testSimulate.

public void testSimulate() throws Exception {
    BytesReference pipelineSource = jsonBuilder().startObject().field("description", "my_pipeline").startArray("processors").startObject().startObject("test").endObject().endObject().endArray().endObject().bytes();
    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 = 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().bytes();
    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", "type", "id", null, null, source);
    assertThat(simulateDocumentBaseResult.getIngestDocument().getSourceAndMetadata(), equalTo(ingestDocument.getSourceAndMetadata()));
    assertThat(simulateDocumentBaseResult.getFailure(), nullValue());
}
Also used : BytesReference(org.elasticsearch.common.bytes.BytesReference) SimulatePipelineResponse(org.elasticsearch.action.ingest.SimulatePipelineResponse) HashMap(java.util.HashMap) SimulatePipelineRequest(org.elasticsearch.action.ingest.SimulatePipelineRequest) GetPipelineResponse(org.elasticsearch.action.ingest.GetPipelineResponse) SimulateDocumentBaseResult(org.elasticsearch.action.ingest.SimulateDocumentBaseResult)

Aggregations

HashMap (java.util.HashMap)1 GetPipelineResponse (org.elasticsearch.action.ingest.GetPipelineResponse)1 SimulateDocumentBaseResult (org.elasticsearch.action.ingest.SimulateDocumentBaseResult)1 SimulatePipelineRequest (org.elasticsearch.action.ingest.SimulatePipelineRequest)1 SimulatePipelineResponse (org.elasticsearch.action.ingest.SimulatePipelineResponse)1 BytesReference (org.elasticsearch.common.bytes.BytesReference)1