Skip to content

Commit 4e69fa1

Browse files
Dry up more of index setting creation in tests (#95602)
Quick follow-up to #95569 of spots that I could automatically identify and refactor, saving another 1.5k LoC in tests.
1 parent e752135 commit 4e69fa1

File tree

140 files changed

+406
-1890
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

140 files changed

+406
-1890
lines changed

modules/ingest-geoip/src/test/java/org/elasticsearch/ingest/geoip/DatabaseNodeServiceTests.java

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -369,13 +369,7 @@ static ClusterState createClusterState(PersistentTasksCustomMetadata tasksCustom
369369
: GeoIpDownloader.DATABASES_INDEX;
370370
Index index = new Index(indexName, UUID.randomUUID().toString());
371371
IndexMetadata.Builder idxMeta = IndexMetadata.builder(index.getName())
372-
.settings(
373-
Settings.builder()
374-
.put("index.version.created", Version.CURRENT)
375-
.put("index.uuid", index.getUUID())
376-
.put("index.number_of_shards", 1)
377-
.put("index.number_of_replicas", 0)
378-
);
372+
.settings(indexSettings(Version.CURRENT, 1, 0).put("index.uuid", index.getUUID()));
379373
if (aliasGeoipDatabase) {
380374
idxMeta.putAlias(AliasMetadata.builder(GeoIpDownloader.DATABASES_INDEX));
381375
}

modules/reindex/src/internalClusterTest/java/org/elasticsearch/migration/AbstractFeatureMigrationIntegTest.java

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -195,12 +195,7 @@ public void createSystemIndexForDescriptor(SystemIndexDescriptor descriptor) thr
195195
}
196196

197197
static Settings createSettings(Version creationVersion, int flagSettingValue) {
198-
return Settings.builder()
199-
.put(IndexMetadata.INDEX_NUMBER_OF_SHARDS_SETTING.getKey(), 1)
200-
.put(IndexMetadata.INDEX_NUMBER_OF_REPLICAS_SETTING.getKey(), 0)
201-
.put(FlAG_SETTING_KEY, flagSettingValue)
202-
.put("index.version.created", creationVersion)
203-
.build();
198+
return indexSettings(creationVersion, 1, 0).put(FlAG_SETTING_KEY, flagSettingValue).build();
204199
}
205200

206201
static String createMapping(boolean descriptorManaged, boolean descriptorInternal) {

modules/reindex/src/test/java/org/elasticsearch/reindex/ReindexSourceTargetValidationTests.java

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -179,13 +179,7 @@ private static IndexMetadata index(String name, String... aliases) {
179179
}
180180

181181
private static IndexMetadata index(String name, @Nullable Boolean writeIndex, String... aliases) {
182-
IndexMetadata.Builder builder = IndexMetadata.builder(name)
183-
.settings(
184-
Settings.builder()
185-
.put("index.version.created", Version.CURRENT.id)
186-
.put("index.number_of_shards", 1)
187-
.put("index.number_of_replicas", 1)
188-
);
182+
IndexMetadata.Builder builder = IndexMetadata.builder(name).settings(indexSettings(Version.CURRENT, 1, 1));
189183
for (String alias : aliases) {
190184
builder.putAlias(AliasMetadata.builder(alias).writeIndex(writeIndex).build());
191185
}

modules/reindex/src/test/java/org/elasticsearch/reindex/RetryTests.java

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -171,11 +171,7 @@ private void testCase(
171171
.put("node.attr.color", "blue")
172172
.build();
173173
final String node = internalCluster().startDataOnlyNode(nodeSettings);
174-
final Settings indexSettings = Settings.builder()
175-
.put("index.number_of_shards", 1)
176-
.put("index.number_of_replicas", 0)
177-
.put("index.routing.allocation.include.color", "blue")
178-
.build();
174+
final Settings indexSettings = indexSettings(1, 0).put("index.routing.allocation.include.color", "blue").build();
179175

180176
// Create the source index on the node with small thread pools so we can block them.
181177
client().admin().indices().prepareCreate("source").setSettings(indexSettings).execute().actionGet();

qa/multi-cluster-search/src/test/java/org/elasticsearch/search/CCSDuelIT.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ private static void indexDocuments(String idPrefix) throws IOException, Interrup
174174
assertTrue(response.isAcknowledged());
175175

176176
int numShards = randomIntBetween(1, 5);
177-
Settings settings = Settings.builder().put("index.number_of_shards", numShards).put("index.number_of_replicas", 0).build();
177+
Settings settings = indexSettings(numShards, 0).build();
178178
String mapping = """
179179
{
180180
"properties": {

server/src/internalClusterTest/java/org/elasticsearch/action/admin/cluster/allocation/ClusterAllocationExplainIT.java

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1213,12 +1213,7 @@ private void prepareIndex(
12131213
client().admin()
12141214
.indices()
12151215
.prepareCreate("idx")
1216-
.setSettings(
1217-
Settings.builder()
1218-
.put("index.number_of_shards", numPrimaries)
1219-
.put("index.number_of_replicas", numReplicas)
1220-
.put(settings)
1221-
)
1216+
.setSettings(indexSettings(numPrimaries, numReplicas).put(settings))
12221217
.setWaitForActiveShards(activeShardCount)
12231218
.get()
12241219
);

server/src/internalClusterTest/java/org/elasticsearch/action/admin/indices/create/ShrinkIndexIT.java

Lines changed: 11 additions & 41 deletions
Original file line numberDiff line numberDiff line change
@@ -105,13 +105,7 @@ public void testCreateShrinkIndexToN() {
105105
client().admin()
106106
.indices()
107107
.prepareResizeIndex("source", "first_shrink")
108-
.setSettings(
109-
Settings.builder()
110-
.put("index.number_of_replicas", 0)
111-
.put("index.number_of_shards", shardSplits[1])
112-
.putNull("index.blocks.write")
113-
.build()
114-
)
108+
.setSettings(indexSettings(shardSplits[1], 0).putNull("index.blocks.write").build())
115109
.get()
116110
);
117111
ensureGreen();
@@ -139,12 +133,7 @@ public void testCreateShrinkIndexToN() {
139133
.indices()
140134
.prepareResizeIndex("first_shrink", "second_shrink")
141135
.setSettings(
142-
Settings.builder()
143-
.put("index.number_of_replicas", 0)
144-
.put("index.number_of_shards", shardSplits[2])
145-
.putNull("index.blocks.write")
146-
.putNull("index.routing.allocation.require._name")
147-
.build()
136+
indexSettings(shardSplits[2], 0).putNull("index.blocks.write").putNull("index.routing.allocation.require._name").build()
148137
)
149138
.get()
150139
);
@@ -230,10 +219,7 @@ public void testShrinkIndexPrimaryTerm() throws Exception {
230219
final long beforeShrinkPrimaryTerm = IntStream.range(0, numberOfShards).mapToLong(indexMetadata::primaryTerm).max().getAsLong();
231220

232221
// now merge source into target
233-
final Settings shrinkSettings = Settings.builder()
234-
.put("index.number_of_replicas", 0)
235-
.put("index.number_of_shards", numberOfTargetShards)
236-
.build();
222+
final Settings shrinkSettings = indexSettings(numberOfTargetShards, 0).build();
237223
assertAcked(client().admin().indices().prepareResizeIndex("source", "target").setSettings(shrinkSettings).get());
238224

239225
ensureGreen(TimeValue.timeValueSeconds(120));
@@ -476,13 +462,7 @@ public void testCreateShrinkWithIndexSort() throws Exception {
476462
() -> client().admin()
477463
.indices()
478464
.prepareResizeIndex("source", "target")
479-
.setSettings(
480-
Settings.builder()
481-
.put("index.number_of_replicas", 0)
482-
.put("index.number_of_shards", "2")
483-
.put("index.sort.field", "foo")
484-
.build()
485-
)
465+
.setSettings(indexSettings(2, 0).put("index.sort.field", "foo").build())
486466
.get()
487467
);
488468
assertThat(exc.getMessage(), containsString("can't override index sort when resizing an index"));
@@ -492,13 +472,7 @@ public void testCreateShrinkWithIndexSort() throws Exception {
492472
client().admin()
493473
.indices()
494474
.prepareResizeIndex("source", "target")
495-
.setSettings(
496-
Settings.builder()
497-
.put("index.number_of_replicas", 0)
498-
.put("index.number_of_shards", "2")
499-
.putNull("index.blocks.write")
500-
.build()
501-
)
475+
.setSettings(indexSettings(2, 0).putNull("index.blocks.write").build())
502476
.get()
503477
);
504478
ensureGreen();
@@ -620,11 +594,9 @@ public void testShrinkThenSplitWithFailedNode() throws Exception {
620594
.indices()
621595
.prepareResizeIndex("original", "shrunk")
622596
.setSettings(
623-
Settings.builder()
624-
.put(IndexMetadata.SETTING_NUMBER_OF_REPLICAS, 1)
625-
.put(IndexMetadata.SETTING_NUMBER_OF_SHARDS, 1)
626-
.putNull(IndexMetadata.INDEX_ROUTING_REQUIRE_GROUP_SETTING.getConcreteSettingForNamespace("_name").getKey())
627-
.build()
597+
indexSettings(1, 1).putNull(
598+
IndexMetadata.INDEX_ROUTING_REQUIRE_GROUP_SETTING.getConcreteSettingForNamespace("_name").getKey()
599+
).build()
628600
)
629601
.setResizeType(ResizeType.SHRINK)
630602
.get()
@@ -644,11 +616,9 @@ public void testShrinkThenSplitWithFailedNode() throws Exception {
644616
.indices()
645617
.prepareResizeIndex("shrunk", "splitagain")
646618
.setSettings(
647-
Settings.builder()
648-
.put(IndexMetadata.SETTING_NUMBER_OF_REPLICAS, 0)
649-
.put(IndexMetadata.SETTING_NUMBER_OF_SHARDS, shardCount)
650-
.putNull(IndexMetadata.INDEX_ROUTING_REQUIRE_GROUP_SETTING.getConcreteSettingForNamespace("_name").getKey())
651-
.build()
619+
indexSettings(shardCount, 0).putNull(
620+
IndexMetadata.INDEX_ROUTING_REQUIRE_GROUP_SETTING.getConcreteSettingForNamespace("_name").getKey()
621+
).build()
652622
)
653623
.setResizeType(ResizeType.SPLIT)
654624
);

server/src/internalClusterTest/java/org/elasticsearch/action/admin/indices/create/SplitIndexIT.java

Lines changed: 6 additions & 37 deletions
Original file line numberDiff line numberDiff line change
@@ -178,10 +178,7 @@ private void splitToN(int sourceShards, int firstSplitShards, int secondSplitSha
178178
ensureYellow();
179179
updateIndexSettings(Settings.builder().put("index.blocks.write", true), "source");
180180
ensureGreen();
181-
Settings.Builder firstSplitSettingsBuilder = Settings.builder()
182-
.put("index.number_of_replicas", 0)
183-
.put("index.number_of_shards", firstSplitShards)
184-
.putNull("index.blocks.write");
181+
Settings.Builder firstSplitSettingsBuilder = indexSettings(firstSplitShards, 0).putNull("index.blocks.write");
185182
if (sourceShards == 1 && useRoutingPartition == false && randomBoolean()) { // try to set it if we have a source index with 1 shard
186183
firstSplitSettingsBuilder.put("index.number_of_routing_shards", secondSplitShards);
187184
}
@@ -220,13 +217,7 @@ private void splitToN(int sourceShards, int firstSplitShards, int secondSplitSha
220217
.indices()
221218
.prepareResizeIndex("first_split", "second_split")
222219
.setResizeType(ResizeType.SPLIT)
223-
.setSettings(
224-
Settings.builder()
225-
.put("index.number_of_replicas", 0)
226-
.put("index.number_of_shards", secondSplitShards)
227-
.putNull("index.blocks.write")
228-
.build()
229-
)
220+
.setSettings(indexSettings(secondSplitShards, 0).putNull("index.blocks.write").build())
230221
.get()
231222
);
232223
ensureGreen();
@@ -339,11 +330,7 @@ public void testSplitIndexPrimaryTerm() throws Exception {
339330
final long beforeSplitPrimaryTerm = IntStream.range(0, numberOfShards).mapToLong(indexMetadata::primaryTerm).max().getAsLong();
340331

341332
// now split source into target
342-
final Settings splitSettings = Settings.builder()
343-
.put("index.number_of_replicas", 0)
344-
.put("index.number_of_shards", numberOfTargetShards)
345-
.putNull("index.blocks.write")
346-
.build();
333+
final Settings splitSettings = indexSettings(numberOfTargetShards, 0).putNull("index.blocks.write").build();
347334
assertAcked(
348335
client().admin()
349336
.indices()
@@ -398,13 +385,7 @@ public void testCreateSplitIndex() throws Exception {
398385
.indices()
399386
.prepareResizeIndex("source", "target")
400387
.setResizeType(ResizeType.SPLIT)
401-
.setSettings(
402-
Settings.builder()
403-
.put("index.number_of_replicas", createWithReplicas ? 1 : 0)
404-
.put("index.number_of_shards", 2)
405-
.putNull("index.blocks.write")
406-
.build()
407-
)
388+
.setSettings(indexSettings(2, createWithReplicas ? 1 : 0).putNull("index.blocks.write").build())
408389
.get()
409390
);
410391
ensureGreen();
@@ -508,13 +489,7 @@ public void testCreateSplitWithIndexSort() throws Exception {
508489
.indices()
509490
.prepareResizeIndex("source", "target")
510491
.setResizeType(ResizeType.SPLIT)
511-
.setSettings(
512-
Settings.builder()
513-
.put("index.number_of_replicas", 0)
514-
.put("index.number_of_shards", 4)
515-
.put("index.sort.field", "foo")
516-
.build()
517-
)
492+
.setSettings(indexSettings(4, 0).put("index.sort.field", "foo").build())
518493
.get()
519494
);
520495
assertThat(exc.getMessage(), containsString("can't override index sort when resizing an index"));
@@ -525,13 +500,7 @@ public void testCreateSplitWithIndexSort() throws Exception {
525500
.indices()
526501
.prepareResizeIndex("source", "target")
527502
.setResizeType(ResizeType.SPLIT)
528-
.setSettings(
529-
Settings.builder()
530-
.put("index.number_of_replicas", 0)
531-
.put("index.number_of_shards", 4)
532-
.putNull("index.blocks.write")
533-
.build()
534-
)
503+
.setSettings(indexSettings(4, 0).putNull("index.blocks.write").build())
535504
.get()
536505
);
537506
ensureGreen();

server/src/internalClusterTest/java/org/elasticsearch/action/support/WaitActiveShardCountIT.java

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@
1313
import org.elasticsearch.action.admin.indices.create.CreateIndexResponse;
1414
import org.elasticsearch.cluster.health.ClusterHealthStatus;
1515
import org.elasticsearch.common.Priority;
16-
import org.elasticsearch.common.settings.Settings;
1716
import org.elasticsearch.core.Strings;
1817
import org.elasticsearch.rest.RestStatus;
1918
import org.elasticsearch.test.ESIntegTestCase;
@@ -30,11 +29,7 @@
3029
*/
3130
public class WaitActiveShardCountIT extends ESIntegTestCase {
3231
public void testReplicationWaitsForActiveShardCount() throws Exception {
33-
CreateIndexResponse createIndexResponse = prepareCreate(
34-
"test",
35-
1,
36-
Settings.builder().put("index.number_of_shards", 1).put("index.number_of_replicas", 2)
37-
).get();
32+
CreateIndexResponse createIndexResponse = prepareCreate("test", 1, indexSettings(1, 2)).get();
3833

3934
assertAcked(createIndexResponse);
4035

server/src/internalClusterTest/java/org/elasticsearch/aliases/IndexAliasesIT.java

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,6 @@
2727
import org.elasticsearch.cluster.metadata.IndexMetadata;
2828
import org.elasticsearch.cluster.metadata.Metadata;
2929
import org.elasticsearch.common.StopWatch;
30-
import org.elasticsearch.common.settings.Settings;
3130
import org.elasticsearch.core.TimeValue;
3231
import org.elasticsearch.index.query.QueryBuilder;
3332
import org.elasticsearch.index.query.QueryBuilders;
@@ -704,15 +703,7 @@ public void testWaitForAliasCreationMultipleShards() throws Exception {
704703

705704
public void testWaitForAliasCreationSingleShard() throws Exception {
706705
logger.info("--> creating index [test]");
707-
assertAcked(
708-
admin().indices()
709-
.create(
710-
new CreateIndexRequest("test").settings(
711-
Settings.builder().put("index.number_of_replicas", 0).put("index.number_of_shards", 1)
712-
)
713-
)
714-
.get()
715-
);
706+
assertAcked(admin().indices().create(new CreateIndexRequest("test").settings(indexSettings(1, 0))).get());
716707

717708
ensureGreen();
718709

0 commit comments

Comments
 (0)