Skip to content

Commit 769aae0

Browse files
committed
Fix off-by one error in TransportSearchActionTests
Closes #54156
1 parent ecb071c commit 769aae0

File tree

1 file changed

+10
-12
lines changed

1 file changed

+10
-12
lines changed

server/src/test/java/org/elasticsearch/action/search/TransportSearchActionTests.java

+10-12
Original file line numberDiff line numberDiff line change
@@ -844,9 +844,8 @@ public void testShouldMinimizeRoundtrips() throws Exception {
844844
}
845845
}
846846

847-
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/54156")
848847
public void testShouldPreFilterSearchShards() {
849-
int numIndices = randomIntBetween(1, 10);
848+
int numIndices = randomIntBetween(2, 10);
850849
Index[] indices = new Index[numIndices];
851850
for (int i = 0; i < numIndices; i++) {
852851
String indexName = randomAlphaOfLengthBetween(5, 10);
@@ -856,23 +855,23 @@ public void testShouldPreFilterSearchShards() {
856855
{
857856
SearchRequest searchRequest = new SearchRequest();
858857
assertFalse(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
859-
indices, randomIntBetween(1, 127)));
858+
indices, randomIntBetween(2, 127)));
860859
assertFalse(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
861860
indices, randomIntBetween(127, 10000)));
862861
}
863862
{
864863
SearchRequest searchRequest = new SearchRequest()
865864
.source(new SearchSourceBuilder().query(QueryBuilders.rangeQuery("timestamp")));
866865
assertFalse(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
867-
indices, randomIntBetween(1, 127)));
866+
indices, randomIntBetween(2, 127)));
868867
assertTrue(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
869868
indices, randomIntBetween(127, 10000)));
870869
}
871870
{
872871
SearchRequest searchRequest = new SearchRequest()
873872
.source(new SearchSourceBuilder().sort(SortBuilders.fieldSort("timestamp")));
874873
assertTrue(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
875-
indices, randomIntBetween(1, 127)));
874+
indices, randomIntBetween(2, 127)));
876875
assertTrue(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
877876
indices, randomIntBetween(127, 10000)));
878877
}
@@ -881,15 +880,14 @@ public void testShouldPreFilterSearchShards() {
881880
.source(new SearchSourceBuilder().sort(SortBuilders.fieldSort("timestamp")))
882881
.scroll("5m");
883882
assertTrue(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
884-
indices, randomIntBetween(1, 127)));
883+
indices, randomIntBetween(2, 127)));
885884
assertTrue(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
886885
indices, randomIntBetween(127, 10000)));
887886
}
888887
}
889888

890-
@AwaitsFix(bugUrl = "https://github.com/elastic/elasticsearch/issues/54156")
891889
public void testShouldPreFilterSearchShardsWithReadOnly() {
892-
int numIndices = randomIntBetween(1, 10);
890+
int numIndices = randomIntBetween(2, 10);
893891
int numReadOnly = randomIntBetween(1, numIndices);
894892
Index[] indices = new Index[numIndices];
895893
ClusterBlocks.Builder blocksBuilder = ClusterBlocks.builder();
@@ -908,15 +906,15 @@ public void testShouldPreFilterSearchShardsWithReadOnly() {
908906
{
909907
SearchRequest searchRequest = new SearchRequest();
910908
assertFalse(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
911-
indices, randomIntBetween(1, 127)));
909+
indices, randomIntBetween(2, 127)));
912910
assertFalse(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
913911
indices, randomIntBetween(127, 10000)));
914912
}
915913
{
916914
SearchRequest searchRequest = new SearchRequest()
917915
.source(new SearchSourceBuilder().query(QueryBuilders.rangeQuery("timestamp")));
918916
assertTrue(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
919-
indices, randomIntBetween(1, 127)));
917+
indices, randomIntBetween(2, 127)));
920918
assertTrue(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
921919
indices, randomIntBetween(127, 10000)));
922920
}
@@ -925,7 +923,7 @@ public void testShouldPreFilterSearchShardsWithReadOnly() {
925923
.source(new SearchSourceBuilder().query(QueryBuilders.rangeQuery("timestamp")));
926924
searchRequest.scroll("5s");
927925
assertTrue(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
928-
indices, randomIntBetween(127, 10000)));
926+
indices, randomIntBetween(2, 127)));
929927
assertTrue(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
930928
indices, randomIntBetween(127, 10000)));
931929
}
@@ -934,7 +932,7 @@ public void testShouldPreFilterSearchShardsWithReadOnly() {
934932
.source(new SearchSourceBuilder().query(QueryBuilders.rangeQuery("timestamp")));
935933
searchRequest.searchType(SearchType.DFS_QUERY_THEN_FETCH);
936934
assertFalse(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
937-
indices, randomIntBetween(127, 10000)));
935+
indices, randomIntBetween(2, 127)));
938936
assertFalse(TransportSearchAction.shouldPreFilterSearchShards(clusterState, searchRequest,
939937
indices, randomIntBetween(127, 10000)));
940938
}

0 commit comments

Comments
 (0)