Examples with RemoteInfo org.opensearch.index.reindex.RemoteInfo used on opensource projects

Example 1 with RemoteInfo

use of org.opensearch.index.reindex.RemoteInfo in project OpenSearch by opensearch-project.

the class RequestConvertersTests method testReindex.

public void testReindex() throws IOException {
    ReindexRequest reindexRequest = new ReindexRequest();
    reindexRequest.setSourceIndices("source_idx");
    reindexRequest.setDestIndex("dest_idx");
    Map<String, String> expectedParams = new HashMap<>();
    if (randomBoolean()) {
        XContentBuilder builder = JsonXContent.contentBuilder().prettyPrint();
        RemoteInfo remoteInfo = new RemoteInfo("http", "remote-host", 9200, null, BytesReference.bytes(matchAllQuery().toXContent(builder, ToXContent.EMPTY_PARAMS)), "user", "pass", emptyMap(), RemoteInfo.DEFAULT_SOCKET_TIMEOUT, RemoteInfo.DEFAULT_CONNECT_TIMEOUT);
        reindexRequest.setRemoteInfo(remoteInfo);
    }
    if (randomBoolean()) {
        reindexRequest.setSourceBatchSize(randomInt(100));
    }
    if (randomBoolean()) {
        reindexRequest.setDestOpType("create");
    }
    if (randomBoolean()) {
        reindexRequest.setDestPipeline("my_pipeline");
    }
    if (randomBoolean()) {
        float requestsPerSecond = (float) randomDoubleBetween(0.0, 10.0, false);
        expectedParams.put(RethrottleRequest.REQUEST_PER_SECOND_PARAMETER, Float.toString(requestsPerSecond));
        reindexRequest.setRequestsPerSecond(requestsPerSecond);
    } else {
        expectedParams.put(RethrottleRequest.REQUEST_PER_SECOND_PARAMETER, "-1");
    }
    if (randomBoolean()) {
        reindexRequest.setDestRouting("=cat");
    }
    if (randomBoolean()) {
        if (randomBoolean()) {
            reindexRequest.setMaxDocs(randomIntBetween(100, 1000));
        } else {
            reindexRequest.setSize(randomIntBetween(100, 1000));
        }
    }
    if (randomBoolean()) {
        reindexRequest.setAbortOnVersionConflict(false);
    }
    if (randomBoolean()) {
        String ts = randomTimeValue();
        reindexRequest.setScroll(TimeValue.parseTimeValue(ts, "scroll"));
    }
    if (reindexRequest.getRemoteInfo() == null && randomBoolean()) {
        reindexRequest.setSourceQuery(new TermQueryBuilder("foo", "fooval"));
    }
    if (randomBoolean()) {
        int slices = randomIntBetween(0, 4);
        reindexRequest.setSlices(slices);
        if (slices == 0) {
            expectedParams.put("slices", AbstractBulkByScrollRequest.AUTO_SLICES_VALUE);
        } else {
            expectedParams.put("slices", Integer.toString(slices));
        }
    } else {
        expectedParams.put("slices", "1");
    }
    setRandomTimeout(reindexRequest::setTimeout, ReplicationRequest.DEFAULT_TIMEOUT, expectedParams);
    setRandomWaitForActiveShards(reindexRequest::setWaitForActiveShards, ActiveShardCount.DEFAULT, expectedParams);
    expectedParams.put("scroll", reindexRequest.getScrollTime().getStringRep());
    expectedParams.put("wait_for_completion", Boolean.TRUE.toString());
    Request request = RequestConverters.reindex(reindexRequest);
    assertEquals("/_reindex", request.getEndpoint());
    assertEquals(HttpPost.METHOD_NAME, request.getMethod());
    assertEquals(expectedParams, request.getParameters());
    assertToXContentBody(reindexRequest, request.getEntity());
}