Unmute TSDBDocValuesFormatSingleNodeTests

The issue was with ZSTD_STORED_FIELDS_FEATURE_FLAG feature flag in release build. The test didn't expect this. Adjusted the test to also take LegacyPerFieldMapperCodec into account.

We should lookinto removeing the ZSTD_STORED_FIELDS_FEATURE_FLAG given that we are not actively looking into using zstd compression for default codec case.

Closes elastic#143197
Closes elastic#143198

Author: martijnvg

muted-tests.yml

@@ -449,12 +449,6 @@ tests:
 - class: org.elasticsearch.index.mapper.ShortFieldMapperTests
   method: testSyntheticSourceWithTranslogSnapshot
   issue: https://github.com/elastic/elasticsearch/issues/143196
-- class: org.elasticsearch.index.codec.tsdb.es819.TSDBDocValuesFormatSingleNodeTests
-  method: testStandardIndexWithTSDBDocValuesFormatSetting
-  issue: https://github.com/elastic/elasticsearch/issues/143197
-- class: org.elasticsearch.index.codec.tsdb.es819.TSDBDocValuesFormatSingleNodeTests
-  method: testTSDBIndexUsesCorrectDocValuesFormat
-  issue: https://github.com/elastic/elasticsearch/issues/143198
 - class: org.elasticsearch.snapshots.SnapshotStressTestsIT
   method: testRandomActivities
   issue: https://github.com/elastic/elasticsearch/issues/143199

server/src/test/java/org/elasticsearch/index/codec/tsdb/es819/TSDBDocValuesFormatSingleNodeTests.java

@@ -16,13 +16,16 @@
 import org.elasticsearch.cluster.metadata.IndexMetadata;
 import org.elasticsearch.common.settings.Settings;
 import org.elasticsearch.index.IndexSettings;
+import org.elasticsearch.index.codec.CodecService;
 import org.elasticsearch.index.codec.Elasticsearch92Lucene103Codec;
+import org.elasticsearch.index.codec.LegacyPerFieldMapperCodec;
 import org.elasticsearch.index.engine.Engine;
 import org.elasticsearch.indices.IndicesService;
 import org.elasticsearch.test.ESSingleNodeTestCase;
 import org.elasticsearch.xcontent.XContentFactory;
 
 import java.util.Set;
+import java.util.function.Function;
 
 import static org.hamcrest.Matchers.equalTo;
 import static org.hamcrest.Matchers.greaterThan;
@@ -86,9 +89,23 @@ private void indexDocuments(String indexName) throws Exception {
     private void assertDocValuesFormat(String indexName, DocValuesFormat expectedFormat, Set<String> expectedFields) {
         var indexService = getInstanceFromNode(IndicesService.class).indexServiceSafe(resolveIndex(indexName));
         var shard = indexService.getShard(0);
-        var codec = (Elasticsearch92Lucene103Codec) shard.getEngineOrNull().config().getCodec();
+        var codec = shard.withEngineOrNull(engine -> engine.config().getCodec());
+        Function<String, DocValuesFormat> docValuesFormatProvider;
+        if (codec instanceof Elasticsearch92Lucene103Codec es92103codec) {
+            docValuesFormatProvider = es92103codec::getDocValuesFormatForField;
+        } else if (codec instanceof CodecService.DeduplicateFieldInfosCodec deduplicateFieldInfosCodec) {
+            if (deduplicateFieldInfosCodec.delegate() instanceof LegacyPerFieldMapperCodec legacyPerFieldMapperCodec) {
+                docValuesFormatProvider = legacyPerFieldMapperCodec::getDocValuesFormatForField;
+            } else {
+                fail("Unexpected codec type: " + codec.getClass().getName());
+                return;
+            }
+        } else {
+            fail("Unexpected codec type: " + codec.getClass().getName());
+            return;
+        }
         for (String field : expectedFields) {
-            DocValuesFormat writerFormat = codec.getDocValuesFormatForField(field);
+            DocValuesFormat writerFormat = docValuesFormatProvider.apply(field);
             assertThat(
                 "IndexWriter codec for field [" + field + "] should return the expected TSDB doc values format instance",
                 writerFormat,