3636import static org .hamcrest .Matchers .containsString ;
3737import static org .hamcrest .Matchers .equalTo ;
3838import static org .hamcrest .Matchers .greaterThanOrEqualTo ;
39+ import static org .hamcrest .Matchers .hasSize ;
3940import static org .hamcrest .Matchers .notNullValue ;
4041
4142public abstract class AbstractStringTypeRollingUpgradeIT extends AbstractRollingUpgradeWithSecurityTestCase {
@@ -115,7 +116,7 @@ protected void testIndexing(boolean shouldIncludeKeywordMultifield) throws Excep
115116 assertThat (((Map <?, ?>) settings .get ("defaults" )).get ("index.mapping.source.mode" ), equalTo ("SYNTHETIC" ));
116117
117118 // then continued - verify that the created data stream using the created template
118- LogsdbIndexingRollingUpgradeIT . assertDataStream (dataStreamName , templateId );
119+ assertDataStream (dataStreamName , templateId );
119120
120121 // when/then - run some queries and verify results
121122 ensureGreen (dataStreamName );
@@ -184,7 +185,7 @@ static void createTemplate(String dataStreamName, String id, String template) th
184185 assertOK (client ().performRequest (putIndexTemplateRequest ));
185186 }
186187
187- private void bulkIndex (String dataStreamName , int numRequest , int numDocs ) throws Exception {
188+ static String bulkIndex (String dataStreamName , int numRequest , int numDocs ) throws Exception {
188189 String firstIndex = null ;
189190 Instant startTime = Instant .now ().minusSeconds (60 * 60 );
190191
@@ -202,9 +203,10 @@ private void bulkIndex(String dataStreamName, int numRequest, int numDocs) throw
202203 firstIndex = (String ) ((Map <?, ?>) ((Map <?, ?>) ((List <?>) responseBody .get ("items" )).get (0 )).get ("create" )).get ("_index" );
203204 }
204205 }
206+ return firstIndex ;
205207 }
206208
207- private String bulkIndexRequestBody (String dataStreamName , int numDocs , Instant startTime ) {
209+ static String bulkIndexRequestBody (String dataStreamName , int numDocs , Instant startTime ) {
208210 StringBuilder requestBody = new StringBuilder ();
209211
210212 for (int j = 0 ; j < numDocs ; j ++) {
@@ -245,7 +247,7 @@ private static String randomAlphasDelimitedBySpace(int maxAlphas, int minCodeUni
245247 return String .join (" " , alphas );
246248 }
247249
248- private void recordSmallestMessage (final String dataStreamName , final String message ) {
250+ private static void recordSmallestMessage (final String dataStreamName , final String message ) {
249251 if (smallestMessageMap .containsKey (dataStreamName ) == false || message .compareTo (smallestMessageMap .get (dataStreamName )) < 0 ) {
250252 smallestMessageMap .put (dataStreamName , message );
251253 }
@@ -341,6 +343,16 @@ static Map<String, Object> getIndexSettingsWithDefaults(String index) throws IOE
341343 }
342344 }
343345
346+ static void assertDataStream (String dataStreamName , String templateId ) throws IOException {
347+ var getDataStreamsRequest = new Request ("GET" , "/_data_stream/" + dataStreamName );
348+ var getDataStreamResponse = client ().performRequest (getDataStreamsRequest );
349+ assertOK (getDataStreamResponse );
350+ var dataStreams = entityAsMap (getDataStreamResponse );
351+ assertThat (ObjectPath .evaluate (dataStreams , "data_streams.0.name" ), equalTo (dataStreamName ));
352+ assertThat (ObjectPath .evaluate (dataStreams , "data_streams.0.indices" ), hasSize (1 ));
353+ assertThat (ObjectPath .evaluate (dataStreams , "data_streams.0.template" ), equalTo (templateId ));
354+ }
355+
344356 static String formatInstant (Instant instant ) {
345357 return DateFormatter .forPattern (FormatNames .STRICT_DATE_OPTIONAL_TIME .getName ()).format (instant );
346358 }
0 commit comments