Skip to content

Commit

Permalink
Adding unit tests
Browse files Browse the repository at this point in the history 
Signed-off-by: Bharathwaj G <[email protected]>
  • Loading branch information
@bharath-techie
bharath-techie committed Apr 6, 2022
1 parent 751c320 commit eba05e5
Show file tree
Hide file tree
Showing 3 changed files with 205 additions and 33 deletions.
4 changes: 4 additions & 0 deletions server/src/main/java/org/opensearch/action/search/TransportCreatePITAction.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -311,6 +311,10 @@ public String id() {
public TimeValue getCreateTime() {
return createTime;
}

public long getKeepAlive() {
return keepAlive;
}
}

public static class UpdatePitContextResponse extends TransportResponse {
Expand Down
2 changes: 1 addition & 1 deletion server/src/main/java/org/opensearch/search/SearchService.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -985,7 +985,7 @@ public void updatePitIdAndKeepAlive(
throw new SearchContextMissingException(request.getSearchContextId());
}
readerContext.setPitId(request.getPitId());
readerContext.tryUpdateKeepAlive(maxKeepAlive);
readerContext.tryUpdateKeepAlive(request.getKeepAlive());
readerContext.setCreateTimestamp(request.getCreateTime());
listener.onResponse(new TransportCreatePITAction.UpdatePitContextResponse(request.getPitId()));
} catch (Exception e) {
Expand Down
232 changes: 200 additions & 32 deletions server/src/test/java/org/opensearch/search/PitSingleNodeTests.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -11,14 +11,18 @@
import org.opensearch.action.ActionFuture;
import org.opensearch.action.search.CreatePITAction;
import org.opensearch.action.search.CreatePITRequest;
import org.opensearch.action.search.SearchPhaseExecutionException;
import org.opensearch.action.search.SearchResponse;
import org.opensearch.common.settings.Settings;
import org.opensearch.common.unit.TimeValue;
import org.opensearch.index.IndexNotFoundException;
import org.opensearch.search.builder.PointInTimeBuilder;
import org.opensearch.test.OpenSearchSingleNodeTestCase;

import java.util.concurrent.CountDownLatch;
import java.util.concurrent.ExecutionException;

import static org.hamcrest.CoreMatchers.equalTo;
import static org.opensearch.action.support.WriteRequest.RefreshPolicy.IMMEDIATE;
import static org.opensearch.test.hamcrest.OpenSearchAssertions.assertHitCount;

Expand All @@ -28,7 +32,16 @@ protected boolean resetNodeAfterTest() {
return true;
}

public void testCreatePIT() throws ExecutionException, InterruptedException {
@Override
protected Settings nodeSettings() {
// very frequent checks
return Settings.builder()
.put(super.nodeSettings())
.put(SearchService.KEEPALIVE_INTERVAL_SETTING.getKey(), TimeValue.timeValueMillis(1))
.build();
}

public void testCreatePITSuccess() throws ExecutionException, InterruptedException {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();

Expand All @@ -48,6 +61,59 @@ public void testCreatePIT() throws ExecutionException, InterruptedException {
service.doClose(); // this kills the keep-alive reaper we have to reset the node after this test
}

public void testCreatePITWithMultipleIndicesSuccess() throws ExecutionException, InterruptedException {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
createIndex("index1", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index1").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
CreatePITRequest request = new CreatePITRequest(TimeValue.timeValueDays(1), true);
request.setIndices(new String[] { "index", "index1" });
SearchService service = getInstanceFromNode(SearchService.class);

ActionFuture<SearchResponse> execute = client().execute(CreatePITAction.INSTANCE, request);
SearchResponse response = execute.get();
assertEquals(4, response.getSuccessfulShards());
assertEquals(4, service.getActiveContexts());
service.doClose();
}

public void testCreatePITWithShardReplicasSuccess() throws ExecutionException, InterruptedException {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 1).build());
client().prepareIndex("index").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();

CreatePITRequest request = new CreatePITRequest(TimeValue.timeValueDays(1), true);
request.setIndices(new String[] { "index" });
ActionFuture<SearchResponse> execute = client().execute(CreatePITAction.INSTANCE, request);
SearchResponse pitResponse = execute.get();

client().prepareIndex("index").setId("2").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
SearchResponse searchResponse = client().prepareSearch("index")
.setSize(2)
.setPointInTime(new PointInTimeBuilder(pitResponse.pointInTimeId()).setKeepAlive(TimeValue.timeValueDays(1)))
.get();
assertHitCount(searchResponse, 1);

SearchService service = getInstanceFromNode(SearchService.class);
assertEquals(2, service.getActiveContexts());
service.doClose();
}

public void testCreatePITWithNonExistentIndex() {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
CreatePITRequest request = new CreatePITRequest(TimeValue.timeValueDays(1), true);
request.setIndices(new String[] { "index", "index1" });
SearchService service = getInstanceFromNode(SearchService.class);

ActionFuture<SearchResponse> execute = client().execute(CreatePITAction.INSTANCE, request);

ExecutionException ex = expectThrows(ExecutionException.class, execute::get);

assertTrue(ex.getMessage().contains("no such index [index1]"));
assertEquals(0, service.getActiveContexts());
service.doClose();
}

public void testCreatePITOnValidIndexAndThenDeleteIndex() throws ExecutionException, InterruptedException {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
Expand All @@ -59,7 +125,7 @@ public void testCreatePITOnValidIndexAndThenDeleteIndex() throws ExecutionExcept
client().admin().indices().prepareDelete("index").get();

IndexNotFoundException ex = expectThrows(IndexNotFoundException.class, () -> {
SearchResponse searchResponse = client().prepareSearch("index")
client().prepareSearch("index")
.setSize(2)
.setPointInTime(new PointInTimeBuilder(pitResponse.pointInTimeId()).setKeepAlive(TimeValue.timeValueDays(1)))
.get();
Expand Down Expand Up @@ -89,34 +155,35 @@ public void testCreatePITOnCloseIndex() {
service.doClose();
}

public void testCreatePITWithMultipleIndices() throws ExecutionException, InterruptedException {
public void testCreatePITOnValidIndexAndThenCloseIndex() throws ExecutionException, InterruptedException {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
createIndex("index1", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index1").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
CreatePITRequest request = new CreatePITRequest(TimeValue.timeValueDays(1), true);
request.setIndices(new String[] { "index", "index1" });
SearchService service = getInstanceFromNode(SearchService.class);

ActionFuture<SearchResponse> execute = client().execute(CreatePITAction.INSTANCE, request);
SearchResponse response = execute.get();
assertEquals(4, response.getSuccessfulShards());
assertEquals(4, service.getActiveContexts());
service.doClose();
}

public void testCreatePITWithNonExistentIndex() {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
CreatePITRequest request = new CreatePITRequest(TimeValue.timeValueDays(1), true);
request.setIndices(new String[] { "index", "index1" });
SearchService service = getInstanceFromNode(SearchService.class);

request.setIndices(new String[] { "index" });
ActionFuture<SearchResponse> execute = client().execute(CreatePITAction.INSTANCE, request);
SearchResponse pitResponse = execute.get();
SearchService service = getInstanceFromNode(SearchService.class);
assertEquals(2, service.getActiveContexts());
client().admin().indices().prepareClose("index").get();
SearchPhaseExecutionException ex = expectThrows(SearchPhaseExecutionException.class, () -> {
SearchResponse searchResponse = client().prepareSearch("index")
.setSize(2)
.setPointInTime(new PointInTimeBuilder(pitResponse.pointInTimeId()).setKeepAlive(TimeValue.timeValueDays(1)))
.get();
});
assertTrue(ex.shardFailures()[0].reason().contains("SearchContextMissingException"));
assertEquals(0, service.getActiveContexts());

ExecutionException ex = expectThrows(ExecutionException.class, execute::get);

assertTrue(ex.getMessage().contains("no such index [index1]"));
// PIT reader contexts are lost after close, verifying it with open index api
client().admin().indices().prepareOpen("index").get();
ex = expectThrows(SearchPhaseExecutionException.class, () -> {
client().prepareSearch("index")
.setSize(2)
.setPointInTime(new PointInTimeBuilder(pitResponse.pointInTimeId()).setKeepAlive(TimeValue.timeValueDays(1)))
.get();
});
assertTrue(ex.shardFailures()[0].reason().contains("SearchContextMissingException"));
assertEquals(0, service.getActiveContexts());
service.doClose();
}
Expand Down Expand Up @@ -145,28 +212,129 @@ public void testMaxOpenPitContexts() throws Exception {
+ "This limit can be set by changing the [search.max_open_pit_context] setting."
)
);
service.doClose();
}

public void testOpenPitContextsConcurrently() throws Exception {
createIndex("index");
final int maxPitContexts = SearchService.MAX_OPEN_PIT_CONTEXT.get(Settings.EMPTY);
CreatePITRequest request = new CreatePITRequest(TimeValue.timeValueDays(1), true);
request.setIndices(new String[] { "index" });
SearchService service = getInstanceFromNode(SearchService.class);
Thread[] threads = new Thread[randomIntBetween(2, 8)];
CountDownLatch latch = new CountDownLatch(threads.length);
for (int i = 0; i < threads.length; i++) {
threads[i] = new Thread(() -> {
latch.countDown();
try {
latch.await();
for (;;) {
try {
client().execute(CreatePITAction.INSTANCE, request).get();
} catch (ExecutionException e) {
assertTrue(
e.getMessage()
.contains(
"Trying to create too many Point In Time contexts. "
+ "Must be less than or equal to: ["
+ SearchService.MAX_OPEN_PIT_CONTEXT.get(Settings.EMPTY)
+ "]. "
+ "This limit can be set by changing the [search.max_open_pit_context] setting."
)
);
return;
}
}
} catch (Exception e) {
throw new AssertionError(e);
}
});
threads[i].setName("opensearch[node_s_0][search]");
threads[i].start();
}
for (Thread thread : threads) {
thread.join();
}
assertThat(service.getActiveContexts(), equalTo(maxPitContexts));
service.doClose();
}

public void testCreatePITWithShardReplicas() throws ExecutionException, InterruptedException {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 1).build());
public void testSearchWithFirstPhaseKeepAliveExpiry() throws ExecutionException, InterruptedException {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();

CreatePITRequest request = new CreatePITRequest(TimeValue.timeValueDays(1), true);
CreatePITRequest request = new CreatePITRequest(TimeValue.timeValueSeconds(1), true);
request.setIndices(new String[] { "index" });
ActionFuture<SearchResponse> execute = client().execute(CreatePITAction.INSTANCE, request);
SearchResponse pitResponse = execute.get();
SearchService service = getInstanceFromNode(SearchService.class);
assertEquals(2, service.getActiveContexts());
// create pit phase 1 temporary keep alive is 30 seconds and keep alive update takes the max out of existing
// keep alive and passed time alive. so waiting for 31 seconds here.
Thread.sleep(31000);
client().prepareIndex("index").setId("2").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();

SearchPhaseExecutionException ex = expectThrows(SearchPhaseExecutionException.class, () -> {
client().prepareSearch("index")
.setSize(2)
.setPointInTime(new PointInTimeBuilder(pitResponse.pointInTimeId()).setKeepAlive(TimeValue.timeValueDays(1)))
.get();
});
assertTrue(ex.shardFailures()[0].reason().contains("SearchContextMissingException"));
assertEquals(0, service.getActiveContexts());
service.doClose();
}

public void testSearchWithPitSecondPhaseKeepAliveExpiry() throws ExecutionException, InterruptedException {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
CreatePITRequest request = new CreatePITRequest(TimeValue.timeValueMinutes(1), true);
request.setIndices(new String[] { "index" });
ActionFuture<SearchResponse> execute = client().execute(CreatePITAction.INSTANCE, request);
SearchResponse pitResponse = execute.get();
SearchService service = getInstanceFromNode(SearchService.class);
assertEquals(2, service.getActiveContexts());
Thread.sleep(31000);
assertEquals(2, service.getActiveContexts());
Thread.sleep(31000);
assertEquals(0, service.getActiveContexts());
client().prepareIndex("index").setId("2").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
SearchResponse searchResponse = client().prepareSearch("index")
.setSize(2)
.setPointInTime(new PointInTimeBuilder(pitResponse.pointInTimeId()).setKeepAlive(TimeValue.timeValueDays(1)))
.get();
assertHitCount(searchResponse, 1);
SearchPhaseExecutionException ex = expectThrows(SearchPhaseExecutionException.class, () -> {
client().prepareSearch("index")
.setSize(2)
.setPointInTime(new PointInTimeBuilder(pitResponse.pointInTimeId()).setKeepAlive(TimeValue.timeValueDays(1)))
.get();
});
assertTrue(ex.shardFailures()[0].reason().contains("SearchContextMissingException"));
service.doClose();
}

public void testSearchWithPitKeepAliveExtension() throws ExecutionException, InterruptedException {
createIndex("index", Settings.builder().put("index.number_of_shards", 2).put("index.number_of_replicas", 0).build());
client().prepareIndex("index").setId("1").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
CreatePITRequest request = new CreatePITRequest(TimeValue.timeValueSeconds(1), true);
request.setIndices(new String[] { "index" });
ActionFuture<SearchResponse> execute = client().execute(CreatePITAction.INSTANCE, request);
SearchResponse pitResponse = execute.get();
SearchService service = getInstanceFromNode(SearchService.class);
assertEquals(2, service.getActiveContexts());
client().prepareSearch("index")
.setSize(2)
.setPointInTime(new PointInTimeBuilder(pitResponse.pointInTimeId()).setKeepAlive(TimeValue.timeValueMinutes(1)))
.get();
client().prepareIndex("index").setId("2").setSource("field", "value").setRefreshPolicy(IMMEDIATE).get();
Thread.sleep(31000);
assertEquals(2, service.getActiveContexts());
Thread.sleep(31000);
assertEquals(0, service.getActiveContexts());
SearchPhaseExecutionException ex = expectThrows(SearchPhaseExecutionException.class, () -> {
client().prepareSearch("index")
.setSize(2)
.setPointInTime(new PointInTimeBuilder(pitResponse.pointInTimeId()).setKeepAlive(TimeValue.timeValueMinutes(1)))
.get();
});
assertTrue(ex.shardFailures()[0].reason().contains("SearchContextMissingException"));
service.doClose();
}

}

0 comments on commit eba05e5

Please sign in to comment.