From 731c32ab0d5a0da49b9118975010049ce08a364e Mon Sep 17 00:00:00 2001 From: Taylor Gray Date: Thu, 8 Jun 2023 00:24:27 -0500 Subject: [PATCH] Add integration test to create and delete PIT with AwsSdk2Transport Signed-off-by: Taylor Gray --- .../integTest/aws/AwsSdk2SearchIT.java | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/java-client/src/test/java/org/opensearch/client/opensearch/integTest/aws/AwsSdk2SearchIT.java b/java-client/src/test/java/org/opensearch/client/opensearch/integTest/aws/AwsSdk2SearchIT.java index 1e7f4ce427..62c27eb64b 100644 --- a/java-client/src/test/java/org/opensearch/client/opensearch/integTest/aws/AwsSdk2SearchIT.java +++ b/java-client/src/test/java/org/opensearch/client/opensearch/integTest/aws/AwsSdk2SearchIT.java @@ -13,12 +13,18 @@ import org.opensearch.client.opensearch.OpenSearchAsyncClient; import org.opensearch.client.opensearch.OpenSearchClient; import org.opensearch.client.opensearch._types.OpenSearchException; +import org.opensearch.client.opensearch._types.Time; import org.opensearch.client.opensearch.core.IndexRequest; import org.opensearch.client.opensearch.core.IndexResponse; import org.opensearch.client.opensearch.core.SearchResponse; +import org.opensearch.client.opensearch.core.pit.CreatePitRequest; +import org.opensearch.client.opensearch.core.pit.CreatePitResponse; +import org.opensearch.client.opensearch.core.pit.DeletePitRequest; +import org.opensearch.client.opensearch.core.pit.DeletePitResponse; import org.opensearch.client.opensearch.indices.CreateIndexRequest; import org.opensearch.client.opensearch.indices.OpenSearchIndicesClient; +import java.util.Collections; import java.util.List; import java.util.concurrent.CompletableFuture; @@ -44,6 +50,20 @@ public void testAsyncAsyncClient() throws Exception { testClientAsync(true); } + @Test + public void testCreateDeletePit() throws Exception { + final OpenSearchClient client = getClient(false, null, null); + + final CreatePitResponse createPitResponse = client.createPit(CreatePitRequest.of(createPitRequest -> createPitRequest + .keepAlive(new Time.Builder().time("10m").build()) + .targetIndexes(Collections.singletonList(TEST_INDEX)))); + Assert.assertNotNull(createPitResponse); + Assert.assertNotNull(createPitResponse.pitId()); + + final DeletePitResponse deletePitResponse = client.deletePit(DeletePitRequest.of(deletePitBuilder -> deletePitBuilder.pitId(Collections.singletonList(createPitResponse.pitId())))); + Assert.assertNotNull(deletePitResponse); + } + void testClient(boolean async) throws Exception { resetTestIndex(async); final OpenSearchClient client = getClient(async, null, null);