Include test features in feature name generation (#116078) (#116089)

Author: thecoop

test/metadata-extractor/src/main/java/org/elasticsearch/extractor/features/HistoricalFeaturesMetadataExtractor.java

@@ -30,6 +30,7 @@
 import java.util.Map;
 import java.util.ServiceLoader;
 import java.util.Set;
+import java.util.stream.Stream;
 
 public class HistoricalFeaturesMetadataExtractor {
     private final ClassLoader classLoader;
@@ -93,7 +94,9 @@ void extractHistoricalFeatureMetadata(CheckedBiConsumer<Map<NodeFeature, Version
         ServiceLoader<FeatureSpecification> featureSpecLoader = ServiceLoader.load(FeatureSpecification.class, classLoader);
         for (FeatureSpecification featureSpecification : featureSpecLoader) {
             historicalFeatures.putAll(featureSpecification.getHistoricalFeatures());
-            featureSpecification.getFeatures().stream().map(NodeFeature::id).forEach(featureNames::add);
+            Stream.concat(featureSpecification.getFeatures().stream(), featureSpecification.getTestFeatures().stream())
+                .map(NodeFeature::id)
+                .forEach(featureNames::add);
         }
         metadataConsumer.accept(historicalFeatures, featureNames);
     }

test/metadata-extractor/src/test/java/org/elasticsearch/extractor/features/HistoricalFeaturesMetadataExtractorTests.java

@@ -31,6 +31,7 @@
 import static org.hamcrest.Matchers.containsInAnyOrder;
 import static org.hamcrest.Matchers.empty;
 import static org.hamcrest.Matchers.hasEntry;
+import static org.hamcrest.Matchers.hasItem;
 import static org.hamcrest.Matchers.hasKey;
 import static org.hamcrest.Matchers.not;
 
@@ -49,6 +50,7 @@ public void testExtractHistoricalMetadata() throws IOException {
         });
         assertThat(nodeFeatureVersionMap, not(anEmptyMap()));
         assertThat(featureNamesSet, not(empty()));
+        assertThat(featureNamesSet, hasItem("test_features_enabled"));
 
         Path outputFile = temporaryFolder.newFile().toPath();
         extractor.generateMetadataFile(outputFile);