99package org .opensearch .autoforcemerge ;
1010
1111import com .carrotsearch .randomizedtesting .annotations .ThreadLeakFilters ;
12+
1213import org .opensearch .cluster .metadata .IndexMetadata ;
1314import org .opensearch .common .settings .Settings ;
1415import org .opensearch .common .unit .TimeValue ;
1718import org .opensearch .core .common .unit .ByteSizeValue ;
1819import org .opensearch .index .IndexSettings ;
1920import org .opensearch .index .engine .SegmentsStats ;
21+ import org .opensearch .index .shard .IndexShard ;
2022import org .opensearch .index .store .remote .file .CleanerDaemonThreadLeakFilter ;
2123import org .opensearch .node .Node ;
2224import org .opensearch .remotestore .RemoteStoreBaseIntegTestCase ;
23- import org .opensearch .test .OpenSearchIntegTestCase ;
24- import org .opensearch .index .shard .IndexShard ;
2525import org .opensearch .test .InternalTestCluster ;
26+ import org .opensearch .test .OpenSearchIntegTestCase ;
2627
2728import java .util .concurrent .ExecutionException ;
2829import java .util .concurrent .atomic .AtomicLong ;
2930
3031import static org .opensearch .gateway .remote .RemoteClusterStateService .REMOTE_CLUSTER_STATE_ENABLED_SETTING ;
31- import static org .opensearch .test .hamcrest .OpenSearchAssertions .* ;
32+ import static org .opensearch .test .hamcrest .OpenSearchAssertions .assertAcked ;
3233
3334@ ThreadLeakFilters (filters = CleanerDaemonThreadLeakFilter .class )
3435@ OpenSearchIntegTestCase .ClusterScope (scope = OpenSearchIntegTestCase .Scope .TEST , numDataNodes = 0 , supportsDedicatedMasters = false )
@@ -190,14 +191,30 @@ public void testAutoForceMergeTriggeringBasicWithFiveShardsOfTwoIndex() throws E
190191 internalTestCluster .startClusterManagerOnlyNode (clusterSettings );
191192 String dataNode = internalTestCluster .startDataOnlyNodes (1 , clusterSettings ).getFirst ();
192193 internalCluster ().startWarmOnlyNodes (1 , clusterSettings ).getFirst ();
193- assertAcked (client ().admin ().indices ().prepareCreate (INDEX_NAME_1 ).setSettings (Settings .builder ()
194- .put (IndexMetadata .SETTING_NUMBER_OF_SHARDS , 3 )
195- .put (IndexMetadata .SETTING_NUMBER_OF_REPLICAS , 0 )
196- .build ()).get ());
197- assertAcked (client ().admin ().indices ().prepareCreate (INDEX_NAME_2 ).setSettings (Settings .builder ()
198- .put (IndexMetadata .SETTING_NUMBER_OF_SHARDS , 2 )
199- .put (IndexMetadata .SETTING_NUMBER_OF_REPLICAS , 0 )
200- .build ()).get ());
194+ assertAcked (
195+ client ().admin ()
196+ .indices ()
197+ .prepareCreate (INDEX_NAME_1 )
198+ .setSettings (
199+ Settings .builder ()
200+ .put (IndexMetadata .SETTING_NUMBER_OF_SHARDS , 3 )
201+ .put (IndexMetadata .SETTING_NUMBER_OF_REPLICAS , 0 )
202+ .build ()
203+ )
204+ .get ()
205+ );
206+ assertAcked (
207+ client ().admin ()
208+ .indices ()
209+ .prepareCreate (INDEX_NAME_2 )
210+ .setSettings (
211+ Settings .builder ()
212+ .put (IndexMetadata .SETTING_NUMBER_OF_SHARDS , 2 )
213+ .put (IndexMetadata .SETTING_NUMBER_OF_REPLICAS , 0 )
214+ .build ()
215+ )
216+ .get ()
217+ );
201218
202219 // Each ingestion request creates a segment here
203220 for (int i = 0 ; i < INGESTION_COUNT ; i ++) {
@@ -224,11 +241,10 @@ public void testAutoForceMergeTriggeringBasicWithFiveShardsOfTwoIndex() throws E
224241 SegmentsStats segmentsStatsForShard3Before = shard3 .segmentStats (false , false );
225242 SegmentsStats segmentsStatsForShard4Before = shard4 .segmentStats (false , false );
226243 SegmentsStats segmentsStatsForShard5Before = shard5 .segmentStats (false , false );
227- AtomicLong totalSegments = new AtomicLong (segmentsStatsForShard1Before .getCount ()
228- + segmentsStatsForShard2Before .getCount ()
229- + segmentsStatsForShard3Before .getCount ()
230- + segmentsStatsForShard4Before .getCount ()
231- + segmentsStatsForShard5Before .getCount ());
244+ AtomicLong totalSegments = new AtomicLong (
245+ segmentsStatsForShard1Before .getCount () + segmentsStatsForShard2Before .getCount () + segmentsStatsForShard3Before .getCount ()
246+ + segmentsStatsForShard4Before .getCount () + segmentsStatsForShard5Before .getCount ()
247+ );
232248
233249 assertTrue (totalSegments .get () > 5 );
234250
@@ -240,11 +256,10 @@ public void testAutoForceMergeTriggeringBasicWithFiveShardsOfTwoIndex() throws E
240256 SegmentsStats segmentsStatsForShard3After = shard3 .segmentStats (false , false );
241257 SegmentsStats segmentsStatsForShard4After = shard4 .segmentStats (false , false );
242258 SegmentsStats segmentsStatsForShard5After = shard5 .segmentStats (false , false );
243- totalSegments .set (segmentsStatsForShard1After .getCount ()
244- + segmentsStatsForShard2After .getCount ()
245- + segmentsStatsForShard3After .getCount ()
246- + segmentsStatsForShard4After .getCount ()
247- + segmentsStatsForShard5After .getCount ());
259+ totalSegments .set (
260+ segmentsStatsForShard1After .getCount () + segmentsStatsForShard2After .getCount () + segmentsStatsForShard3After .getCount ()
261+ + segmentsStatsForShard4After .getCount () + segmentsStatsForShard5After .getCount ()
262+ );
248263 // refresh to clear old segments
249264 flushAndRefresh (INDEX_NAME_1 );
250265 flushAndRefresh (INDEX_NAME_2 );
@@ -256,4 +271,3 @@ public void testAutoForceMergeTriggeringBasicWithFiveShardsOfTwoIndex() throws E
256271 assertAcked (client ().admin ().indices ().prepareDelete (INDEX_NAME_2 ).get ());
257272 }
258273}
259-
0 commit comments